计算机科学 ›› 2020, Vol. 47 ›› Issue (7): 118-124.doi: 10.11896/jsjkx.190600161

所属专题: 医学图像 医学图像

• 计算机图形学&多媒体 • 上一篇    下一篇

基于堆叠式双向LSTM的心电图自动识别算法

王文刀, 王润泽, 魏鑫磊, 漆云亮, 马义德   

  1. 兰州大学信息科学与工程学院 兰州730000
  • 收稿日期:2019-06-26 出版日期:2020-07-15 发布日期:2020-07-16
  • 通讯作者: 马义德(mayide510510@qq.com)
  • 作者简介:897675693@qq.com
  • 基金资助:
    国家自然科学基金(61175012,60872109,60572011);甘肃省自然科学基金(18JR3RA288);中央高校基本科研业务费专项资金(lzujbky-2017-it72)

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)

摘要: 针对日趋增长的心电图数据分析需求,提出了一种新的心电图分类算法。首先对原始数据进行截断固定长度、样本均衡、求取信号的瞬时频率和光谱熵等预处理操作,数据经过预处理后模型能够更好地从其中提取特征进行学习;在训练过程中采用两个双向LSTM(BILSIM)网络堆叠组成的模型,堆叠式的双向LSTM(BILSIM)模型是一种改进的循环神经网络模型,相较于卷积神经网络,循环神经网络更加适合用来处理像心电图这样的序列数据。该模型在Windows下的MATLAB2018b上进行训练和测试,CUDA版本为9.0,采用分类准确率作为衡量模型性能的指标在两个数据集上进行了测试,一个是2017年生理信号挑战赛的数据(下文简称2017数据集),该模型在此数据集上最终分类准确率为97.4%;另一个是2018年生理信号挑战赛的数据(下文简称2018数据集),最终的分类准确率为77.6%,并在所属的MATLAB组获得了第三名的成绩。该算法与传统LSTM网络的结果相比,在2017数据集上提升了5.6%的准确率,在2018数据集上提升了7.6%的准确率;与单层的双向LSTM网络的结果相比,在2017数据集上提升了4.2%的准确率,在2018数据集上提升了5.7%的准确率,这充分验证了该算法的可行性和优势。

关键词: 堆叠式双向LSTM网络, 深度学习, 心电图分类, 心律失常

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

中图分类号: 

  • 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] 徐涌鑫, 赵俊峰, 王亚沙, 谢冰, 杨恺.
时序知识图谱表示学习
Temporal Knowledge Graph Representation Learning
计算机科学, 2022, 49(9): 162-171. https://doi.org/10.11896/jsjkx.220500204
[2] 饶志双, 贾真, 张凡, 李天瑞.
基于Key-Value关联记忆网络的知识图谱问答方法
Key-Value Relational Memory Networks for Question Answering over Knowledge Graph
计算机科学, 2022, 49(9): 202-207. https://doi.org/10.11896/jsjkx.220300277
[3] 汤凌韬, 王迪, 张鲁飞, 刘盛云.
基于安全多方计算和差分隐私的联邦学习方案
Federated Learning Scheme Based on Secure Multi-party Computation and Differential Privacy
计算机科学, 2022, 49(9): 297-305. https://doi.org/10.11896/jsjkx.210800108
[4] 王剑, 彭雨琦, 赵宇斐, 杨健.
基于深度学习的社交网络舆情信息抽取方法综述
Survey of Social Network Public Opinion Information Extraction Based on Deep Learning
计算机科学, 2022, 49(8): 279-293. https://doi.org/10.11896/jsjkx.220300099
[5] 郝志荣, 陈龙, 黄嘉成.
面向文本分类的类别区分式通用对抗攻击方法
Class Discriminative Universal Adversarial Attack for Text Classification
计算机科学, 2022, 49(8): 323-329. https://doi.org/10.11896/jsjkx.220200077
[6] 姜梦函, 李邵梅, 郑洪浩, 张建朋.
基于改进位置编码的谣言检测模型
Rumor Detection Model Based on Improved Position Embedding
计算机科学, 2022, 49(8): 330-335. https://doi.org/10.11896/jsjkx.210600046
[7] 孙奇, 吉根林, 张杰.
基于非局部注意力生成对抗网络的视频异常事件检测方法
Non-local Attention Based Generative Adversarial Network for Video Abnormal Event Detection
计算机科学, 2022, 49(8): 172-177. https://doi.org/10.11896/jsjkx.210600061
[8] 侯钰涛, 阿布都克力木·阿布力孜, 哈里旦木·阿布都克里木.
中文预训练模型研究进展
Advances in Chinese Pre-training Models
计算机科学, 2022, 49(7): 148-163. https://doi.org/10.11896/jsjkx.211200018
[9] 周慧, 施皓晨, 屠要峰, 黄圣君.
基于主动采样的深度鲁棒神经网络学习
Robust Deep Neural Network Learning Based on Active Sampling
计算机科学, 2022, 49(7): 164-169. https://doi.org/10.11896/jsjkx.210600044
[10] 苏丹宁, 曹桂涛, 王燕楠, 王宏, 任赫.
小样本雷达辐射源识别的深度学习方法综述
Survey of Deep Learning for Radar Emitter Identification Based on Small Sample
计算机科学, 2022, 49(7): 226-235. https://doi.org/10.11896/jsjkx.210600138
[11] 胡艳羽, 赵龙, 董祥军.
一种用于癌症分类的两阶段深度特征选择提取算法
Two-stage Deep Feature Selection Extraction Algorithm for Cancer Classification
计算机科学, 2022, 49(7): 73-78. https://doi.org/10.11896/jsjkx.210500092
[12] 程成, 降爱莲.
基于多路径特征提取的实时语义分割方法
Real-time Semantic Segmentation Method Based on Multi-path Feature Extraction
计算机科学, 2022, 49(7): 120-126. https://doi.org/10.11896/jsjkx.210500157
[13] 王君锋, 刘凡, 杨赛, 吕坦悦, 陈峙宇, 许峰.
基于多源迁移学习的大坝裂缝检测
Dam Crack Detection Based on Multi-source Transfer Learning
计算机科学, 2022, 49(6A): 319-324. https://doi.org/10.11896/jsjkx.210500124
[14] 楚玉春, 龚航, 王学芳, 刘培顺.
基于YOLOv4的目标检测知识蒸馏算法研究
Study on Knowledge Distillation of Target Detection Algorithm Based on YOLOv4
计算机科学, 2022, 49(6A): 337-344. https://doi.org/10.11896/jsjkx.210600204
[15] 周志豪, 陈磊, 伍翔, 丘东亮, 梁广升, 曾凡巧.
基于SMOTE-SDSAE-SVM的车载CAN总线入侵检测算法
SMOTE-SDSAE-SVM Based Vehicle CAN Bus Intrusion Detection Algorithm
计算机科学, 2022, 49(6A): 562-570. https://doi.org/10.11896/jsjkx.210700106
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!