Search papers, labs, and topics across Lattice.
This paper presents a novel approach for integrating aleatoric and epistemic uncertainties in high-dimensional regression tasks, which is crucial for enhancing the reliability of deep learning predictions in complex applications like medical image segmentation. By employing a low-rank plus diagonal covariance structure, the method captures essential correlations in output spaces while mitigating the computational burden associated with full covariance matrices. The results demonstrate significant improvements in uncertainty quantification across various tasks, including image inpainting and depth estimation, leading to more robust and transparent predictions.
Jointly modeling aleatoric and epistemic uncertainties can dramatically enhance prediction reliability in high-dimensional outputs, outperforming traditional methods.
Uncertainty Quantification (UQ) plays a vital role in enhancing the reliability of deep learning model predictions, especially in scenarios with high-dimensional output spaces. This paper addresses the dual nature of uncertainty -- aleatoric and epistemic -- focusing on their joint integration in high-dimensional regression tasks. For example, in applications like medical image segmentation or restoration, aleatoric uncertainty captures inherent data noise, while epistemic uncertainty quantifies the model's confidence in unfamiliar conditions. Modeling both jointly enables more reliable predictions by reflecting both unavoidable variability and knowledge gaps, whereas modeling only one limits transparency and robustness. We propose a novel approach that approximates the resulting joint uncertainty using a low-rank plus diagonal covariance structure, capturing essential output correlations while avoiding the computational burdens of full covariance matrices. Unlike prior work, our method explicitly combines aleatoric and epistemic uncertainties into a unified second-order distribution that supports robust downstream analyses like sampling and log-likelihood evaluation. We further introduce stabilization strategies for efficient training and inference, achieving superior UQ in the tasks of image inpainting, colorization, optical flow, and depth estimation.
