A LSTM and CNN Based Assemble Neural Network Framework for Arrhythmias Classification

Fan Liu, Xingshe Zhou, Jinli Cao, Zhu Wang, Hua Wang, Yanchun Zhang

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

23 Scopus citations

Abstract

This paper puts forward a LSTM and CNN based assemble neural network framework to distinguish different types of arrhythmias by integrating stacked bidirectional long shot-term memory (SB-LSTM) network and two-dimensional convolutional neural network (TD-CNN). Particularly, SB-LSTM is used to mine the long-term dependencies contained in electrocardiogram (ECG) from two directions to model the overall variation trends of ECG, while TD-CNN aims at extracting local information of ECG to characterize the local features of ECG. Moreover, we design an ensemble empirical mode decomposition (EEMD) based signal decomposition layer and a support vector machine based intermediate result fusion layer, by which ECG can be analyzed more effectively, and the final classification results can be more accurate and robust. Experimental results on public INCART arrhythmia database show that our model surpasses three state-of-the-art methods, and obtains 99.1% of accuracy, 99.3% of sensitivity and 98.5% of specificity.

Original languageEnglish
Title of host publication2019 IEEE International Conference on Acoustics, Speech, and Signal Processing, ICASSP 2019 - Proceedings
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages1303-1307
Number of pages5
ISBN (Electronic)9781479981311
DOIs
StatePublished - May 2019
Event44th IEEE International Conference on Acoustics, Speech, and Signal Processing, ICASSP 2019 - Brighton, United Kingdom
Duration: 12 May 201917 May 2019

Publication series

NameICASSP, IEEE International Conference on Acoustics, Speech and Signal Processing - Proceedings
Volume2019-May
ISSN (Print)1520-6149

Conference

Conference44th IEEE International Conference on Acoustics, Speech, and Signal Processing, ICASSP 2019
Country/TerritoryUnited Kingdom
CityBrighton
Period12/05/1917/05/19

Keywords

  • Arrhythmia Classification
  • CNN
  • ECG
  • Ensemble Empirical Mode Decomposition (EEMD)
  • LSTM

Fingerprint

Dive into the research topics of 'A LSTM and CNN Based Assemble Neural Network Framework for Arrhythmias Classification'. Together they form a unique fingerprint.

Cite this