Investigating ECG Diagnosis with Ambiguous Labels using Partial Label Learning
By: Sana Rahmani, Javad Hashemi, Ali Etemad
Label ambiguity is an inherent problem in real-world electrocardiogram (ECG) diagnosis, arising from overlapping conditions and diagnostic disagreement. However, current ECG models are trained under the assumption of clean and non-ambiguous annotations, which limits both the development and the meaningful evaluation of models under real-world conditions. Although Partial Label Learning (PLL) frameworks are designed to learn from ambiguous labels, their effectiveness in medical time-series domains, ECG in particular, remains largely unexplored. In this work, we present the first systematic study of PLL methods for ECG diagnosis. We adapt nine PLL algorithms to multi-label ECG diagnosis and evaluate them using a diverse set of clinically motivated ambiguity generation strategies, capturing both unstructured (e.g., random) and structured ambiguities (e.g., cardiologist-derived similarities, treatment relationships, and diagnostic taxonomies). Our experiments on the PTB-XL and Chapman datasets demonstrate that PLL methods vary substantially in their robustness to different types and degrees of ambiguity. Through extensive analysis, we identify key limitations of current PLL approaches in clinical settings and outline future directions for developing robust and clinically aligned ambiguity-aware learning frameworks for ECG diagnosis.
Similar Papers
Amortized Variational Inference for Partial-Label Learning: A Probabilistic Approach to Label Disambiguation
Machine Learning (CS)
Teaches computers to learn from messy, unsure information.
Transferring Clinical Knowledge into ECGs Representation
Machine Learning (CS)
Makes heart monitors understand sickness better.
Partial Label Clustering
Machine Learning (CS)
Helps computers group similar things better.