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

引入一种改进的预测不确定性信息论度量

Introducing an Improved Information-Theoretic Measure of Predictive Uncertainty

ELLIS Unit Linz and LIT AI Lab, Institute for Machine Learning(ELLIS 林茨分部与 LIT 人工智能实验室,机器学习研究所);Johannes Kepler University Linz(约翰开普勒林茨大学)

Kajetan Schweighofer, Lukas Aichberger, Mykyta Ielanskyi, Sepp Hochreiter

AI总结:

针对现有基于BMA熵的预测不确定性度量的缺陷,提出一种理论完善的度量,在合成任务和ImageNet上验证其更合理且适用于现实应用。

摘要:

将机器学习模型应用于现实世界的决策,需要区分模型已知的内容与未知的内容。评估模型知识的一个关键因素是量化其预测不确定性。预测不确定性通常通过贝叶斯模型平均(BMA)预测分布的熵来衡量。然而,这种当前预测不确定性度量的适当性最近受到了质疑。我们针对这些局限性提供了新的见解。我们的分析表明,当前度量错误地假设BMA预测分布等同于生成数据集的真实模型的预测分布。因此,我们引入了一种具有理论基础的度量来克服这些局限性。我们通过实验验证了所引入的预测不确定性度量的优势。我们发现,在受控的合成任务中,我们引入的度量表现更为合理。此外,我们在ImageNet上的评估表明,我们引入的度量在利用预测不确定性的现实世界应用中具有优势。

英文摘要:

Applying a machine learning model for decision-making in the real world requires to distinguish what the model knows from what it does not. A critical factor in assessing the knowledge of a model is to quantify its predictive uncertainty. Predictive uncertainty is commonly measured by the entropy of the Bayesian model average (BMA) predictive distribution. Yet, the properness of this current measure of predictive uncertainty was recently questioned. We provide new insights regarding those limitations. Our analyses show that the current measure erroneously assumes that the BMA predictive distribution is equivalent to the predictive distribution of the true model that generated the dataset. Consequently, we introduce a theoretically grounded measure to overcome these limitations. We experimentally verify the benefits of our introduced measure of predictive uncertainty. We find that our introduced measure behaves more reasonably in controlled synthetic tasks. Moreover, our evaluations on ImageNet demonstrate that our introduced measure is advantageous in real-world applications utilizing predictive uncertainty.

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

↑