A new deep learning model for assisted diagnosis on electrocardiogram

Math Biosci Eng. 2019 Mar 22;16(4):2481-2491. doi: 10.3934/mbe.2019124.

Abstract

In order to enhance the accuracy of computer aided electrocardiogram analysis, we propose a deep learning model called CBRNN to assist diagnosis on electrocardiogram for clinical medical service. It combines two sub networks which are convolutional neural network (CNN) and bi-directional recurrent neural network (BRNN). In the model, CNN with one-dimension convolution is employed to extract features for each lead of ECG, and BRNN is used to fuse features of different leads to represent deeper features. In the training step, we use more than 40 thousand training data and more than 19 thousand validation data to obtain the optimal parameters of the model. Besides, by validating our model on more than CCDD 120,000 real data, it achieves an 87.69% accuracy rate, higher than popular deep learning models such as CNN and ResNet. Our model has better accuracy than state-of-the-art models and it is also slightly higher than the average accuracy of human judgement. It can be served for the first round screening of ECG examination clinical diagnosis.

Keywords: bi-directional recurrent neural network; clinical medicine; convolutional neural network; electrocardiogram; multi-lead.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Cardiology*
  • Deep Learning*
  • Diagnosis, Computer-Assisted / methods*
  • Electrocardiography*
  • Humans
  • Machine Learning
  • Medical Errors
  • Medical Informatics
  • Models, Cardiovascular
  • Neural Networks, Computer
  • Signal Processing, Computer-Assisted*
  • Skin / pathology*