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

Special Issue: Medical Imaging ; Medical Imaging

• 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: Arrhythmia, Deep learning, Electrocardiogram classification, Stacked bidirectional LSTM network

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 Classification
Based 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.Arrhythmia
Classification 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 Recurrent
Neural Networks[J].Studies in Computational Intelligence,2008,552(35):385-443.
[18]DONGHYUN.Long Short-Term Memory Recurrent Neural
Network - 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 Machine
Translation of Rare Words with Subword Units[J].Computer Science,2015,27(5):224-245.
[23]PONS J,LIDY T,SERRA X.Experimenting with Musically
Motivated 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 Recurrent
Neural 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] XU Yong-xin, ZHAO Jun-feng, WANG Ya-sha, XIE Bing, YANG Kai. Temporal Knowledge Graph Representation Learning [J]. Computer Science, 2022, 49(9): 162-171.
[2] RAO Zhi-shuang, JIA Zhen, ZHANG Fan, LI Tian-rui. Key-Value Relational Memory Networks for Question Answering over Knowledge Graph [J]. Computer Science, 2022, 49(9): 202-207.
[3] TANG Ling-tao, WANG Di, ZHANG Lu-fei, LIU Sheng-yun. Federated Learning Scheme Based on Secure Multi-party Computation and Differential Privacy [J]. Computer Science, 2022, 49(9): 297-305.
[4] WANG Jian, PENG Yu-qi, ZHAO Yu-fei, YANG Jian. Survey of Social Network Public Opinion Information Extraction Based on Deep Learning [J]. Computer Science, 2022, 49(8): 279-293.
[5] HAO Zhi-rong, CHEN Long, HUANG Jia-cheng. Class Discriminative Universal Adversarial Attack for Text Classification [J]. Computer Science, 2022, 49(8): 323-329.
[6] JIANG Meng-han, LI Shao-mei, ZHENG Hong-hao, ZHANG Jian-peng. Rumor Detection Model Based on Improved Position Embedding [J]. Computer Science, 2022, 49(8): 330-335.
[7] SUN Qi, JI Gen-lin, ZHANG Jie. Non-local Attention Based Generative Adversarial Network for Video Abnormal Event Detection [J]. Computer Science, 2022, 49(8): 172-177.
[8] HU Yan-yu, ZHAO Long, DONG Xiang-jun. Two-stage Deep Feature Selection Extraction Algorithm for Cancer Classification [J]. Computer Science, 2022, 49(7): 73-78.
[9] CHENG Cheng, JIANG Ai-lian. Real-time Semantic Segmentation Method Based on Multi-path Feature Extraction [J]. Computer Science, 2022, 49(7): 120-126.
[10] HOU Yu-tao, ABULIZI Abudukelimu, ABUDUKELIMU Halidanmu. Advances in Chinese Pre-training Models [J]. Computer Science, 2022, 49(7): 148-163.
[11] ZHOU Hui, SHI Hao-chen, TU Yao-feng, HUANG Sheng-jun. Robust Deep Neural Network Learning Based on Active Sampling [J]. Computer Science, 2022, 49(7): 164-169.
[12] SU Dan-ning, CAO Gui-tao, WANG Yan-nan, WANG Hong, REN He. Survey of Deep Learning for Radar Emitter Identification Based on Small Sample [J]. Computer Science, 2022, 49(7): 226-235.
[13] WANG Jun-feng, LIU Fan, YANG Sai, LYU Tan-yue, CHEN Zhi-yu, XU Feng. Dam Crack Detection Based on Multi-source Transfer Learning [J]. Computer Science, 2022, 49(6A): 319-324.
[14] CHU Yu-chun, GONG Hang, Wang Xue-fang, LIU Pei-shun. Study on Knowledge Distillation of Target Detection Algorithm Based on YOLOv4 [J]. Computer Science, 2022, 49(6A): 337-344.
[15] ZHU Wen-tao, LAN Xian-chao, LUO Huan-lin, YUE Bing, WANG Yang. Remote Sensing Aircraft Target Detection Based on Improved Faster R-CNN [J]. Computer Science, 2022, 49(6A): 378-383.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!