2023/11/27 by Muhammet Fatih Alkan, Alkan, Muhammet, Gruschen Veldtman +3
Computer Science · Medicine · #Computational Physics and Python Applications #ECG Monitoring and Analysis #FOS: Computer and information sciences #FOS: Electrical engineering #Machine Learning (cs.LG) #Phonocardiography and Auscultation Techniques #Signal Processing (eess.SP) #electronic engineering #information engineering
paper · doi:10.48550/arxiv.2312.09437
openalex publication_date 2023/11/27 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28
Congenital heart disease (CHD) is a relatively rare disease that affects patients at birth and results in extremely heterogeneous anatomical and functional defects. 12-lead ECG signal is routinely collected in CHD patients because it provides significant biomarkers for disease prognosis. However, developing accurate machine learning models is challenging due to the lack of large available datasets. Here, we suggest exploiting the Riemannian geometry of the spatial covariance structure of the ECG signal to improve classification. Firstly, we use covariance augmentation to mix samples across the Riemannian geodesic between corresponding classes. Secondly, we suggest to project the covariance matrices to their respective class Riemannian mean to enhance the quality of feature extraction via tangent space projection. We perform several ablation experiments and demonstrate significant improvement compared to traditional machine learning models and deep learning on ECG time series data.