MFB-CBRNN: A Hybrid Network for MI Detection Using 12-Lead ECGs

This paper proposes a novel hybrid network named multiple-feature-branch convolutional bidirectional recurrent neural network (MFB-CBRNN) for myocardial infarction (MI) detection using 12-lead ECGs. The model efficiently combines convolutional neural network-based and recurrent neural network-based structures. Each feature branch consists of several one-dimensional convolutional and pooling layers, corresponding to a certain lead. All the feature branches are independent from each other, which are utilized to learn the diverse features from different leads. Moreover, a bidirectional long short term memory network is employed to summarize all the feature branches. Its good ability of feature aggregation has been proved by the experiments. Furthermore, the paper develops a novel optimization method, lead random mask (LRM), to alleviate overfitting and implement an implicit ensemble like dropout. The model with LRM can achieve a more accurate MI detection. Class-based and subject-based fivefold cross validations are both carried out using Physikalisch-Technische Bundesanstalt diagnostic database. Totally, there are 148 MI and 52 healthy control subjects involved in the experiments. The MFB-CBRNN achieves an overall accuracy of 99.90% in class-based experiments, and an overall accuracy of 93.08% in subject-based experiments. Compared with other related studies, our algorithm achieves a comparable or even better result on MI detection. Therefore, the MFB-CBRNN has a good generalization capacity and is suitable for MI detection using 12-lead ECGs. It has a potential to assist the real-world MI diagnostics and reduce the burden of cardiologists.

[1]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[2]  M. Mitra,et al.  Cross Wavelet Transform Based Analysis of Electrocardiogram Signals , 2013 .

[3]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

[4]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[5]  Geoffrey E. Hinton,et al.  Speech recognition with deep recurrent neural networks , 2013, 2013 IEEE International Conference on Acoustics, Speech and Signal Processing.

[6]  Reza Tafreshi,et al.  Automated analysis of ECG waveforms with atypical QRS complex morphologies , 2014, Biomed. Signal Process. Control..

[7]  Qiang Chen,et al.  Network In Network , 2013, ICLR.

[8]  Mark E Josephson,et al.  Use of the electrocardiogram in acute myocardial infarction. , 2003, The New England journal of medicine.

[9]  Samarendra Dandapat,et al.  Multiscale Energy and Eigenspace Approach to Detection and Localization of Myocardial Infarction , 2015, IEEE Transactions on Biomedical Engineering.

[10]  Maarten L. Simoons,et al.  The third universal definition of myocardial infarction , 2013 .

[11]  Michael Tschannen,et al.  Convolutional recurrent neural networks for electrocardiogram classification , 2017, 2017 Computing in Cardiology (CinC).

[12]  Leo Breiman,et al.  Bias, Variance , And Arcing Classifiers , 1996 .

[13]  N. Ouyang,et al.  Use of an artificial neural network to analyse an ECG with QS complex in V1–2 leads , 1997, Medical and Biological Engineering and Computing.

[14]  Simon Haykin,et al.  Neural Networks and Learning Machines , 2010 .

[15]  Samy Bengio,et al.  Scheduled Sampling for Sequence Prediction with Recurrent Neural Networks , 2015, NIPS.

[16]  Gholamreza Attarodi,et al.  A New Pattern Recognition Method for Detection and Localization of Myocardial Infarction Using T-Wave Integral and Total Integral as Extracted Features from One Cycle of ECG Signal , 2014 .

[17]  Hao Wang,et al.  Multiple-feature-branch convolutional neural network for myocardial infarction diagnosis using electrocardiogram , 2018, Biomed. Signal Process. Control..

[18]  Willis J. Tompkins,et al.  A Real-Time QRS Detection Algorithm , 1985, IEEE Transactions on Biomedical Engineering.

[19]  Lovekesh Vig,et al.  Anomaly detection in ECG time signals via deep long short-term memory networks , 2015, 2015 IEEE International Conference on Data Science and Advanced Analytics (DSAA).

[20]  Huazhong Yang,et al.  Patient-specific ECG classification based on recurrent neural networks and clustering technique , 2017, 2017 13th IASTED International Conference on Biomedical Engineering (BioMed).

[21]  Jeffrey M. Hausdorff,et al.  Physionet: Components of a New Research Resource for Complex Physiologic Signals". Circu-lation Vol , 2000 .

[22]  Yuan Yu,et al.  TensorFlow: A system for large-scale machine learning , 2016, OSDI.

[23]  Muhammad Arif,et al.  Detection and Localization of Myocardial Infarction using K-nearest Neighbor Classifier , 2012, Journal of Medical Systems.

[24]  Ramesh Kumar Sunkaria,et al.  Inferior myocardial infarction detection using stationary wavelet transform and machine learning approach , 2017, Signal, Image and Video Processing.

[25]  Jin He,et al.  Real-Time Multilead Convolutional Neural Network for Myocardial Infarction Detection , 2018, IEEE Journal of Biomedical and Health Informatics.

[26]  Özal Yildirim,et al.  A novel wavelet sequence based on deep bidirectional LSTM network model for ECG signal classification , 2018, Comput. Biol. Medicine.

[27]  U. Rajendra Acharya,et al.  Automated detection and localization of myocardial infarction using electrocardiogram: a comparative study of different leads , 2016, Knowl. Based Syst..

[28]  Nitin Aji Bhaskar Performance Analysis of Support Vector Machine and Neural Networks in Detection of Myocardial Infarction , 2015 .

[29]  Masoumeh Haghpanahi,et al.  Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network , 2019, Nature Medicine.

[30]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.

[31]  U. Rajendra Acharya,et al.  Automated characterization and classification of coronary artery disease and myocardial infarction by decomposition of ECG signals: A comparative study , 2017, Inf. Sci..

[32]  Guang-Zhong Yang,et al.  Deep Learning for Health Informatics , 2017, IEEE Journal of Biomedical and Health Informatics.

[33]  Borys Surawicz,et al.  Chou's Electrocardiography in Clinical Practice , 2020 .

[34]  U. Rajendra Acharya,et al.  Application of stacked convolutional and long short-term memory network for accurate identification of CAD ECG signals , 2018, Comput. Biol. Medicine.

[35]  Celia Shahnaz,et al.  Detection of inferior myocardial infarction using shallow convolutional neural networks , 2017, 2017 IEEE Region 10 Humanitarian Technology Conference (R10-HTC).

[36]  Hermann Ney,et al.  Translation Modeling with Bidirectional Recurrent Neural Networks , 2014, EMNLP.

[37]  Abhishek Das,et al.  Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization , 2016, 2017 IEEE International Conference on Computer Vision (ICCV).

[38]  Deepta Rajan,et al.  A Generative Modeling Approach to Limited Channel ECG Classification , 2018, 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[39]  U. Rajendra Acharya,et al.  Application of deep convolutional neural network for automated detection of myocardial infarction using ECG signals , 2017, Inf. Sci..

[40]  Li Sun,et al.  ECG Analysis Using Multiple Instance Learning for Myocardial Infarction Detection , 2012, IEEE Transactions on Biomedical Engineering.

[41]  Kuldip K. Paliwal,et al.  Bidirectional recurrent neural networks , 1997, IEEE Trans. Signal Process..

[42]  Sandeep Raj,et al.  Development of robust, fast and efficient QRS complex detector: a methodological review , 2018, Australasian Physical & Engineering Sciences in Medicine.

[43]  Tingting Lv,et al.  Bidirectional Recurrent Neural Network And Convolutional Neural Network (BiRCNN) For ECG Beat Classification , 2018, 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[44]  Jürgen Schmidhuber,et al.  Long Short-Term Memory , 1997, Neural Computation.

[45]  Anthony McGrath,et al.  Understanding the ECG Part 2: ECG basics , 2015 .