Repository logo
 

Bayesian segnet: Model uncertainty in deep convolutional encoder-decoder architectures for scene understanding

Accepted version
Peer-reviewed

Type

Conference Object

Change log

Authors

Badrinarayanan, V 

Abstract

We present a deep learning framework for probabilistic pixel-wise semantic segmentation, which we term Bayesian SegNet. Semantic segmentation is an important tool for visual scene understanding and a meaningful measure of uncertainty is essential for decision making. Our contribution is a practical system which is able to predict pixel-wise class labels with a measure of model uncertainty. We achieve this by Monte Carlo sampling with dropout at test time to generate a posterior distribution of pixel class labels. In addition, we show that modelling uncertainty improves segmentation performance by 2-3% across a number of state of the art architectures such as SegNet, FCN and Dilation Network, with no additional parametrisation. We also observe a significant improvement in performance for smaller datasets where modelling uncertainty is more effective. We benchmark Bayesian SegNet on the indoor SUN Scene Understanding and outdoor CamVid driving scenes datasets.

Description

Keywords

cs.CV, cs.CV, cs.NE

Journal Title

British Machine Vision Conference 2017, BMVC 2017

Conference Name

British Machine Vision Conference 2017

Journal ISSN

Volume Title

Publisher

British Machine Vision Association

Rights

All rights reserved
Sponsorship
Toyota Corporation