Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Risk-Calibrated Learning: Minimizing Fatal Errors in Medical AI

About

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.

Abolfazl Mohammadi-Seif, Ricardo Baeza-Yates• 2026

Related benchmarks

TaskDatasetResultRank
Medical Image ClassificationISIC 2018
Accuracy76.1
40
Medical Image ClassificationBreakHis
Acc56
21
Medical Image ClassificationSICAP Prostate Histo v2
CER0.81
4
Medical Image ClassificationBrain MRI Radiology
CER0.00e+0
4
Showing 4 of 4 rows

Other info

Follow for update