Optimal Estimation of the Best Mean in Multi-Armed Bandits
Takayuki Osogami, Junya Honda, Junpei Komiyama
Abstract
We study the problem of estimating the mean reward of the best arm in a multi-armed bandit (MAB) setting. Specifically, given a target precision ε and confidence level 1 − δ , the goal is to return an ε -accurate estimate of the largest mean reward with probability at least 1 − δ , while minimizing the number of samples. We first establish an instance-dependent lower bound on the sample complexity, which requires handling the infinitely many possible candidates of the estimated best mean. This lower bound is expressed in a non-convex optimization problem, which becomes the main difficulty of this problem, preventing the direct application of standard techniques such as Track-and-Stop to provably achieve optimality. To overcome this difficulty, we introduce several new algorithmic and analytical techniques and propose an algorithm that achieves the asymptotic lower bound with matching constants in the leading term. Our method combines a confidence ellipsoid-based stopping condition with a two-phase sampling strategy tailored to manage non-convexity proposed algorithm is simple, nearly free of hyperparameters, and achieves the instance-dependent, asymptotically optimal sample complexity. Experimental results support our theoretical guarantees and demonstrate the practical effectiveness of our method.
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 de1069e3-5e84-4bee-9c4a-dff91946efebCited by top-tier papers1
Ask how each one uses itBuilds on7
- Fast Pure Exploration via Frank-WolfePo-An Wang, Ruo-Chun Tzeng, Alexandre ProutièreNeurIPS 2021 · 56 citations
- An Optimal Elimination Algorithm for Learning a Best ArmAvinatan Hassidim, Ron Kupfer, Yaron SingerNeurIPS 2020 · 17 citations
- An ε-Best-Arm Identification Algorithm for Fixed-Confidence and BeyondMarc Jourdan, Rémy Degenne, Emilie KaufmannNeurIPS 2023 · 15 citations
- Biases in Evaluation of Molecular Optimization Methods and Bias Reduction StrategiesHiroshi Kajino, Kohei Miyaguchi, Takayuki OsogamiICML 2023 · 1 citation
- Is Best-of-N the Best of Them? Coverage, Scaling, and Optimality in Inference-Time AlignmentAudrey Huang, Adam Block, Qinghua Liu, Nan Jiang et al.ICML 2025
Related papers
- Optimal Multi-Fidelity Best-Arm IdentificationRiccardo Poiani, Rémy Degenne, Emilie Kaufmann, Alberto Maria Metelli et al.NeurIPS 2024 · 9 citations
- Multi-Fidelity Best-Arm IdentificationRiccardo Poiani, Alberto Maria Metelli, Marcello RestelliNeurIPS 2022 · 12 citations
- Optimal Best-arm Identification in Linear BanditsYassir Jedra, Alexandre ProutièreNeurIPS 2020 · 99 citations
- Best Arm Identification in Contaminated Stochastic BanditsArpan Mukherjee, Ali Tajer, Pin-Yu Chen, Payel DasNeurIPS 2021 · 1 citation
- Fixed Confidence Best Arm Identification in the Bayesian SettingKyoungseok Jang, Junpei Komiyama, Kazutoshi YamazakiNeurIPS 2024
