Comprehensive electrocardiographic diagnosis based on deep learning.

Journal: Artificial intelligence in medicine
Published Date:

Abstract

Cardiovascular disease (CVD) is the leading cause of death worldwide, and coronary artery disease (CAD) is a major contributor. Early-stage CAD can progress if undiagnosed and left untreated, leading to myocardial infarction (MI) that may induce irreversible heart muscle damage, resulting in heart chamber remodeling and eventual congestive heart failure (CHF). Electrocardiography (ECG) signals can be useful to detect established MI, and may also be helpful for early diagnosis of CAD. For the latter especially, the ECG perturbations can be subtle and potentially misclassified during manual interpretation and/or when analyzed by traditional algorithms found in ECG instrumentation. For automated diagnostic systems (ADS), deep learning techniques are favored over conventional machine learning techniques, due to the automatic feature extraction and selection processes involved. This paper highlights various deep learning algorithms exploited for the classification of ECG signals into CAD, MI, and CHF conditions. The Convolutional Neural Network (CNN), followed by combined CNN and Long Short-Term Memory (LSTM) models, appear to be the most useful architectures for classification. A 16-layer LSTM model was developed in our study and validated using 10-fold cross-validation. A classification accuracy of 98.5% was achieved. Our proposed model has the potential to be a useful diagnostic tool in hospitals for the classification of abnormal ECG signals.

Authors

  • Oh Shu Lih
    Department of Electronic & Computer Engineering, Ngee Ann Polytechnic, Singapore.
  • V Jahmunah
    Department of Electronics and Computer Engineering, Ngee Ann Polytechnic, Singapore.
  • Tan Ru San
    National Heart Centre, Singapore.
  • Edward J Ciaccio
    Department of Medicine, Celiac Disease Center, Columbia University, New York, USA.
  • Toshitaka Yamakawa
    Department of Computer Science and Electrical Engineering, Kumamoto University, Japan.
  • Masayuki Tanabe
    Department of Computer Science and Electrical Engineering, Kumamoto University, Japan; International Research Organization for Advanced Science and Technology (IROAST) Kumamoto University, Kumamoto, Japan.
  • Makiko Kobayashi
    Department of Computer Science and Electrical Engineering, Kumamoto University, Japan.
  • Oliver Faust
    Department of Engineering and Mathematics, Sheffield Hallam University, United Kingdom. Electronic address: o.faust@shu.ac.uk.
  • U Rajendra Acharya
    School of Business (Information Systems), Faculty of Business, Education, Law & Arts, University of Southern Queensland, Darling Heights, Australia.