Propagating Uncertainty in Multi-Stage Bayesian Convolutional Neural Networks with Application to Pulmonary Nodule Detection

2017/12/01 by Onur Özdemir, Ozdemir, Onur, Benjamin Woodward +3 · 2 citations
Medicine · #COVID-19 diagnosis using AI #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #Lung Cancer Diagnosis and Treatment #Machine Learning (stat.ML) #Phonocardiography and Auscultation Techniques

paper · pdf · doi:10.48550/arxiv.1712.00497

openalex publication_date 2017/12/01 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Motivated by the problem of computer-aided detection (CAD) of pulmonary nodules, we introduce methods to propagate and fuse uncertainty information in a multi-stage Bayesian convolutional neural network (CNN) architecture. The question we seek to answer is "can we take advantage of the model uncertainty provided by one deep learning model to improve the performance of the subsequent deep learning models and ultimately of the overall performance in a multi-stage Bayesian deep learning architecture?". Our experiments show that propagating uncertainty through the pipeline enables us to improve the overall performance in terms of both final prediction accuracy and model confidence.

Citations

• Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning

Cited by

• Decomposition-Based Modular Conformal Prediction for Two-Stage Modeling
• Trustworthy clinical AI solutions: A unified review of uncertainty quantification in Deep Learning models for medical image analysis

Related

• Modeling Uncertainty by Learning a Hierarchy of Deep Neural Connections
• Exploring Instance-Level Uncertainty for Medical Detection
• A 3D Probabilistic Deep Learning System for Detection and Diagnosis of\n Lung Cancer Using Low-Dose CT Scans
• On Calibrated Model Uncertainty in Deep Learning
• Probabilistic Neighbourhood Component Analysis: Sample Efficient Uncertainty Estimation in Deep Learning