Search papers, labs, and topics across Lattice.
The paper introduces Risk-Calibrated Learning, a training technique that minimizes semantically incoherent, high-confidence errors in medical image classification by incorporating a confusion-aware clinical severity matrix into the loss function. This method explicitly distinguishes between visually ambiguous errors and catastrophic structural errors, allowing for targeted suppression of critical false negatives. Experiments across four medical imaging modalities demonstrate a significant reduction in Critical Error Rate (CER) compared to baselines, with relative safety improvements ranging from 20% to 92.4%.
Medical AI can be made dramatically safer鈥攔educing critical errors by up to 92%鈥攕imply by weighting the loss function with a clinically-informed confusion matrix.
Deep learning models often achieve expert-level accuracy in medical image classification but suffer from a critical flaw: semantic incoherence. These high-confidence mistakes that are semantically incoherent (e.g., classifying a malignant tumor as benign) fundamentally differ from acceptable errors which stem from visual ambiguity. Unlike safe, fine-grained disagreements, these fatal failures erode clinical trust. To address this, we propose Risk-Calibrated Learning, a technique that explicitly distinguishes between visual ambiguity (fine-grained errors) and catastrophic structural errors. By embedding a confusion-aware clinical severity matrix M into the optimization landscape, our method suppresses critical errors (false negatives) without requiring complex architectural changes. We validate our approach in four different imaging modalities: Brain Tumor MRI, ISIC 2018 (Dermoscopy), BreaKHis (Breast Histopathology), and SICAPv2 (Prostate Histopathology). Extensive experiments demonstrate that our Risk-Calibrated Loss consistently reduces the Critical Error Rate (CER) for all four datasets, achieving relative safety improvements ranging from 20.0% (on breast histopathology) to 92.4% (on prostate histopathology) compared to state-of-the-art baselines such as Focal Loss. These results confirm that our method offers a superior safety-accuracy trade-off across both CNN and Transformer architectures.
