arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.08151cs.LGstat.ML
收藏

面向预测的贝叶斯主动学习

Prediction-Oriented Bayesian Active Learning

University of Oxford(牛津大学);Microsoft Research(微软研究院)

Freddie Bickford Smith, Andreas Kirsch, Sebastian Farquhar, Yarin Gal, Adam Foster, Tom Rainforth

AI总结:

本文提出期望预测信息增益(EPIG)采集函数,通过在预测空间而非参数空间度量信息增益,解决了 BALD 忽略输入分布导致的数据选择次优问题,并在多个数据集和模型上验证了其优于 BALD 的预测性能。

摘要:

主动学习的信息论方法传统上侧重于最大化关于模型参数所获取的信息,最常见的是通过优化 BALD 分数来实现。我们强调,从预测性能的角度来看,这可能是次优的。例如,BALD 缺乏输入分布的概念,因此容易优先选择相关性有限的数据。为了解决这个问题,我们提出了期望预测信息增益(EPIG),这是一种在预测空间而非参数空间中度量信息增益的采集函数。我们发现,与 BALD 相比,在一系列数据集和模型上使用 EPIG 能带来更强的预测性能,因此它提供了一种有吸引力的直接替代方案。

英文摘要:

Information-theoretic approaches to active learning have traditionally focused on maximising the information gathered about the model parameters, most commonly by optimising the BALD score. We highlight that this can be suboptimal from the perspective of predictive performance. For example, BALD lacks a notion of an input distribution and so is prone to prioritise data of limited relevance. To address this we propose the expected predictive information gain (EPIG), an acquisition function that measures information gain in the space of predictions rather than parameters. We find that using EPIG leads to stronger predictive performance compared with BALD across a range of datasets and models, and thus provides an appealing drop-in replacement.

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

↑