Optimizing Electrocardiogram Signal Augmentation for Realistic Synthetic Data in Deep Learning Model

• Published in: Signal Processing Algorithms, Architectures, Arrangements, and Applications Conference proceedings pp. 54 - 59
• Main Authors: Safdar, Muhammad Farhan, Palka, Piotr, Faresi, Ahmed Al, Nowak, Robert Marek
• Format: Conference Proceeding
• Language: English
• Published: Division of Signal Processing and Electronic Systems, Poznan University of Technology (DSPES PUT) 25.09.2024
• Subjects: Atrial Fibrillation Diagnostic; Biological system modeling; Classification algorithms; Convolutional Neural Network; Data augmentation; Data models; Deep learning; Electrocardiogram classification; Electrocardiography; Heart; Signal processing algorithms; Synthetic data; Time-Shift; Training
• ISSN: 2326-0319
• DOI: 10.23919/SPA61993.2024.10715629

Electrocardiograms (ECG) are non-invasive signals and have proven useful in assessing the heart condition. Given the necessity for extensive datasets in ECG classification using deep learning (DL) models, there is a critical imperative to devise data augmentation methods capable of generating synthetic but realistic dataset suitable for training DL model. In this study, we propose a novel approach for augmenting ECG signals, aiming to produce realistic signals while optimizing memory usage and resource requirements. Building upon our previous work in ECG signal augmentation, we revisit the methodology to address limitations observed in the generation of synthetic signals. The existing method segmented ECG signals into fixed-length segments and combined them, occasionally resulting in unrealistic heart cycles within the signals in extreme condition. To address this issue, our proposed technique incorporates R peak detection, signal segmentation, and reordering based on the R-peaks information. We evaluated the proposed method using three benchmark datasets, including PTB-XL, Chapman-Shaoxin from PhysioNet, and the dataset from China Physiological Signal Challenge 2018 (CPSC-2018), for classifying atrial fibrillation from normal samples. Our approach achieved an accuracy of 0.83, sensitivity of 0.86, specificity of 0.80, F_{1}-score of 0.83, and precision of 0.80. These results underscore the effectiveness and efficiency of our method in augmenting ECG signals for various applications in healthcare and biomedical research.