arXiv:2605.01502cs.CV2026-05中稿 · IEEE ICIP 2026

用解码器层间信息量估算分割不确定性,单次前向无需复杂计算。

RADMI: Latent Information Aggregation as a Proxy for Model Uncertainty

论文配图:RADMI: Latent Information Aggregation as a Proxy for Model Uncertainty
图 1 · 摘自论文原文
  • 通过测量解码器相邻层间互信息,实现单次前向的不确定性估计。
  • 在地震岩相分割任务中,皮尔逊相关性领先第二名5.5%,斯皮尔曼相关性领先10.7%。
  • 生成清晰边界定位的不确定性图,适合高精度密集预测场景。

认知不确定性估计对于识别深度学习系统输出不可靠的区域至关重要。然而,现有方法需依赖计算成本高昂的集成方法或多次随机前向传播,限制了其在分割等密集预测任务中的可扩展性。我们提出分辨率聚合解码器互信息(RADMI),一种单次前向传播方法,通过测量分割网络中连续解码器层间的互信息(MI)来估计预测不确定性。我们发现,层间互信息升高与预测不确定性相关,因为网络在模糊区域(如类别边界)需整合冲突的上下文信息。在地震岩相分割基准测试中,RADMI在所有单次前向方法中与深度集成不确定性相关性最高,皮尔逊相关系数领先第二名5.5%,斯皮尔曼相关系数领先10.7%。相比缺乏空间精度或计算开销大的基线方法,RADMI无需架构修改即可生成锐利、边界局部化的不确定性图。结果表明,归一化信息流的线性聚合为编码器-解码器架构提供了合理且高效的预测不确定性代理。

原文摘要 · Abstract (English)

Epistemic uncertainty estimation is essential for identifying regions where deep learning system outputs may be unreliable. However, existing approaches require computationally expensive ensemble methods or multiple stochastic forward passes, limiting their scalability to dense prediction tasks like segmentation. We propose Resolution-Aggregated Decoder Mutual Information (RADMI), a single-pass method that estimates prediction uncertainty by measuring mutual information (MI) between consecutive decoder layers in segmentation networks. We observe that elevated inter-layer MI correlates with prediction uncertainty, as the network must integrate conflicting contextual information at ambiguous regions such as class boundaries. Evaluating on a seismic facies segmentation benchmark, RADMI achieves the highest correlation with deep ensemble uncertainty among all single-pass methods, outperforming the next-best baselines by 5.5% in Pearson and 10.7% in Spearman correlation coefficients. Compared to baselines that either lack spatial precision or demand significant computational overhead, RADMI yields sharp, boundary-localized uncertainty maps without architectural modifications. Our results suggest that linear aggregation of normalized information flow provides a principled and efficient proxy for prediction uncertainty in encoder-decoder architectures.

不确定性估计分割任务互信息单次前向

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