UCB for Large-Scale Pure Exploration: Beyond Sub-Gaussianity

Abstract:Selecting the best alternative from a finite set is the central objective of ranking and selection (R&S) and best arm identification (BAI). Traditional R&S or BAI approaches have predominantly relied on Gaussian or sub-Gaussian assumptions on the performance distributions of all alternatives, which limit their applicability to non-sub-Gaussian---especially heavy-tailed---problems. The need to move beyond sub-Gaussianity may become even more critical in large-scale problems, which tend to be especially sensitive to distributional specifications. In this paper, motivated by the widespread use of upper confidence bound (UCB) algorithms in sequential decision making, we investigate their performance in large-scale, non-sub-Gaussian R&S settings. We consider the simplest category of UCB algorithms, where the UCB value for each alternative is defined as the sample mean plus an exploration bonus that depends only on its own sample size. We abstract this into a meta-UCB algorithm and propose letting it select the alternative with the largest sample size as the best upon stopping. For this meta-UCB algorithm, we first derive a distribution-free lower bound on the probability of correct selection. Building on this bound, we first study the indifference-zone formulation and show that the meta-UCB algorithm---and therefore a broad class of UCB algorithms---achieves sample optimality as long as the variances are uniformly bounded. We then extend sample optimality to a non-indifference-zone configuration with polynomially shrinking mean gaps under the same variance assumption. These results demonstrate the applicability of UCB algorithms to large-scale R&S problems with non-sub-Gaussian distributions. Numerical experiments support our results and provide additional insights into the comparative behaviors of UCB algorithms within and beyond our meta-UCB framework.

Machine Learning (stat.ML); Machine Learning (cs.LG)

arXiv:2511.22273 [stat.ML]

(or arXiv:2511.22273v2 [stat.ML] for this version)

https://doi.org/10.48550/arXiv.2511.22273 Focus to learn more arXiv-issued DOI via DataCite

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.