Computer Science ›› 2020, Vol. 47 ›› Issue (7): 118-124.doi: 10.11896/jsjkx.190600161

• Computer Graphics & Multimedia • Previous Articles     Next Articles

Automatic Recognition of ECG Based on Stacked Bidirectional LSTM

WANG Wen-dao, WANG Run-ze, WEI Xin-lei, QI Yun-liang, MA Yi-de   

  1. School of Information Science and Engineering,Lanzhou University,Lanzhou 730000,China
  • Received:2019-06-26 Online:2020-07-15 Published:2020-07-16
  • About author:WANG Wen-dao,born in 1995,postgraduate.His main research interests include in-depth study and processing of medical signals.
    MA Yi-de,born in 1963,Ph.D,professor,is a member of China Computer Federation.His main research interests include digital image processing,biomedical engineering and artificial neural network.
  • Supported by:
    This work was supported by the National Natural Science Foundation of China (61175012,60872109,60572011),Natural Science Foundation of Gansu Province,China(18JR3RA288) and Fundamental Research Funds for the Central Universities of Ministry of Education of China (lzujbky-2017-it72)

Abstract: For the growing demand of ECG data analysis,a new ECG classification algorithm is proposed.Firstly,the original data are truncated by fixed length,sample equilibrium is obtained,and the pre-processing operations such as instantaneous frequency and spectral entropy of the signal are obtained.After the data is preprocessed,the model can better extract features from the data for learning.In training progress,a two-way LSTM network stacking model is adopted.The stacked two-way LSTM model is an improved cyclic neural network model.Compared with convolutional neural networks,the cyclic neural network is more sui-table for processing sequence data such as electrocardiogram.The experiment is conducted using MATLAB2018b under Windows for training and testing.The CUDA version is 9.0.The classification accuracy rate is used as an indicator to measure the performance of the model.The model is tested on two data sets,one is the data of the 2017 Physiological Signal Challenge(hereinafter referred to as the 2017 dataset),the final classification accuracy rate is 97.4%;the other is the data of the 2018 Physiological Signal Challenge (hereinafter referred to as the 2018 dataset),and the final classification accuracy rate is 77.6% on this dataset.The MATLAB group to which it belongs has achieved the third place.This algorithm improves the accuracy of 5.6% in the 2017 dataset and 7.6% in the 2018 dataset compared to the results of the traditional LSTM network.Compared to the results of a single-layer bidirectional LSTM network,in the 2017 data set,the accuracy rate improves 4.2%,and the accuracy rate improves 5.7% in the 2018 data set,which fully verifies the feasibility and advantages of the algorithm.

Key words: Stacked bidirectional LSTM network, Arrhythmia, Electrocardiogram classification, Deep learning

CLC Number: 

  • TP274
[1] MEMBERS W G,MOZAFFARIAN D,BENJAMIN E J,et al.Executive Summary:Heart Disease and Stroke Statistics--2016 Update:A Report from the American Heart Association[J].Circulation,2016,127(1):143-152.
[2] 何方田.临床心电图详解与诊断[M].杭州:浙江大学出版社,2010.
[3] ENGIN M.ECG Beat Classification Using Neuro-Fuzzy Net-work[J].Pattern Recognition Letters,2004,25(15):1715-1722.
[4] YU S N,CHEN Y H.Electrocardiogram Beat ClassificationBased on Wavelet Transformation and Probabilistic Neural Network[J].Pattern Recognition Letters,2007,28(10):1142-1150.
[5] GRAVES A.Long Short-Term Memory[J].Neural Computation,1997,9(8):1735-1780.
[6] TEIJEIRO T,GARCIA C A,CASTRO D,et al.ArrhythmiaClassification from the Abductive Interpretation of Short Single-Lead ECG records[J/OL].http://www.cinc.org/archives/2017/pdf/166-054.pdf.
[7] 陈新.临床心律失常学[M].北京:人民卫生出版社,2009.
[8] LECUNY Y,BENGIO Y,HINTON G.Deep Learning[J].Nature,2015,521 (7553):365-436.
[9] MA Y H.Review of in-depth study [J].Reading Abstracts(Middle),2017,804(6):594-628.
[10] OUYANG K.Biomedical Signal Processing [J].Beijing Biomedi-cal Engineering,2006,445(9):655-720.
[11] JIN L P,DONG J.Deep Learning Algorithms for Clinical ECG Analysis [J].Chinese Science:Information Science,2015,45(3):398-416.
[12] LIPTON Z C,KALE D C,ELKAN C,et al.Learning to Diagnose with Lstm Recurrent Neural Networks[J].ComputerScie-nce,2015,123(4):223-268.
[13] ZIHLMANN M,PEREKRESTENKO D,TSCHANNEN M.Convolutional Recurrent Neural Networks for Electrocardiogram Classification[J].Computer Science,2017,57(12):148-205.
[14] RAJPURKAR P,HANNUN A Y,HAGHPANAHI M,et al.Cardiologist - level Arrhythmia Detection with Convolutional Neural Networks [J].Computer Science,2017,225(5):308-374.
[15] GRAVES A,SCHMIDHUBER J.Framewise Phoneme Classification with Bidirectional LSTM and Other Neural Network Architectures[J].Neural Networks,2005,18(5/6):602-610.
[16] PRAKASH A,HASAN S A,LEE K,et al.Neural Paraphrase Generation with Stacked Residual Lstm Networks[J].2016,65(8):439-512.
[17] GRAVES A.Supervised Sequence Labelling with RecurrentNeural Networks[J].Studies in Computational Intelligence,2008,552(35):385-443.
[18] DONGHYUN.Long Short-Term Memory Recurrent NeuralNetwork - based Acoustic Model Using Connectionist Temporal Classification on a Large-scale Training Corpus[J].China Communications,2017,14(9):23-31.
[19] MIKOLOV T,ZWEIG G.Context Dependent Recurrent Neural Network Language Model[C]//Spoken Language Technology Workshop.2013.
[20] GRAVES A,MOHAMED A R,HINTON G.Speech Recognition with Deep Recurrent Neural Networks[C]//2013 IEEE International Conference on Acoustics,Speech and Signal Proces-sing.IEEE,2013.
[21] YUANFANG R,CHONG T,FEI L I,et al.Relation Classification Via Sequence Features and Bi-directional LSTMs[J].Wuhan University Journal of Natural Sciences,2017,22(6):489-497.
[22] SENNRICH R,HADDOW B,BIRCH A.Neural MachineTranslation of Rare Words with Subword Units[J].Computer Science,2015,27(5):224-245.
[23] PONS J,LIDY T,SERRA X.Experimenting with MusicallyMotivated Convolutional Neural Networks[C]//2016 14th International Workshop on Content-Based Multimedia Indexing (CBMI).IEEE,2016.
[24] WANG D L.Deep Learning Reinvents the Hearing Aid[J].IEEE Spectrum,2017,54(3):32-37.
[25] GRAVES A.Supervised Sequence Labelling with RecurrentNeural Networks[J].Studies in Computational Intelligence,2008,385(6):57-128.
[26] SRIVASTAVA N,HINTON G,KRIZHEVSKY A,et al.Dropout:A Simple Way to Prevent Neural Networks from Overfitting[J].Journal of Machine Learning Research,2014,15(1):1929-1958.
[27] MULVEY J M,RUSZCZYNSKI A.A New Scenario Decompo-sition Method for Large-scale Stochastic Optimization[J].Opera-tions Research,1995,43(3):477-490.
[28] HE K,ZHANG X,REN S,et al.Delving Deep Into Rectifiers:Surpassing Human-level Performance on Imagenet Classification[J].Computer Science,2015,246(12):450-480.
[29] TOWNSEND J T.Theoretical Analysis of an Alphabetic Confusion Matrix[J].Perception & Psychophysics,1971,9(1):40-50.
[1] WANG Rui-ping, JIA Zhen, LIU Chang, CHEN Ze-wei, LI Tian-rui. Deep Interest Factorization Machine Network Based on DeepFM [J]. Computer Science, 2021, 48(1): 226-232.
[2] YU Wen-jia, DING Shi-fei. Conditional Generative Adversarial Network Based on Self-attention Mechanism [J]. Computer Science, 2021, 48(1): 241-246.
[3] TONG Xin, WANG Bin-jun, WANG Run-zheng, PAN Xiao-qin. Survey on Adversarial Sample of Deep Learning Towards Natural Language Processing [J]. Computer Science, 2021, 48(1): 258-267.
[4] DING Yu, WEI Hao, PAN Zhi-song, LIU Xin. Survey of Network Representation Learning [J]. Computer Science, 2020, 47(9): 52-59.
[5] HE Xin, XU Juan, JIN Ying-ying. Action-related Network:Towards Modeling Complete Changeable Action [J]. Computer Science, 2020, 47(9): 123-128.
[6] YE Ya-nan, CHI Jing, YU Zhi-ping, ZHAN Yu-liand ZHANG Cai-ming. Expression Animation Synthesis Based on Improved CycleGan Model and Region Segmentation [J]. Computer Science, 2020, 47(9): 142-149.
[7] DENG Liang, XU Geng-lin, LI Meng-jie, CHEN Zhang-jin. Fast Face Recognition Based on Deep Learning and Multiple Hash Similarity Weighting [J]. Computer Science, 2020, 47(9): 163-168.
[8] BAO Yu-xuan, LU Tian-liang, DU Yan-hui. Overview of Deepfake Video Detection Technology [J]. Computer Science, 2020, 47(9): 283-292.
[9] YUAN Ye, HE Xiao-ge, ZHU Ding-kun, WANG Fu-lee, XIE Hao-ran, WANG Jun, WEI Ming-qiang, GUO Yan-wen. Survey of Visual Image Saliency Detection [J]. Computer Science, 2020, 47(7): 84-91.
[10] LIU Yan, WEN Jing. Complex Scene Text Detection Based on Attention Mechanism [J]. Computer Science, 2020, 47(7): 135-140.
[11] ZHANG Zhi-yang, ZHANG Feng-li, TAN Qi, WANG Rui-jin. Review of Information Cascade Prediction Methods Based on Deep Learning [J]. Computer Science, 2020, 47(7): 141-153.
[12] JIANG Wen-bin, FU Zhi, PENG Jing, ZHU Jian. 4Bit-based Gradient Compression Method for Distributed Deep Learning System [J]. Computer Science, 2020, 47(7): 220-226.
[13] CHEN Jin-yin, ZHANG Dun-Jie, LIN Xiang, XU Xiao-dong and ZHU Zi-ling. False Message Propagation Suppression Based on Influence Maximization [J]. Computer Science, 2020, 47(6A): 17-23.
[14] CHENG Zhe, BAI Qian, ZHANG Hao, WANG Shi-pu and LIANG Yu. Improving Hi-C Data Resolution with Deep Convolutional Neural Networks [J]. Computer Science, 2020, 47(6A): 70-74.
[15] HE Lei, SHAO Zhan-peng, ZHANG Jian-hua and ZHOU Xiao-long. Review of Deep Learning-based Action Recognition Algorithms [J]. Computer Science, 2020, 47(6A): 139-147.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
[1] LEI Li-hui and WANG Jing. Parallelization of LTL Model Checking Based on Possibility Measure[J]. Computer Science, 2018, 45(4): 71 -75 .
[2] SUN Qi, JIN Yan, HE Kun and XU Ling-xuan. Hybrid Evolutionary Algorithm for Solving Mixed Capacitated General Routing Problem[J]. Computer Science, 2018, 45(4): 76 -82 .
[3] ZHANG Jia-nan and XIAO Ming-yu. Approximation Algorithm for Weighted Mixed Domination Problem[J]. Computer Science, 2018, 45(4): 83 -88 .
[4] WU Jian-hui, HUANG Zhong-xiang, LI Wu, WU Jian-hui, PENG Xin and ZHANG Sheng. Robustness Optimization of Sequence Decision in Urban Road Construction[J]. Computer Science, 2018, 45(4): 89 -93 .
[5] SHI Wen-jun, WU Ji-gang and LUO Yu-chun. Fast and Efficient Scheduling Algorithms for Mobile Cloud Offloading[J]. Computer Science, 2018, 45(4): 94 -99 .
[6] ZHOU Yan-ping and YE Qiao-lin. L1-norm Distance Based Least Squares Twin Support Vector Machine[J]. Computer Science, 2018, 45(4): 100 -105 .
[7] LIU Bo-yi, TANG Xiang-yan and CHENG Jie-ren. Recognition Method for Corn Borer Based on Templates Matching in Muliple Growth Periods[J]. Computer Science, 2018, 45(4): 106 -111 .
[8] GENG Hai-jun, SHI Xin-gang, WANG Zhi-liang, YIN Xia and YIN Shao-ping. Energy-efficient Intra-domain Routing Algorithm Based on Directed Acyclic Graph[J]. Computer Science, 2018, 45(4): 112 -116 .
[9] CUI Qiong, LI Jian-hua, WANG Hong and NAN Ming-li. Resilience Analysis Model of Networked Command Information System Based on Node Repairability[J]. Computer Science, 2018, 45(4): 117 -121 .
[10] WANG Zhen-chao, HOU Huan-huan and LIAN Rui. Path Optimization Scheme for Restraining Degree of Disorder in CMT[J]. Computer Science, 2018, 45(4): 122 -125 .