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 uses prototype-based inference, which bases decisions on similarity to learned representations of real ECG segments, to provide transparent and reliable case-based explanations. It employs a structured multi-branch architecture that mirrors clinical interpretation workflows, integrating a 1D CNN with global prototypes for rhythm classification, a 2D CNN with temporally localized prototypes for morphology-based inference, and a 2D CNN with global prototypes for diffusion anomalies. Each branch is trained with a prototype loss designed for multi-label learning and combines a novel contrastive loss that promotes clustering, separation, diversity, and proper separation between prototypes of unrelated classes. We evaluate ProtoECGNet on all 71 diagnostic labels of the PTB-XL dataset, demonstrating that it performs competitively against state-of-the-art black-box models while providing structured case-based explanations. The representativeness and clarity of the prototypes were verified through structured clinician reviews. 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 demonstrate the feasibility of achieving both interpretability and performance in electrocardiogram classification using a prototype-based deep learning model.
Presenting an effective prototype learning method for multi-label classification problems.
Validate the reliability of the prototype through clinical review.
A practical approach to developing an electrocardiogram diagnostic support system.
Limitations:
Only performance validation was performed on the PTB-XL dataset, so further research on generalization performance is needed.
The interpretability of a prototype relies on the subjective assessment of clinicians. Therefore, the development of objective evaluation criteria is necessary.
The model's complexity may be high, resulting in high computational costs.
Further validation is needed in diverse ECG datasets and clinical settings.
👍