Rare-Class Collapse in ECG-Based Ventricular Tachycardia and Fibrillation Detection: A Systematic Benchmark of Class-Imbalance Mitigation from Reweighting to Cascade Classification
Tiruwa, K. R.
Show abstract
Ventricular tachycardia (VT) and ventricular fibrillation (VF) are the leading electrical causes of sudden cardiac death, but automated detection is limited by strong class imbalance, where lethal arrhythmias account for fewer than 22% of ECG segments. In this setting, standard classifiers can achieve high accuracy by predicting normal rhythm in most cases while missing many lethal events, a failure mode referred to as rare-class collapse. We evaluated six imbalance-handling approaches: naive logistic regression, inverse-frequency reweighting, label-distribution-aware margin loss (LDAM), cost-sensitive training, two-stage cascade classification, and anomaly detection on 15,614 ECG segments from three PhysioNet databases (VTaC, VFDB, CUDB), with an overall normal-to-lethal ratio of 3.6:1. All methods were assessed at a fixed operating point of 95% specificity using recall, area under the precision-recall curve (AUPRC), and missed-lethal-event rate (MLER). The naive model achieved 45.1% recall (MLER = 0.549), missing 564 of 1,027 lethal events despite 84.1% accuracy. The two-stage cascade performed best, with 65.2% recall, AUPRC of 0.821, and MLER of 0.348, reducing missed events by 37% and achieving the highest decision-curve net benefit. Per-source analysis showed near-complete VF detection (recall up to 0.975) but much lower VT detection (recall 0.183), suggesting a feature-space limitation due to spectral similarity between organized VT and rapid sinus rhythm. Overall, the results show that evaluation metrics strongly influence the visibility of rare-class failure, and that cascade-based methods outperform simpler reweighting approaches for detecting lethal arrhythmias.
Matching journals
The top 7 journals account for 50% of the predicted probability mass.