Automatic Cardiac Arrhythmia Classification Using Residual Network Combined With Long Short-Term Memory

Yun Kwan Kim, Minji Lee, Hee Seok Song, Seong Whan Lee

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

Diagnosis and classification of arrhythmia, which is associated with abnormal electrical activities in the heart, are critical for clinical treatments. Previous studies focused on the diagnosis of atrial fibrillation, which is the most common arrhythmia in adults. The classification performance achieved by studies on other arrhythmia types is not satisfactory for clinical use owing to the small number of classes (minority classes). In this study, we propose a novel framework for automatic classification that combines a residual network with a squeeze-and-excitation block and a bidirectional long short-term memory. Eight-, four-, and two-class performances were evaluated on the MIT-BIH arrhythmia database (MITDB), the MIT-BIH atrial fibrillation database (AFDB), and the PhysioNet/Computing in the cardiology challenge 2017 database (CinC DB), respectively, and they were superior to the performance achieved by conventional methods. In addition, the classwise F1-score in the minority classes was higher than those of the methods adopted in existing studies. To measure the generalization ability of the proposed framework, AFDB and CinC DB were tested using an MITDB-trained model, and superior performance was achieved compared with ShallowConvNet and DeepConvNet. We performed a cross-subject experiment using AFDB and obtained a statistically higher performance using the proposed method compared with typical machine learning methods. The proposed framework can enable the direct diagnosis of arrhythmia types in clinical trials based on the accurate detection of the minority class.

Original languageEnglish
Article number4005817
JournalIEEE Transactions on Instrumentation and Measurement
Volume71
DOIs
Publication statusPublished - 2022

Keywords

  • Arrhythmia classification
  • Augmentation
  • Electrocardiography (ECG)
  • Few shot
  • Long short-term memory
  • Residual network (ResNet)
  • Squeeze-and-excitation (SE) block

ASJC Scopus subject areas

  • Instrumentation
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Automatic Cardiac Arrhythmia Classification Using Residual Network Combined With Long Short-Term Memory'. Together they form a unique fingerprint.

Cite this