English

Optimal $\delta$-Correct Best-Arm Selection for Heavy-Tailed Distributions

Machine Learning 2023-11-27 v3 Probability Machine Learning

Abstract

Given a finite set of unknown distributions or arms that can be sampled, we consider the problem of identifying the one with the maximum mean using a δ\delta-correct algorithm (an adaptive, sequential algorithm that restricts the probability of error to a specified δ\delta) that has minimum sample complexity. Lower bounds for δ\delta-correct algorithms are well known. δ\delta-correct algorithms that match the lower bound asymptotically as δ\delta reduces to zero have been previously developed when arm distributions are restricted to a single parameter exponential family. In this paper, we first observe a negative result that some restrictions are essential, as otherwise, under a δ\delta-correct algorithm, distributions with unbounded support would require an infinite number of samples in expectation. We then propose a δ\delta-correct algorithm that matches the lower bound as δ\delta reduces to zero under the mild restriction that a known bound on the expectation of (1+ϵ)th(1+\epsilon)^{th} moment of the underlying random variables exists, for ϵ>0\epsilon > 0. We also propose batch processing and identify near-optimal batch sizes to speed up the proposed algorithm substantially. The best-arm problem has many learning applications, including recommendation systems and product selection. It is also a well-studied classic problem in the simulation community.

Keywords

Cite

@article{arxiv.1908.09094,
  title  = {Optimal $\delta$-Correct Best-Arm Selection for Heavy-Tailed Distributions},
  author = {Shubhada Agrawal and Sandeep Juneja and Peter Glynn},
  journal= {arXiv preprint arXiv:1908.09094},
  year   = {2023}
}

Comments

Updated version of work that appeared in ALT 2020

R2 v1 2026-06-23T10:55:43.887Z