Stacking Segment-based CNN with SVM for Recognition of Atrial Fibrillation from Single-lead ECG Recordings

Quang H Nguyen, Binh P Nguyen, Trung B. Nguyen, Trang T T Do, James F. Mbinta, Colin Simpson

Research output: Contribution to journalArticlepeer-review

Abstract

Background and Objective:
Atrial fibrillation (AF) is the most common form of cardiac rhythm disorder. Early detection of AF can result in a lower risk of stroke, heart failure, systemic thromboembolism, and coronary artery disease. AF detection however is challenging due to the need for specialised equipment and professional technicians. Hand-held electrocardiogram (ECG) devices, including wearables, are now available and provide a potential mechanism for detecting AF. We wished to identify AF from short single-lead ECG recordings using a machine learning method.
Methods:
We predicted AF from ECG signals by stacking a support vector machine (SVM) on statistical features of segment-based recognition units produced by a convolutional neural network. We used the ECG dataset from the PhysioNet / Computing in Cardiology Challenge 2017, which contained 8528 ECG recordings, to validate our method.
Results:
ECG recordings were categorised into four classes with an average F1 score of 84.19% under 5-fold cross-validations.
Conclusions:
Our model performed better than other state-of-the-art methods applied to the same dataset using the same metric. This stacking method can be generalised for other problems related to medical signals as it does not require expertise in analysing ECG data.
Original languageEnglish
Article number102672
JournalBiomedical Signal Processing and Control
Volume68
Early online date4 May 2021
DOIs
Publication statusE-pub ahead of print - 4 May 2021

Fingerprint

Dive into the research topics of 'Stacking Segment-based CNN with SVM for Recognition of Atrial Fibrillation from Single-lead ECG Recordings'. Together they form a unique fingerprint.

Cite this