Abstract
Deep learning-based electrocardiogram (ECG) classification has shownimpressive performance but clinical adoption has been slowed by the lack oftransparent and faithful explanations. Post hoc methods such as saliency mapsmay fail to reflect a model's true decision process. Prototype-based reasoningoffers a more transparent alternative by grounding decisions in similarity tolearned representations of real ECG segments, enabling faithful, case-basedexplanations. We introduce ProtoECGNet, a prototype-based deep learning modelfor interpretable, multi-label ECG classification. ProtoECGNet employs astructured, multi-branch architecture that reflects clinical interpretationworkflows: it integrates a 1D CNN with global prototypes for rhythmclassification, a 2D CNN with time-localized prototypes for morphology-basedreasoning, and a 2D CNN with global prototypes for diffuse abnormalities. Eachbranch is trained with a prototype loss designed for multi-label learning,combining clustering, separation, diversity, and a novel contrastive loss thatencourages appropriate separation between prototypes of unrelated classes whileallowing clustering for frequently co-occurring diagnoses. We evaluateProtoECGNet on all 71 diagnostic labels from the PTB-XL dataset, demonstratingcompetitive performance relative to state-of-the-art black-box models whileproviding structured, case-based explanations. To assess prototype quality, weconduct a structured clinician review of the final model's projectedprototypes, finding that they are rated as representative and clear.ProtoECGNet shows that prototype learning can be effectively scaled to complex,multi-label time-series classification, offering a practical path towardtransparent and trustworthy deep learning models for clinical decision support.