Towards Improved Intermediate Layer Variational Inference for Uncertainty Estimation
Abstract
DNNs have been excelling on many computer vision tasks, achieving many milestones and are continuing to prove their validity. It is important for DNNs to express their uncertainties to be taken into account for the whole application output, whilst maintaining their performance. The intermediate layer variational inference has been a promising approach to estimate uncertainty in real-time beating state-of-the-art approaches. In this work, we propose an enhancement of the intermediate layer variational inference by adding Dirichlet distribution to the network architecture. This improves, on one hand, the uncertainty estimation reliability and on the other hand can detect out-of-distribution samples. Results show that with the addition of Dirichlet distributions the DNN is able to maintain its segmentation performance whilst boosting its uncertainty estimation capabilities.
Cite
Text
Hammam et al. "Towards Improved Intermediate Layer Variational Inference for Uncertainty Estimation." European Conference on Computer Vision Workshops, 2022. doi:10.1007/978-3-031-25072-9_36Markdown
[Hammam et al. "Towards Improved Intermediate Layer Variational Inference for Uncertainty Estimation." European Conference on Computer Vision Workshops, 2022.](https://mlanthology.org/eccvw/2022/hammam2022eccvw-improved/) doi:10.1007/978-3-031-25072-9_36BibTeX
@inproceedings{hammam2022eccvw-improved,
title = {{Towards Improved Intermediate Layer Variational Inference for Uncertainty Estimation}},
author = {Hammam, Ahmed and Bonarens, Frank and Ghobadi, Seyed Eghbal and Stiller, Christoph},
booktitle = {European Conference on Computer Vision Workshops},
year = {2022},
pages = {526-542},
doi = {10.1007/978-3-031-25072-9_36},
url = {https://mlanthology.org/eccvw/2022/hammam2022eccvw-improved/}
}