English

Bayesian Best-Arm Identification with Abstention: A Polynomial-to-Exponential Phase Transition

Machine Learning 2026-06-28 v1 Information Theory Machine Learning

Abstract

We study the Bayesian fixed-budget best-arm identification problem in which a learner can abstain from making a terminal recommendation. Subject to an abstention budget α\alpha, we analyze the probability of undetected error--the risk of recommending a suboptimal arm without abstaining. Our central finding is that abstention induces a phase transition: without abstention, the error probability decays polynomially in the sampling budget TT; in contrast, introducing any small positive abstention budget shifts this to an exponential decay. For Gaussian priors and rewards, in the regime TT\to\infty followed by α0\alpha\downarrow0, we establish exact matching information-theoretic lower bounds and algorithmic upper bounds on the optimal error exponent, which takes the form exp(α2T8κν2)\exp(-\frac{\alpha^{2}T}{8\kappa_{\nu}^{2}}). The hardness parameter κν\kappa_{\nu} represents the prior density of the top-two gap at zero, highlighting that nearly tied instances drive the fundamental error. We introduce an adaptive algorithm, PGWS, that successfully achieves this optimal exponent by expending its abstention budget on statistically ambiguous instances. We further demonstrate that this polynomial-to-exponential improvement is exclusively a Bayesian phenomenon--in the frequentist setting, abstention only affects lower-order exponent terms. We also extend our results beyond the Gaussian model.

Keywords

Cite

@article{arxiv.2606.29203,
  title  = {Bayesian Best-Arm Identification with Abstention: A Polynomial-to-Exponential Phase Transition},
  author = {Yuqi Huang and Yunlong Hou and Vincent Y. F. Tan},
  journal= {arXiv preprint arXiv:2606.29203},
  year   = {2026}
}