Open AccessA new deep learning model for assisted diagnosis on electrocardiogram
Author(s) -
Eric Ke Wang,
Xi Liu,
Rui Sun,
Fan Wang,
Le Pan,
Chengtao Cai,
Antonia Dimitrakopoulou-Srauss,
Zhe Nie,
Yue Ping Li
Publication year - 2019
Publication title -
mathematical biosciences and engineering
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.451
H-Index - 45
eISSN - 1551-0018
pISSN - 1547-1063
DOI - 10.3934/mbe.2019124
Subject(s) - artificial intelligence , computer science , deep learning , convolutional neural network , convolution (computer science) , pattern recognition (psychology) , fuse (electrical) , machine learning , data modeling , artificial neural network , database , electrical engineering , engineering
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.