# Bayesian QuickNAT: Model Uncertainty in Deep Whole-Brain Segmentation for Structure-wise Quality Control

arXiv: https://arxiv.org/abs/1811.09800 · HTML: https://mlsanity.com/paper/1811.09800/

## Key facts

- Cited by 39 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 24 Nov 2018.

## Built on by (most cited first, 30 of 39)

- [A Survey of Uncertainty in Deep Neural Networks](https://mlsanity.com/paper/2107.03342/) (2021)
- [Trustworthy clinical AI solutions: a unified review of uncertainty quantification in deep learning models for medical image analysis](https://mlsanity.com/paper/2210.03736/) (2022)
- [A Survey on Epistemic (Model) Uncertainty in Supervised Learning: Recent Advances and Applications](https://mlsanity.com/paper/2111.01968/) (2021)
- [Evidential Uncertainty Quantification: A Variance-Based Perspective](https://mlsanity.com/paper/2311.11367/) (2023)
- [Label fusion and training methods for reliable representation of inter-rater uncertainty](https://mlsanity.com/paper/2202.07550/) (2022)
- [QUBIQ: Uncertainty Quantification for Biomedical Image Segmentation Challenge](https://mlsanity.com/paper/2405.18435/) (2024)
- [Comparative Benchmarking of Failure Detection Methods in Medical Image Segmentation: Unveiling the Role of Confidence Aggregation](https://mlsanity.com/paper/2406.03323/) (2024)
- [Beyond Unimodal: Generalising Neural Processes for Multimodal Uncertainty Estimation](https://mlsanity.com/paper/2304.01518/) (2023)
- [Uncertainty Estimation for Multi-view Data: The Power of Seeing the Whole Picture](https://mlsanity.com/paper/2210.02676/) (2022)
- [GaIA: Graphical Information Gain based Attention Network for Weakly Supervised Point Cloud Semantic Segmentation](https://mlsanity.com/paper/2210.01558/) (2022)
- [Structural-Based Uncertainty in Deep Learning Across Anatomical Scales: Analysis in White Matter Lesion Segmentation](https://mlsanity.com/paper/2311.08931/) (2023)
- [Beyond Voxel Prediction Uncertainty: Identifying brain lesions you can trust](https://mlsanity.com/paper/2209.10877/) (2022)
- [A Review of Bayesian Uncertainty Quantification in Deep Probabilistic Image Segmentation](https://mlsanity.com/paper/2411.16370/) (2024)
- [Uncertainty-Aware Vision-Language Segmentation for Medical Imaging](https://mlsanity.com/paper/2602.14498/) (2026)
- [Uncertainty-Aware Body Composition Analysis with Deep Regression Ensembles on UK Biobank MRI](https://mlsanity.com/paper/2101.06963/) (2021)
- [SUPER-Net: Trustworthy Image Segmentation via Uncertainty Propagation in Encoder-Decoder Networks](https://mlsanity.com/paper/2111.05978/) (2021)
- [A Decoupled Uncertainty Model for MRI Segmentation Quality Estimation](https://mlsanity.com/paper/2109.02413/) (2021)
- [Towards Reliable Medical Image Segmentation by Modeling Evidential Calibrated Uncertainty](https://mlsanity.com/paper/2301.00349/) (2023)
- [Deep and Statistical Learning in Biomedical Imaging: State of the Art in 3D MRI Brain Tumor Segmentation](https://mlsanity.com/paper/2103.05529/) (2021)
- [Automated Olfactory Bulb Segmentation on High Resolutional T2-Weighted MRI](https://mlsanity.com/paper/2108.04267/) (2021)
- [A Critical Synthesis of Uncertainty Quantification and Foundation Models in Monocular Depth Estimation](https://mlsanity.com/paper/2501.08188/) (2025)
- [Uncertainty-Aware Segmentation Quality Prediction via Deep Learning Bayesian Modeling: Comprehensive Evaluation and Interpretation on Skin Cancer and Liver Segmentation](https://mlsanity.com/paper/2508.01460/) (2025)
- [SIAM: Head and Brain MRI Segmentation from Few High-Quality Templates via Synthetic Training](https://mlsanity.com/paper/2605.02737/) (2026)
- [Hardware-aware Real-time Myocardial Segmentation Quality Control in Contrast Echocardiography](https://mlsanity.com/paper/2109.06909/) (2021)
- [Automatic quality control framework for more reliable integration of machine learning-based image segmentation into medical workflows](https://mlsanity.com/paper/2112.03277/) (2021)
- [Exploring Structure-Wise Uncertainty for 3D Medical Image Segmentation](https://mlsanity.com/paper/2211.00303/) (2022)
- [SACDNet: Towards Early Type 2 Diabetes Prediction with Uncertainty for Electronic Health Records](https://mlsanity.com/paper/2301.04844/) (2023)
- [Uncertainty Quantification for Image-based Traffic Prediction across Cities](https://mlsanity.com/paper/2308.06129/) (2023)
- [Automatic driving lane change safety prediction model based on LSTM](https://mlsanity.com/paper/2403.06993/) (2024)
- [Uncertainty in Extreme Multi-label Classification](https://mlsanity.com/paper/2210.10160/) (2022)

## Abstract (arXiv)

We introduce Bayesian QuickNAT for the automated quality control of whole-brain segmentation on MRI T1 scans. Next to the Bayesian fully convolutional neural network, we also present inherent measures of segmentation uncertainty that allow for quality control per brain structure. For estimating model uncertainty, we follow a Bayesian approach, wherein, Monte Carlo (MC) samples from the posterior distribution are generated by keeping the dropout layers active at test time. Entropy over the MC samples provides a voxel-wise model uncertainty map, whereas expectation over the MC predictions provides the final segmentation. Next to voxel-wise uncertainty, we introduce four metrics to quantify structure-wise uncertainty in segmentation for quality control. We report experiments on four out-of-sample datasets comprising of diverse age range, pathology and imaging artifacts. The proposed structure-wise uncertainty metrics are highly correlated with the Dice score estimated with manual annotation and therefore present an inherent measure of segmentation quality. In particular, the intersection over union over all the MC samples is a suitable proxy for the Dice score. In addition to quality control at scan-level, we propose to incorporate the structure-wise uncertainty as a measure of confidence to do reliable group analysis on large data repositories. We envisage that the introduced uncertainty metrics would help assess the fidelity of automated deep learning based segmentation methods for large-scale population studies, as they enable automated quality control and group analyses in processing large data repositories.
