Breaking the Barrier: Better Batched Best Arm Identification with Adaptive Grids
arXiv:2501.17370
Abstract
We investigate the problem of batched best arm identification in multi-armed bandits, where we aim to identify the best arm from a set of arms while minimizing both the number of samples and batches. We introduce an algorithm that achieves near-optimal sample complexity and features an instance-sensitive batch complexity, which breaks the barrier. The main contribution of our algorithm is a novel sample allocation scheme that effectively balances exploration and exploitation for batch sizes. Experimental results indicate that our approach is more batch-efficient across various setups. We also extend this framework to the problem of batched best arm identification in linear bandits and achieve similar improvements.
arXiv comment21 pages, published at ICLR 2025