Uncertainty-driven training for three-dimensional calibrated lung nodule classification

Abstract

In this work, we present an uncertainty-driven training framework for three-dimensional computed tomography (CT) lung nodule classification, where validation-based uncertainty estimates guide loss reweighting to enhance predictive performance and probability calibration. Two Uncertainty Quantification (UQ) methods are considered: Monte Carlo Dropout (MCD) and Evidential Deep Learning (EDL). Both provide per-class uncertainty estimates that modulate the loss and encourage focus on hard or unreliable classes. The framework is evaluated with ResNet, DenseNet, EfficientNet, Vision Transformer (ViT), and Swin Transformer backbones on two datasets: the clinical LIDC-IDRI cohort and the NoduleMNIST3D benchmark. Uncertainty-driven training achieves classification performance similar to conventional training while substantially improving calibration, with an expected calibration error (ECE) reduced by up to 65% on LIDC-IDRI. EDL attains competitive performance on shallower architectures with single-pass inference, whereas MCD is more robust on deeper networks. Analysis across architectural families reveals that uncertainty-driven training benefits convolutional backbones more consistently than transformer-based architectures: EDL in particular degrades on ViT, suggesting that the Dirichlet evidence parameterisation may interact unfavourably with attention-based architectures at lower input resolutions. A posteriori temperature scaling proves highly effective across all configurations, indicating that a simple scalar calibration can be competitive even without explicit uncertainty-aware training. Our results indicate that integrating UQ into the training loop can significantly improve probabilistic calibration and support more trustworthy deployment of three-dimensional medical imaging models.

The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.

More in Image & Video Processing

• arXiv preprint · 17 September 2026

  Compression Hurts, Pooling Helps: Information Loss in Rayleigh-Scale Estimation from B-Mode Ultrasound

  D. Hudson Smith and Ahmer Raza
• arXiv preprint · 18 September 2026

  Adaptive Color Grading

  Trevor D. Canham, Abhijith Punnappurath and Michael S. Brown
• arXiv preprint · 17 September 2026

  The segmentation ceiling: why explicit left-ventricular masks do not improve learned ejection-fraction regression

  Farshid Farhadi Khouzani, Paul La Plante, Bryar Mustafa Shareef and 1 other
• arXiv preprint · 18 September 2026

  WS-NeRF: A Mamba-Driven World-State-Aware Adaptive Deblurring Neural Radiance Field

  Hang Jiang, Jinghao Wang, Yiming Zhang and 3 others
• arXiv preprint · 17 September 2026

  HyperAMS-Net: Adaptive Multi-Scale Spatial Hypergraph Network for Brain Disorder Classification

  Proloy Kumar Mondal, Md Kamran Hussin Chowdhury and Hoi Leong Lee

All new papers in Image & Video Processing