Explainable artificial intelligence for heart rate variability in ECG signal

• Main Authors: Sanjana K., Sowmya V., Gopalakrishnan E.A., Soman K.P.
• Format: Article
• Language: English
• Published: Wiley 2020-12-01
• Series: Healthcare Technology Letters
• Subjects: signal classification; learning (artificial intelligence); cardiovascular system; diseases; electrocardiography; medical signal processing; convolutional neural nets; tachycardia disease; deep learning model; atrial fibrillation; ventricular fibrillation; sinus tachycardia; deep learning models; cu-ventricular tachycardia data; cardiac diseases; deep learning architectures; ecg signal; electrocardiogram signal; mit-bih malignant ventricular ectopy database; rcnn model
• Online Access: https://digital-library.theiet.org/content/journals/10.1049/htl.2020.0033

Electrocardiogram (ECG) signal is one of the most reliable methods to analyse the cardiovascular system. In the literature, there are different deep learning architectures proposed to detect various types of tachycardia diseases, such as atrial fibrillation, ventricular fibrillation, and sinus tachycardia. Even though all types of tachycardia diseases have fast beat rhythm as the common characteristic feature, existing deep learning architectures are trained with the corresponding disease-specific features. Most of the proposed works lack the interpretation and understanding of the results obtained. Hence, the objective of this letter is to explore the features learned by the deep learning models. For the detection of the different types of tachycardia diseases, the authors used a transfer learning approach. In this method, the model is trained with one of the tachycardia diseases called atrial fibrillation and tested with other tachycardia diseases, such as ventricular fibrillation and sinus tachycardia. The analysis was done using different deep learning models, such as RNN, LSTM, GRU, CNN, and RSCNN. RNN achieved an accuracy of 96.47% for atrial fibrillation data set, 90.88% accuracy for CU-ventricular tachycardia data set, and also achieved an accuracy of 94.71, and 94.18% for MIT-BIH malignant ventricular ectopy database for ECG lead I and lead II, respectively. The RNN model could only achieve an accuracy of 23.73% for the sinus tachycardia data set. A similar trend is shown by other models. From the analysis, it was evident that even though tachycardia diseases have fast beat rhythm as their common feature, the model was not able to detect different types of tachycardia diseases. The deep learning model could only detect atrial fibrillation and ventricular fibrillation and failed in the case of sinus tachycardia. From the analysis, they were able to interpret that, along with the fast beat rhythm, the model has learned the absence of P-wave which is a common feature for ventricular fibrillation and atrial fibrillation but sinus tachycardia disease has an upright positive P-wave. The time-based analysis is conducted to find the time complexity of the models. The analysis conveyed that RNN and RSCNN models could achieve better performance with lesser time complexity.