Journal of Shandong University(Engineering Science) ›› 2021, Vol. 51 ›› Issue (3): 30-36.doi: 10.6040/j.issn.1672-3961.0.2020.445

Previous Articles     Next Articles

Automatic detection research of arrhythmia based on CNN-LSTM hybrid model

TAO Liang1, LIU Baoning1, LIANG Wei2*   

  1. 1. School of Information and Electrical Engineering, Shandong Jianzhu University, Jinan 250101, Shandong, China;
    2. School of Electrical Engineering and Automation, Qilu University of Technology, Shandong Academy of Sciences, Jinan 250353, Shandong, China
  • Online:2021-06-20 Published:2021-06-24

Abstract: A hybrid algorithm of convolutional neural network and long short-term memory network was proposed for automatic detection of arrhythmias. The model structure was composed of 5 convolutional layers, 5 pooled layers, 1 LSTM layer and 1 fully connected layer. By taking advantage of CNN's ability to automatically extract features and LSTM's ability to capture dependencies before and after time series, the simple preprocessed ECG signal data were directly input into the hybrid model. The whole model combined the two steps of feature extraction and classifier classification, so as to identify five different arrhythmias more efficiently and accurately. The accuracy, sensitivity and specificity of the test set were 99.48%, 99.47% and 99.86% respectively. The experimental results showed that the proposed method could efficiently and accurately identify different types of arrhythmias.

Key words: CNN, LSTM, automatic detection, electrocardiogram, arrhythmia

CLC Number: 

  • TP391
[1] VAN DRIEST S L,WELLS Q S, STALLINGS S, et al. Association of arrhythmia-related genetic variants with phenotypes documented in electronic medical records[J]. JAMA, 2016, 315(1):47-57.
[2] KANANI P,PADOLE M. ECG heartbeat arrhythmia classification using time-series augmented signals and deep learning approach[J]. Procedia Computer Science, 2020, 171:524-531.
[3] LI Qiao, CADATHUR R, GARI D C. Ventricular fibrillation and tachycardia classification using a machine learning approach[J]. IEEE Transactions on Bio-medical Engineering, 2014, 61(6):1607-1613.
[4] SHADNAZ A, ALIREZA M, MARYAM M. Automatic detection of atrial fibrillation using stationary wavelet transform and support vector machine[J]. Computers in Biology and Medicine, 2015, 60:132-142.
[5] MA Fengying, ZHANG Jingyao, LIANG Wei, et al. Automated classification of atrial fibrillation using artificial neural network for wearable devices[J]. Mathematical Problems in Engineering, 2020, 2020:1-6.
[6] SAHOO S, SUBUDHI A, DASH M, et al. Automatic classification of cardiac arrhythmias based on hybrid features and decision tree algorithm[J]. International Journal of Automation and Computing, 2020, 17(4):551-561.
[7] LITJENS G, KOOI T, BEJNORDI B E, et al. A survey on deep learning in medical image analysis[J]. Medical Image Analysis, 2017, 42:60-88.
[8] XIA Yong, NAREN W, WANG Kuanquan, et al. Detecting atrial fibrillation by deep convolutional neural networks[J]. Computers in Biology and Medicine, 2018, 93:84-92.
[9] ROMDHANE T F, ALHICHRI H, OUNI R, et al. Electrocardiogram heartbeat classification based on a deep convolutional neural network and focal loss[J]. Computers in Biology and Medicine, 2020, 123:103866.
[10] NIU Jinghao, TANG Yongqiang, SUN Zhengya,et al. Inter-patient ECG classification with symbolic representations and multi-perspective convolutional neural networks[J]. IEEE Journal of Biomedical and Health Informatics, 2020, 24(5):1321-1332.
[11] SHARMA A, GARG N, PATIDAR S, et al. Automated pre-screening of arrhythmia using hybrid combination of Fourier-Bessel expansion and LSTM[J]. Computers in Biology and Medicine, 2020, 120:103753.
[12] SAEED S, MOHAMMADHOSEIN O, MATIN H. LSTM-based ECG classification for continuous moni-toring on personal wearable devices[J]. IEEE Journal of Biomedical and Health Informatics, 2020, 24(2):515-523.
[13] HAMMAD M, ZHANG Shanzhuo, WANG Kuanquan. A novel two-dimensional ECG feature extraction and classification algorithm based on convolution neural network for human authentication[J]. Future Generation Computer Systems, 2019, 101: 180-196.
[14] FRANCISCO O,DANIEL R. Deep convolutional and LSTMrecurrent neural networks for multimodal wearable activity recognition[J]. Sensors, 2016, 16(1):115.
[15] GOLDBERGER A L, AMARAL L A N, GLASS L, et al. Physiobank, physiotoolkit, and physionet: com-ponents of a new research resource for complex physiologic signals[J]. Circulation: Journal of the American Heart Association, 2000, 101(23):215-220.
[16] PAN J, TOMPKINS W J. A real-time QRS detection algorithm[J]. IEEE Transactions on Bio-medical Engineering, 1985, 32(3):230-236.
[17] SOREN D, EMILY J, PETER M. Singular values for ReLUlayers[J]. IEEE Transactions on Neural Networks and Learning Systems, 2019:3594-3605.
[18] ZHOU Ji, WANG Haide, WEI Jinlong, et al. Adaptive moment estimation for polynomial nonlinear equalizer in PAM8-based optical interconnects[J]. Optics Express, 2019, 27(22):32210-32216.
[19] SAHOO S, KANUNGO B, BEHERA S, et al. Multiresolution wavelet transform based feature extraction and ECG classification to detect cardiac abnormalities[J]. Measurement, 2017, 108:55-66.
[20] ELHAJ F A, SALIM N, HARRIS A R, et al. Arrhythmia recognition and classification using combined linear and nonlinear features of ECG signals[J]. Computer Methods and Programs in Biomedicine, 2016, 127:52-63.
[21] LI Taiyong, ZHOU Min. ECG classification using wavelet packet entropy and random forests[J]. Entropy, 2016, 18(8):285.
[22] ACHARYA U R, OH S L, HAGIWARA Y, et al. A deep convolutional neural network model to classify heartbeats[J]. Computers in Biology and Medicine, 2017, 89:389-396.
[23] 张异凡,黄亦翔,汪开正,等.用于心律失常识别的LSTM和CNN并行组合模型[J].哈尔滨工业大学学报,2019,51(10):76-82. ZHANG Yifan, HUANG Yixiang, WANG Kaizheng, et al. LSTM and CNN parallel combination models for arrhythmia recognition[J]. Journal of Harbin Institute of Technology, 2019, 51(10):76-82.
[1] LIU Shuai, WANG Lei, DING Xutao. Emotional EEG recognition based on Bi-LSTM [J]. Journal of Shandong University(Engineering Science), 2020, 50(4): 35-39.
[2] An ZHU,Chu XU. Method for super-resolution using parallel interlaced sampling [J]. Journal of Shandong University(Engineering Science), 2020, 50(2): 10-16,26.
[3] Longmao HU,Xuegang HU. Identification of the same product feature based on multi-dimension similarity and sentiment word expansion [J]. Journal of Shandong University(Engineering Science), 2020, 50(2): 50-59.
[4] HE Zhengyi, ZENG Xianhua, GUO Jiang. An ensemble method with convolutional neural network and deep belief network for gait recognition and simulation [J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2018, 48(3): 88-95.
[5] YANG Xiulin1, HUANG Shuo2*, DENG Miao1, ZHANG Jihong1,3. Image fusion method based on saliency computation and adaptive PCNN [J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2014, 44(2): 35-42.
Full text



[1] ZHANG Yong-hua,WANG An-ling,LIU Fu-ping . The reflected phase angle of low frequent inhomogeneous[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2006, 36(2): 22 -25 .
[2] LI Liang, LUO Qiming, CHEN Enhong. Graph-based ranking model for object-level search
[3] CHEN Rui, LI Hongwei, TIAN Jing. The relationship between the number of magnetic poles and the bearing capacity of radial magnetic bearing[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2018, 48(2): 81 -85 .
[4] WANG Bo,WANG Ning-sheng . Automatic generation and combinatory optimization of disassembly sequence for mechanical-electric assembly[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2006, 36(2): 52 -57 .
[5] LI Ke,LIU Chang-chun,LI Tong-lei . Medical registration approach using improved maximization of mutual information[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2006, 36(2): 107 -110 .
[6] JI Tao,GAO Xu/sup>,SUN Tong-jing,XUE Yong-duan/sup>,XU Bing-yin/sup> . Characteristic analysis of fault generated traveling waves in 10 Kv automatic blocking and continuous power transmission lines[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2006, 36(2): 111 -116 .
[8] QIN Tong, SUN Fengrong*, WANG Limei, WANG Qinghao, LI Xincai. 3D surface reconstruction using the shape based interpolation guided by maximal discs[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2010, 40(3): 1 -5 .
[9] ZHANG Ying,LANG Yongmei,ZHAO Yuxiao,ZHANG Jianda,QIAO Peng,LI Shanping . Research on technique of aerobic granular sludge cultivationby seeding EGSB anaerobic granular sludge[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2006, 36(4): 56 -59 .
[10] Yue Khing Toh1, XIAO Wendong2, XIE Lihua1. Wireless sensor network for distributed target tracking: practices via real test bed development[J]. JOURNAL OF SHANDONG UNIVERSITY (ENGINEERING SCIENCE), 2009, 39(1): 50 -56 .