Daily Arxiv

This is a page that curates AI-related papers published worldwide.
All content here is summarized using Google Gemini and operated on a non-profit basis.
Copyright for each paper belongs to the authors and their institutions; please make sure to credit the source when sharing.

ProtoECGNet: Case-Based Interpretable Deep Learning for Multi-Label ECG Classification with Contrastive Learning

Created by
  • Haebom

Author

Sahil Sethi, David Chen, Thomas Statchen, Michael C. Burkhart, Nipun Bhandari, Bashar Ramadan, Brett Beaulieu-Jones

Outline

This paper presents ProtoECGNet, an interpretable multi-label deep learning model for electrocardiogram (ECG) classification. ProtoECGNet employs a structured multi-branch architecture that reflects clinical interpretation workflows. It performs rhythm classification using a 1D CNN and global prototypes, morphology-based inference using a 2D CNN and time-localized prototypes, and diffusive abnormality classification using a 2D CNN and global prototypes. Each branch is trained with a prototype loss function designed for multi-label learning and incorporates a novel contrastive loss that promotes clustering, separability, diversity, and proper separation between prototypes of unrelated classes. We evaluate ProtoECGNet on 71 diagnostic labels from the PTB-XL dataset, demonstrating competitive performance against state-of-the-art black-box models and providing structured, case-based explanations. Clinician review of the prototypes confirmed their representativeness and clarity. ProtoECGNet demonstrates that prototype learning can effectively scale to complex multi-label time series classification, providing a practical path toward transparent and reliable deep learning models for clinical decision support.

Takeaways, Limitations

Takeaways:
We present an interpretable and reliable deep learning model for electrocardiogram classification.
We demonstrate that prototype-based learning can be effectively applied to complex multi-label time series classification.
Provides structured, case-based explanations that are easy for clinicians to understand.
Achieve competitive performance with cutting-edge black box models.
Limitations:
In this paper, we evaluated only the PTB-XL dataset, so further research is needed to evaluate generalization performance on other datasets.
The quality assessment of prototypes relies on the subjective assessment of clinicians. More objective assessment methods may be needed.
Due to the complexity of the model, it can be computationally expensive for real-time applications.
👍