arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.08142cs.LGstat.ML

方差门控集成:一种面向认知不确定性的估计框架

Variance-Gated Ensembles: An Epistemic-Aware Framework for Uncertainty Estimation

Faculty of Computer Science, Dalhousie University, Halifax, NS(计算机科学学院,达尔豪西大学,哈利法克斯,NS)

H. Martin Gillis, Isaac Xu, Thomas Trappenberg

AI总结:

提出方差门控集成(VGE)框架,通过从集成统计量计算信噪比门控注入认知敏感性,实现高效且可微的不确定性估计,在计算效率与性能上匹配或超越现有方法。

摘要:

机器学习应用需要快速且可靠的逐样本不确定性估计。常见方法是使用贝叶斯或近似方法的预测分布,并将不确定性加性分解为偶然(即数据相关)和认知(即模型相关)分量。然而,加性分解最近受到质疑,有证据表明当使用有限集成采样和/或不匹配的预测分布时,该分解会失效。本文介绍方差门控集成(VGE),一种直观、可微的框架,通过从集成统计量计算的信噪比门控注入认知敏感性。VGE提供:(i)方差门控边际不确定性(VGMU)分数,将决策边际与集成预测方差耦合;(ii)方差门控归一化(VGN)层,通过每类可学习的集成成员概率归一化,将方差门控不确定性机制推广到训练。我们推导出闭合形式的向量-雅可比积,使得通过集成样本均值和方差进行端到端训练成为可能。VGE在保持计算效率的同时,匹配或超越最先进的信息论基线。因此,VGE为集成模型中的认知感知不确定性估计提供了一种实用且可扩展的方法。

英文摘要:

Machine learning applications require fast and reliable per-sample uncertainty estimation. A common approach is to use predictive distributions from Bayesian or approximation methods and additively decompose uncertainty into aleatoric (i.e., data-related) and epistemic (i.e., model-related) components. However, additive decomposition has recently been questioned, with evidence that it breaks down when using finite-ensemble sampling and/or mismatched predictive distributions. This paper introduces Variance-Gated Ensembles (VGE), an intuitive, differentiable framework that injects epistemic sensitivity via a signal-to-noise gate computed from ensemble statistics. VGE provides: (i) a Variance-Gated Margin Uncertainty (VGMU) score that couples decision margins with ensemble predictive variance; and (ii) a Variance-Gated Normalization (VGN) layer that generalizes the variance-gated uncertainty mechanism to training via per-class, learnable normalization of ensemble member probabilities. We derive closed-form vector-Jacobian products enabling end-to-end training through ensemble sample mean and variance. VGE matches or exceeds state-of-the-art information-theoretic baselines while remaining computationally efficient. As a result, VGE provides a practical and scalable approach to epistemic-aware uncertainty estimation in ensemble models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑