用于最优臂识别与流体分析的最优Top-Two方法
机器学习
2024-12-17 v2 信息论
math.IT
机器学习
摘要
Top- 方法在解决最优臂识别(BAI)问题中变得流行。最优臂,即在有限多个臂中具有最大均值的臂,通过一种算法来识别:在任何序贯步骤中,该算法以固定概率 独立地拉动经验最优臂,否则拉动最佳挑战者臂。这保证了错误选择的概率低于指定的 。对于 BAI 问题,样本复杂度的信息论下界是众所周知的,并且随着 ,计算代价高的插入式方法可以渐近地匹配该下界。对于任何 ,上述 top 2 算法的样本复杂度与下界相差一个常数因子。然而,确定匹配下界的最优 被证明是困难的。在本文中,我们解决了这一问题并提出了一种最优的 top-2 型算法。我们考虑一个以阈值为锚定的分配函数。如果它超过阈值,则算法对经验最优臂进行采样。否则,它对挑战者臂进行采样。我们证明了所提出的算法在 时是最优的。我们的分析依赖于识别出分配的极限流体动力学,该动力学满足拼接在一起的常微分方程组,并描述了算法所遵循的渐近路径。我们依赖隐函数定理来证明这些流体常微分方程的存在性与唯一性,并证明所提出的算法始终接近该常微分方程的解。
引用
@article{arxiv.2403.09123,
title = {Optimal Top-Two Method for Best Arm Identification and Fluid Analysis},
author = {Agniv Bandyopadhyay and Sandeep Juneja and Shubhada Agrawal},
journal= {arXiv preprint arXiv:2403.09123},
year = {2024}
}
备注
To appear in NeurIPS 2024