arrow
Return

Variational Visible Layers: A Practical Framework for Uncertainty Estimation

delete2026-01-01
delete0
PRE
AI
A
Abboud, Zeinab *
H
Hervé Lombaert
K
Kadoury, Samuel
DOI:10.1007/978-3-032-05185-1_64delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Uncertainty estimation is critical for reliable decision-making in medical imaging. State-of-the-art uncertainty methods require significant computational overhead and complex modeling. In this work, we present and explore a simple, effective approach to incorporating Bayesian uncertainty into deterministic networks by replacing the first and/or last layer (visible layers) with their variational Bayesian counterpart. This lightweight modification enables uncertainty quantification through mean-field variational estimation, making it practical for realworld medical applications. We evaluate the methods on ISIC and LIDCIDRI for the segmentation task and DermaMNIST and ChestMNIST for the classification task using post-hoc and jointly-trained visible layers. We demonstrate that variational visible layers enable uncertainty-based failure detection for both in-distribution and near-out-of-distribution samples, preserving task performance while reducing the number of variational parameters required for Bayesian estimation. We provide an easyto-implement solution for integrating uncertainty estimation into existing pipelines.
Keywords:
Uncertainty
Mean-Field Variational Inference
Bayesian
Neural Networks
Classification
Segmentation

Journal

M
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2025, PT XIV
IF:
0
Papers:
59
Citations:
0

Organization

U
universite de montreal
Scholars:
4.6W
Papers: 3.8W
Citations: 46
Cited Papers

Cited Papers

U-Net: Convolutional Networks for Biomedical Image Segmentation
err2015-11-18
err0
PREAI
errOlaf Ronneberger; Philipp Fischer; Thomas Brox
errShare
errSave
A review of uncertainty estimation and its application in medical imaging
err2023-06-01
err0
errOAAI
errKe Zou; Zhihao Chen; Xuedong Yuan; Xiaojing Shen; Meng Wang; Huazhu Fu
errShare
errSave
The Cancer Imaging Archive (TCIA): Maintaining and Operating a Public Information Repository
err2013-07-25
err0
errOAAI
errKenneth Clark; Bruce Vendt; Kirk Smith; John Freymann; Justin Kirby; Paul Koppel; Stephen Moore; Stanley Phillips; David Maffitt; Michael Pringle; Lawrence Tarbox; Fred Prior
errShare
errSave
A review of uncertainty quantification in deep learning: Techniques, applications and challenges
err2021-12-01
err1.2K
errOAAI
errAbdar, Moloud; Pourpanah, Farhad; Hussain, Sadiq; Rezazadegan, Dana; Liu, Li; Ghavamzadeh, Mohammad; Fieguth, Paul; Cao, Xiaochun; Khosravi, Abbas; Acharya, U. Rajendra; Makarenkov, Vladimir; Nahavandi, Saeid
errShare
errSave
err
IF0
err
err0
PREAI
err
errShare
errSave
MedMNIST v2 - A large-scale lightweight benchmark for 2D and 3D biomedical image classification
err2023-01-19
err0
errOAAI
errJiancheng Yang; Rui Shi; Donglai Wei; Zequan Liu; Lin Zhao; Bilian Ke; Hanspeter Pfister; Bingbing Ni
errShare
errSave
researcher View more