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

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 un...

Full description

Saved in:
Bibliographic Details
Main Authors Ozdemir, Onur, Woodward, Benjamin, Berlin, Andrew A
Format Journal Article
LanguageEnglish
Published 01.12.2017
Subjects
Computer Science - Computer Vision and Pattern Recognition
Statistics - Machine Learning
Online AccessGet full text

Cover

More Information
Summary: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.
DOI:10.48550/arxiv.1712.00497