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

固定预算贝叶斯最佳臂识别中的UCB探索

UCB Exploration for Fixed-Budget Bayesian Best Arm Identification

Institute of Science and Technology for Brain-inspired Intelligence, Fudan University(复旦大学脑科学智能技术研究院);School of Mathematics and Statistics, Wuhan University(武汉大学数学与统计学院)

Rong J. B. Zhu, Yanqi Qiu

AI总结:

本文针对固定预算贝叶斯最佳臂识别问题,提出一种通过学习先验信息增强UCB探索的算法,理论证明其简单遗憾上界为 $\tilde{O}(\sqrt{K/n})$,并在实验中一致优于现有基线方法。

摘要:

我们研究固定预算设置下的最佳臂识别(BAI)问题。基于上置信界(UCB)的自适应分配方法(如UCBE)在BAI中表现良好。然而,众所周知,其最优遗憾在理论上依赖于具体实例,我们证明这在许多固定预算BAI问题中是一种人为假象。本文提出了一种UCB探索算法,在贝叶斯设置下对固定预算BAI问题具有理论和实证上的高效性。其核心思想是学习先验信息,这可以提升基于UCB的BAI算法的性能,正如它在累积遗憾最小化问题中所起的作用一样。我们为贝叶斯BAI问题建立了失败概率和简单遗憾的界,给出了阶为 $\tilde{O}(\sqrt{K/n})$ 的上界(忽略对数因子),其中 $n$ 表示预算,$K$ 表示臂的数量。此外,我们通过实证结果表明,我们的方法始终优于最先进的基线方法。

英文摘要:

We study best-arm identification (BAI) in the fixed-budget setting. Adaptive allocations based on upper confidence bounds (UCBs), such as UCBE, are known to work well in BAI. However, it is well-known that its optimal regret is theoretically dependent on instances, which we show to be an artifact in many fixed-budget BAI problems. In this paper we propose an UCB exploration algorithm that is both theoretically and empirically efficient for the fixed budget BAI problem under a Bayesian setting. The key idea is to learn prior information, which can enhance the performance of UCB-based BAI algorithm as it has done in the cumulative regret minimization problem. We establish bounds on the failure probability and the simple regret for the Bayesian BAI problem, providing upper bounds of order $\tilde{O}(\sqrt{K/n})$, up to logarithmic factors, where $n$ represents the budget and $K$ denotes the number of arms. Furthermore, we demonstrate through empirical results that our approach consistently outperforms state-of-the-art baselines.

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

↑