Optimal Transport based Data Augmentation for Heart Disease Diagnosis and Prediction

01/25/2022

∙

In this paper, we focus on a new method of data augmentation to solve the data imbalance problem within imbalanced ECG datasets to improve the robustness and accuracy of heart disease detection. By using Optimal Transport, we augment the ECG disease data from normal ECG beats to balance the data among different categories. We build a Multi-Feature Transformer (MF-Transformer) as our classification model, where different features are extracted from both time and frequency domains to diagnose various heart conditions. Learning from 12-lead ECG signals, our model is able to distinguish five categories of cardiac conditions. Our results demonstrate 1) the classification models' ability to make competitive predictions on five ECG categories; 2) improvements in accuracy and robustness reflecting the effectiveness of our data augmentation method.

READ FULL TEXT

Optimal Transport based Data Augmentation for Heart Disease Diagnosis and Prediction

GeoECG: Data Augmentation via Wasserstein Geodesic Perturbation for Robust Electrocardiogram Prediction

Data Augmentation for Electrocardiograms

Transforming ECG Diagnosis:An In-depth Review of Transformer-based DeepLearning Models in Cardiovascular Disease Detection

ECGT2T: Electrocardiogram synthesis from Two asynchronous leads to Ten leads

ECG Classification based on Wasserstein Scalar Curvature

Generalizing electrocardiogram delineation: training convolutional neural networks with synthetic data augmentation

Transfer Knowledge from Natural Language to Electrocardiography: Can We Detect Cardiovascular Disease Through Language Models?

Optimal Transport based Data Augmentation for Heart Disease Diagnosis and Prediction

Related Research

GeoECG: Data Augmentation via Wasserstein Geodesic Perturbation for Robust Electrocardiogram Prediction

Data Augmentation for Electrocardiograms

Transforming ECG Diagnosis:An In-depth Review of Transformer-based DeepLearning Models in Cardiovascular Disease Detection

ECGT2T: Electrocardiogram synthesis from Two asynchronous leads to Ten leads

ECG Classification based on Wasserstein Scalar Curvature

Generalizing electrocardiogram delineation: training convolutional neural networks with synthetic data augmentation

Transfer Knowledge from Natural Language to Electrocardiography: Can We Detect Cardiovascular Disease Through Language Models?