cs.CVApr 14, 2026

Risk-Calibrated Learning: Minimizing Fatal Errors in Medical AI

Authors: Abolfazl Mohammadi-Seif, Ricardo Baeza-Yates

Organizations: Department of Engineering, Universitat Pompeu Fabra, Barcelona, Spain

Abstract

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.

Figures & tables

Explore similar work

CardsList
  1. MedFormer-UR: Uncertainty-Routed Transformer for Medical Image Classification

    Apr 10, 2026Mohammed Maaz Sibhai, Abedalrhman Alkhateeb, Saad B. AhmedMedical Image ClassificationVision Transformer

  2. Risk-Aware Robust Learning: Reducing Clinical Risk under Label Noise in Medical Image Classification

    Apr 26, 2026Maycon R. S. Pereira, Filipe R. CordeiroCost-Sensitive LearningMedical Image Classification

  3. Confidence-Uncertainty Boundary Calibration for Bayesian Deep Learning in Medical Image Analysis

    Feb 12, 2026Hua Xu, Julián D. Arias-Londoño, Juan I. Godino-LlorenteBayesian Neural NetworksPost-Hoc Calibration