Title Arrhythmia classification from ECG signals using transformers and data balancing techniques /
Authors Malūkaitė, Jaunė ; Bernatavičienė, Jolita ; Treigys, Povilas
DOI 10.15388/LMITT.2024.13
Full Text Download
Is Part of Lietuvos magistrantų informatikos ir IT tyrimai: konferencijos darbai, 2024 m. gegužės 10 d... Vilnius : Vilniaus universiteto leidykla. 2024, p. 90-97.. eISSN 2783-784X
Keywords [eng] ECG signals ; classification ; deep learning ; transformer ; focal loss ; data balancing techniques ; heartbeats
Abstract [eng] While many arrhythmias pose minimal threat, certain heart rhythm irregularities elevate the potential for stroke or heart failure. The complexity arises particularly with the supraventricular premature heartbeat which has a resemblance to a normal beat and occurs infrequently. Consequently, this research proposes a data balancing and classification technique that enhances the accuracy of identifying mentioned hard-to-classify heartbeats while maintaining robust metrics for other classes. The study introduces a deep learning framework combined with a multi-head attention transformer, for balancing – under-sampling and synthetic minority oversampling are used. To evaluate the proposed model, various experiments based on real data were conducted. The results were compared with an existing model used in chest belt heartbeat monitoring, and the results show that the transformer model achieved better performance for supraventricular premature heartbeats, at the same time reaching high overall and per-class metrics.
Published Vilnius : Vilniaus universiteto leidykla
Type Conference paper
Language English
Publication date 2024
CC license CC license description