Best Arm Identification with Fixed Budget: A Large Deviation Perspective
Po-An Wang, Ruo-Chun Tzeng, Alexandre Proutière
Abstract
We consider the problem of identifying the best arm in stochastic Multi-Armed Bandits (MABs) using a fixed sampling budget. Characterizing the minimal instance-specific error probability for this problem constitutes one of the important remaining open problems in MABs. When arms are selected using a static sampling strategy, the error probability decays exponentially with the number of samples at a rate that can be explicitly derived via Large Deviation techniques. Analyzing the performance of algorithms with adaptive sampling strategies is however much more challenging. In this paper, we establish a connection between the Large Deviation Principle (LDP) satisfied by the empirical proportions of arm draws and that satisfied by the empirical arm rewards. This connection holds for any adaptive algorithm, and is leveraged (i) to improve error probability upper bounds of some existing algorithms, such as the celebrated (Successive Rejects) algorithm , and (ii) to devise and analyze new algorithms. In particular, we present (Continuous Rejects), a truly adaptive algorithm that can reject arms in any round based on the observed empirical gaps between the rewards of various arms. Applying our Large Deviation results, we prove that enjoys better performance guarantees than existing algorithms, including . Extensive numerical experiments confirm this observation.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 99a8e80f-0b16-45b5-9bb4-743d5c613b20Cited by top-tier papers5
- Efficient Prompt Optimization Through the Lens of Best Arm IdentificationChengshuai Shi, Kun Yang, Zihan Chen, Jundong Li et al.NeurIPS 2024 · 44 citations
- In-Context Learning for Pure ExplorationAlessio Russo, Ryan Welch, Aldo PacchianoICLR 2026 · 5 citations
- On Universally Optimal Algorithms for A/B TestingPo-An Wang, Kaito Ariu, Alexandre ProutièreICML 2024 · 4 citations
- Training a Generally Curious AgentFahim Tajwar, Yiding Jiang, Abitha Thankaraj, Sumaita Sadia Rahman et al.ICML 2025
- Variance Driven Exploration: A Provable and Efficient Methodology for Pure Exploration in Highly Stochastic EnvironmentsKhang Luong, Nam Nguyen, Hoang Ta, Hung Tran-The et al.ICML 2026
Builds on3
- Fast Pure Exploration via Frank-WolfePo-An Wang, Ruo-Chun Tzeng, Alexandre ProutièreNeurIPS 2021 · 56 citations
- Minimax Optimal Fixed-Budget Best Arm Identification in Linear BanditsJunwen Yang, Vincent Y. F. TanNeurIPS 2022 · 38 citations
- Minimax Optimal Algorithms for Fixed-Budget Best Arm IdentificationJunpei Komiyama, Taira Tsuchiya, Junya HondaNeurIPS 2022 · 27 citations
Related papers
- Optimal Best-arm Identification in Linear BanditsYassir Jedra, Alexandre ProutièreNeurIPS 2020 · 99 citations
- Exploring Best Arm with Top Reward-Cost Ratio in Stochastic BanditsZhida Qin, Xiaoying Gan, Jia Liu, Hongqiu Wu et al.INFOCOM 2020 · 7 citations
- Best Arm Identification in Contaminated Stochastic BanditsArpan Mukherjee, Ali Tajer, Pin-Yu Chen, Payel DasNeurIPS 2021 · 1 citation
- Quantile Bandits for Best Arms IdentificationMengyan Zhang, Cheng Soon OngICML 2021 · 13 citations
- Best Arm Identification for Stochastic Rising BanditsMarco Mussi, Alessandro Montenegro, Francesco Trovò, Marcello Restelli et al.ICML 2024 · 4 citations
