arXiv:2507.18106cs.CVcs.AI2025-07被引 2

用自由能建模分布不确定性,提升异常样本检测精度

Distributional Uncertainty for Out-of-Distribution Detection

  • 基于贝塔分布的自由能密度估计,精细捕捉未知区域不确定性
  • 无需随机采样,直接从参数学习不确定性,计算更高效
  • 在鱼眼场景等真实数据集上表现优异,适合需要可靠置信度的视觉任务

深度神经网络的不确定性估计是检测分布外(OoD)样本的常用方法,通常分布外样本表现出高预测不确定性。然而,传统方法如蒙特卡洛丢弃法仅关注模型或数据不确定性,难以契合OoD检测的语义目标。为此,我们提出自由能后验网络(Free-Energy Posterior Network),联合建模分布不确定性,并利用自由能识别OoD与误分类区域。该方法包含两项关键贡献:(1)采用贝塔分布参数化的自由能密度估计器,实现对模糊或未见区域的细粒度不确定性估计;(2)在后验网络中集成损失函数,可直接从学习参数中估计不确定性,无需随机采样。结合残差预测分支(RPL)框架,该方法超越了后处理能量阈值,使网络能通过贝塔分布方差学习OoD区域,提供语义合理且计算高效的不确定性感知分割方案。我们在Fishyscapes、RoadAnomaly和Segment-Me-If-You-Can等真实世界基准上验证了其有效性。

原文摘要 · Abstract (English)

Estimating uncertainty from deep neural networks is a widely used approach for detecting out-of-distribution (OoD) samples, which typically exhibit high predictive uncertainty. However, conventional methods such as Monte Carlo (MC) Dropout often focus solely on either model or data uncertainty, failing to align with the semantic objective of OoD detection. To address this, we propose the Free-Energy Posterior Network, a novel framework that jointly models distributional uncertainty and identifying OoD and misclassified regions using free energy. Our method introduces two key contributions: (1) a free-energy-based density estimator parameterized by a Beta distribution, which enables fine-grained uncertainty estimation near ambiguous or unseen regions; and (2) a loss integrated within a posterior network, allowing direct uncertainty estimation from learned parameters without requiring stochastic sampling. By integrating our approach with the residual prediction branch (RPL) framework, the proposed method goes beyond post-hoc energy thresholding and enables the network to learn OoD regions by leveraging the variance of the Beta distribution, resulting in a semantically meaningful and computationally efficient solution for uncertainty-aware segmentation. We validate the effectiveness of our method on challenging real-world benchmarks, including Fishyscapes, RoadAnomaly, and Segment-Me-If-You-Can.

不确定性估计异常检测分割自由能

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。