Search papers, labs, and topics across Lattice.
This paper introduces ERTS, an explainability-based reliability training signal designed to enhance the efficiency of ECG classification by filtering out unreliable samples during training. By leveraging Grad-CAM attention maps, ERTS prioritizes samples with coherent and localized patterns while excluding those with low focus scores, thereby optimizing gradient updates. The method demonstrates consistent improvements in macro-F1 scores across three ECG datasets and various architectures, highlighting the potential of explanation quality to enhance both model reliability and training efficiency.
Explanation quality can be harnessed to filter out noise in ECG classification, leading to significant efficiency gains in model training.
Training deep neural networks for clinical time-series analysis is computationally demanding, yet many healthcare settings lack the resources required for repeated model development and deployment. This challenge is particularly evident in electrocardiogram classification, where large datasets and long training schedules make efficiency practically important. Progressive Data Dropout reduces training cost by excluding samples from gradient updates once they are learned, but it relies on model confidence and may retain samples that are difficult due to noise or ambiguity rather than useful signal. In this work, we introduce ERTS, an explainability-based reliability training signal for efficient ECG classification. ERTS uses explanation quality during training to distinguish between informative and unreliable uncertainty. Building on progressive data selection, we compute Grad-CAM attention maps for candidate samples and derive a focus score that measures whether model predictions are supported by coherent and localised patterns. Samples with low focus are filtered out, while those with meaningful attention are prioritised for gradient updates. We evaluate ERTS across three ECG datasets and multiple backbone architectures, showing consistent improvements in macro-F1 alongside reduced effective training cost. These results suggest that explanation quality can serve as a practical signal for improving both efficiency and reliability in clinical time-series learning. Code will be released.
