Interpretable ECG Analysis for Myocardial Infarction Detection through Counterfactuals

Abstract

In the evolving landscape of ECG signal analysis, the challenge of limited transparency in machine learning models remains a significant barrier to their effective integration into clinical practice. This study addresses this issue by investigating the use of counterfactual explanations to improve model interpretability for clinicians, particularly in differentiating healthy subjects from myocardial infarction patients. Utilizing the PTB-XL dataset, we developed a methodology for systematic feature extraction and refinement to prepare for counterfactual analysis. This led to the creation of the Visualizing Counterfactual Clues on Electrocardiograms (VCCE) method, designed to improve the practicality of counterfactual explanations in a clinical setting. The validity of our approach was assessed using custom metrics that reflect the diagnostic relevance of counterfactuals, evaluated with the help of two cardiologists. Our findings suggest that this approach could support future efforts in using ECGs to predict patient outcomes for cardiac conditions, achieving interpretation validity scores of 23.29 $\pm$ 1.04 and 20.28 $\pm$ 0.99 out of 25 for high and moderate-quality interpretations, respectively. Clinical alignment scores of 0.83 $\pm$ 0.12 for high-quality and 0.57 $\pm$ 0.10 for moderate-quality interpretations underscore the potential clinical applicability of our method. The methodology and findings of this study contribute to the ongoing discussion on enhancing the interpretability of machine learning models in cardiology, offering a concept that bridges the gap between advanced data analysis techniques and clinical decision-making. The source code for this study is available at https://github.com/tanyelai/vcce.