Interpretation of EKG with Image Recognition and Convolutional Neural Networks

Dataset used in the training and evaluation of "Interpretation of EKG with Image Recognition and Convolutional Neural Networks"

Abstract:

Electrocardiograms (EKG) form the backbone of all cardiovascular diagnosis, treatment and follow up. Given the pivotal role it plays in modern medicine, there have been multiple efforts to computerize the EKG interpretation with algorithms to improve efficiency and accuracy. Unfortunately, many of these algorithms are machine specific and run-on proprietary signals generated by that machine, hence not generalizable. We propose the development of an image recognition model which can be used to read standard EKG strips. A convolutional neural network (CNN) was trained to classify 12-lead EKGs between 7 clinically important diagnostic classes. An austere variation of the MobileNetV3 model was trained from the ground up on publicly available labeled training set. The precision per class varies from 52% to 91%. This is a novel approach to EKG interpretation as an image recognition problem.