Detection of Atrial Fibrillation from Short ECG Signals Using a Hybrid Deep Learning Model

Xiaodan Wu, Zeyu Sui, Chao Hsien Chu, Guanjie Huang

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Atrial fibrillation (AF) is one of the most common arrhythmic complications. The diagnosis of AF usually requires long-term monitoring on the patient’s electrocardiogram (ECG) and then either having a domain expert examine the results, or extracting key features and then using a heuristic rule or data mining method to detect. Recently, researchers have attempted to use deep learning models, such as convolution neural networks (CNN) and/or Long Short-Term Memory (LSTM) neural networks to skip the feature extraction process and achieve good classification results. In this paper we propose a hybrid CNN-LSTM model which uses the short ECG signal from the PhysioNet/CinC Challenges 2017 dataset to explore and evaluate the relative performance of four data mining algorithms and three deep learning architectures, CNN, LSTM and CNN-LSTM. Our results show that all deep learning architectures except LSTM performed much better than machine learning algorithms without needing complicated feature extraction. CNN-LSTM is the best performer, achieving 97.08% accuracy, 95.52% sensitivity, 98.57% specificity, 98.46% precision, 0.99 AUC (Area under the ROC curve) value and 0.97 F1 score. With proper design of configuration, deep learning can be effective for automatic AF detection while data mining methods require domain knowledge and an extensive feature extraction and selection process to get satisfactory results.

Original languageEnglish (US)
Title of host publicationSmart Health - International Conference, ICSH 2019, Proceedings
EditorsHsinchun Chen, Daniel Zeng, Xiangbin Yan, Chunxiao Xing
PublisherSpringer
Pages269-282
Number of pages14
ISBN (Print)9783030344818
DOIs
Publication statusPublished - Jan 1 2019
Event7th International Conference for Smart Health, ICSH 2019 - Shenzhen, China
Duration: Jul 1 2019Jul 2 2019

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume11924 LNCS
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference7th International Conference for Smart Health, ICSH 2019
CountryChina
CityShenzhen
Period7/1/197/2/19

    Fingerprint

All Science Journal Classification (ASJC) codes

  • Theoretical Computer Science
  • Computer Science(all)

Cite this

Wu, X., Sui, Z., Chu, C. H., & Huang, G. (2019). Detection of Atrial Fibrillation from Short ECG Signals Using a Hybrid Deep Learning Model. In H. Chen, D. Zeng, X. Yan, & C. Xing (Eds.), Smart Health - International Conference, ICSH 2019, Proceedings (pp. 269-282). (Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics); Vol. 11924 LNCS). Springer. https://doi.org/10.1007/978-3-030-34482-5_24