Convolutional Neural Network-Based ECG-Assisted Diagnosis for Coal Workers

Abstract

Objective: To process and extract electrocardiogram (ECG, ECG, or EKG) features using a convolutional neural network (CNN) to establish an ECG-assisted diagnosis model. Methods: Coal workers who underwent physical examinations at Gequan Mine Hospital and Dongpang Mine Hospital of Hebei Jizhong Energy from July 2020 to September 2020 were selected as the study subjects. The ECG images were preprocessed. We use Python software and convolutional neural network to establish ECG images recognition and classification model.We usecalibration curve, calibration-in-the-large, Brier score, specificity, sensitivity, F1 score, Kappa value, accuracy, and area under the curve (AUC) of ROC to evaluate the performance of the model. Results: The number of abnormal ECG results was 849, and the rate of abnormal results was 25.02%. The test set accuracies of the sinus bradycardia model, nonspecific intraventricular conduction delay model, myocardial ischemia model, and sinus tachycardia model were 97.66%, 96.49%, 93.62%, and 93.02%, respectively; sensitivities were 96.63%, 96.30%, 96.88% and 95.24%, respectively; specificities were 98.78%, 96.67%, 86.67%, and 90.90%, respectively; Brier scores were 0.03, 0.07, 0.09, and 0.11, respectively; Calibration-in-the-large values were 0.026, 0.110, 0.041, and 0.098, respectively. Conclusions: The convolutional neural network model can accurately identify the main ECG abnormality types of coal workers. Additionally, the main ECG abnormalities in these coal company workers were sinus bradycardia, non-specific intraventricular conduction delay, myocardial ischemia, and sinus tachycardia.