SIM-ECG: A Signal Importance Mask-driven ECGClassification System
Not on the map. Why?
Abstract (arXiv)
Heart disease is the number one killer, and ECGs can assist in the early diagnosis and prevention of deadly outcomes. Accurate ECG interpretation is critical in detecting heart diseases; however, they are often misinterpreted due to a lack of training or insufficient time spent to detect minute anomalies. Subsequently, researchers turned to machine learning to assist in the analysis. However, existing systems are not as accurate as skilled ECG readers, and black-box approaches to providing diagnosis result in a lack of trust by medical personnel in a given diagnosis. To address these issues, we propose a signal importance mask feedback-based machine learning system that continuously accepts feedback, improves accuracy, and ex-plains the resulting diagnosis. This allows medical personnel to quickly glance at the output and either accept the results, validate the explanation and diagnosis, or quickly correct areas of misinterpretation, giving feedback to the system for improvement. We have tested our system on a publicly available dataset consisting of healthy and disease-indicating samples. We empirically show that our algorithm is better in terms of standard performance measures such as F-score and MacroAUC compared to normal training baseline (without feedback); we also show that our model generates better interpretability maps.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 17 of its 36 references that are in the map. First posted on arXiv 28 Oct 2021.
Builds on
17 of 36 references.
References · 17
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2013Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Mapscited by 2,348
- 2017Right for the Right Reasons: Training Differentiable Models by Constraining their Explanationscited by 293
- 2019Automatic diagnosis of the 12-lead ECG using a deep neural networkcited by 164
- 2020Deep Learning for ECG Analysis: Benchmarks and Insights from PTB-XLcited by 158
- 2019Interpretations are useful: penalizing explanations to align neural networks with prior knowledgecited by 144
- 2014Part-based R-CNNs for Fine-grained Category Detectioncited by 107
- 2018ECG Heartbeat Classification: A Deep Transferable Representationcited by 76
- 2016Rationale-Augmented Convolutional Neural Networks for Text Classificationcited by 39
- 2019Embedding Human Knowledge into Deep Neural Network via Attention Mapcited by 33
- 2020HAN-ECG: An Interpretable Atrial Fibrillation Detection Model Using Hierarchical Attention Networkscited by 14
- 2020Interpretable Deep Learning for Automatic Diagnosis of 12-lead Electrocardiogramcited by 11
5 more
- 2018Classification of 12-Lead ECG Signals with Bi-directional LSTM Networkcited by 8
- 2021Improving the trustworthiness of image classification models by utilizing bounding-box annotationscited by 4
- 2017Detection of Inferior Myocardial Infarction using Shallow Convolutional Neural Networkscited by 3
- 2013Computer Aided ECG Analysis - State of the Art and Upcoming Challengescited by 1
- 2018Using Multitask Learning to Improve 12-Lead Electrocardiogram Classificationcited by 1
Built on by
Not cited yet.