Development of Interpretable Machine Learning Models to Detect Arrhythmia based on ECG Data

  • 2022-05-05 18:29:33
  • Shourya Verma
  • 0

Abstract

The analysis of electrocardiogram (ECG) signals can be time consuming as itis performed manually by cardiologists. Therefore, automation through machinelearning (ML) classification is being increasingly proposed which would allowML models to learn the features of a heartbeat and detect abnormalities. Thelack of interpretability hinders the application of Deep Learning inhealthcare. Through interpretability of these models, we would understand how amachine learning algorithm makes its decisions and what patterns are beingfollowed for classification. This thesis builds Convolutional Neural Network(CNN) and Long Short-Term Memory (LSTM) classifiers based on state-of-the-artmodels and compares their performance and interpretability to shallowclassifiers. Here, both global and local interpretability methods are exploitedto understand the interaction between dependent and independent variablesacross the entire dataset and to examine model decisions in each sample,respectively. Partial Dependence Plots, Shapley Additive Explanations,Permutation Feature Importance, and Gradient Weighted Class Activation Maps(Grad-Cam) are the four interpretability techniques implemented on time-seriesML models classifying ECG rhythms. In particular, we exploit Grad-Cam, which isa local interpretability technique and examine whether its interpretabilityvaries between correctly and incorrectly classified ECG beats within eachclass. Furthermore, the classifiers are evaluated using K-Fold cross-validationand Leave Groups Out techniques, and we use non-parametric statistical testingto examine whether differences are significant. It was found that Grad-CAM wasthe most effective interpretability technique at explaining predictions ofproposed CNN and LSTM models. We concluded that all high performing classifierslooked at the QRS complex of the ECG rhythm when making predictions.

 

Quick Read (beta)

loading the full paper ...