Wang, Chen, Wang, Xiang, Zhang, Jiawei et al. (5 more authors) (2021) Uncertainty Estimation for Stereo Matching Based on Evidential Deep Learning. Pattern Recognition. 108498. ISSN 0031-3203
Abstract
Although deep learning-based stereo matching approaches have achieved excellent performance in recent years, it is still a non-trivial task to estimate the uncertainty of the produced disparity map. In this paper, we propose a novel approach to estimate both aleatoric and epistemic uncertainties for stereo matching in an end-to-end way. We introduce an evidential distribution, named Normal Inverse-Gamma (NIG) distribution, whose parameters can be used to calculate the uncertainty. Instead of directly regressed from aggregated features, the uncertainty parameters are predicted for each potential disparity and then averaged via the guidance of matching probability distribution. Furthermore, considering the sparsity of ground truth in real scene datasets, we design two additional losses. The first one tries to enlarge uncertainty on incorrect predictions, so uncertainty becomes more sensitive to erroneous regions. The second one enforces the smoothness of the uncertainty in the regions with smooth disparity. Most stereo matching models, such as PSM-Net, GA-Net, and AA-Net, can be easily integrated with our approach. Experiments on multiple benchmark datasets show that our method improves stereo matching results. We prove that both aleatoric and epistemic uncertainties are well-calibrated with incorrect predictions. Particularly, our method can capture increased epistemic uncertainty on out-of-distribution data, making it effective to prevent a system from potential fatal consequences. Code is available at https://github.com/Dawnstar8411/StereoMatching-Uncertainty
Metadata
Item Type: | Article |
---|---|
Authors/Creators: |
|
Keywords: | Stereo Matching,Uncertainty Estimation,Evidential Deep Learning |
Dates: |
|
Institution: | The University of York |
Academic Units: | The University of York > Faculty of Sciences (York) > Computer Science (York) |
Depositing User: | Pure (York) |
Date Deposited: | 09 Dec 2021 16:00 |
Last Modified: | 19 Dec 2024 00:13 |
Published Version: | https://doi.org/10.1016/j.patcog.2021.108498 |
Status: | Published online |
Refereed: | Yes |
Identification Number: | 10.1016/j.patcog.2021.108498 |
Related URLs: | |
Open Archives Initiative ID (OAI ID): | oai:eprints.whiterose.ac.uk:181424 |