Visualization deep learning model for automatic arrhythmias classification.

Researchers

Journal

Modalities

Models

Abstract

With the improvement of living standards, heart disease has become one of the common diseases that threaten human health. As a widely used reliable, non-invasive technology, the Electrocardiography (ECG) data has been increasingly used for diagnosing cardiovascular diseases. With the rapid growth of ECG examinations and the shortage of cardiologists, accurate and automatic recognition of ECG patterns has become a research hotspot. In order to improve the accuracy in detecting abnormal ECG patterns, this paper proposes a hybrid 1D Resnet-GRU consisting of the Resnet and gated recurrent unit (GRU) modules to implement classification of arrhythmias from 12-lead ECG recordings. In addition, the focal Loss function is used to solve the problem of unbalanced datasets. Based on the proposed 1D Resnet-GRU model, the Grad-CAM++ mechanism, one of class-discriminative visualization methods, has been employed to the trained network model and generate thermal images superimposed on raw signals, so as to improve interpretability and transparency for arrhythmia classification. Experimental results show the proposed Resnet-GRU method can improve the F_1 score and accuracy effectively, and Grad-CAM++ can provide interpretability and clinical information for arrhythmia classification.© 2022 Institute of Physics and Engineering in Medicine.