English

On the Power of Adaptivity for $\varepsilon$-Best Arm Identification in Linear Bandits

Machine Learning 2026-05-18 v1

Abstract

We study the minimax sample complexity of ε\varepsilon-best arm identification in linear bandits. Given a compact action set X\mathcal{X} that spans Rd\mathbb{R}^d and an unknown reward vector θRd\theta\in\mathbb{R}^d, the goal is to output an arm x^X\widehat{x}\in\mathcal{X} such that x^,θmaxxXx,θε\langle \widehat{x},\theta\rangle \ge \max_{x\in\mathcal{X}} \langle x,\theta\rangle - \varepsilon with probability at least 1δ1-\delta, using as few samples as possible. First, we present a non-adaptive fixed-design method with sample complexity O ⁣(dlog(1/δ)ε2+w(X)2ε2)\mathcal{O}\!\left(\frac{d\log(1/\delta)}{\varepsilon^2}+\frac{w(\mathcal{X})^2}{\varepsilon^2}\right), where w(X)w(\mathcal{X}) is a Gaussian width term dependent on X\mathcal{X}, and we prove a matching lower bound Ω ⁣(dlog(1/δ)ε2+w(X)2ε2)\Omega\!\left(\frac{d\log(1/\delta)}{\varepsilon^2}+\frac{w(\mathcal{X})^2}{\varepsilon^2}\right) for all non-adaptive fixed-design methods. We then turn to adaptive sampling. We raise an important structural question: beyond the canonical basis, are there structured action sets for which adaptivity yields only logarithmic-factor improvements over the optimal non-adaptive rate? We answer in the affirmative for several natural action sets, namely the hypercube, the 2\ell_2 ball, mm-sets, and multi-task multi-armed bandits. Finally, we provide the first construction of an action set X\mathcal{X} for which adaptivity yields a polynomial-factor improvement over every non-adaptive algorithm. A key ingredient behind this separation is an 2\ell_2-norm estimation subroutine: we design an adaptive algorithm that uses O ⁣(dlog(1/δ)ε2)\mathcal{O}\!\left(\frac{d\log(1/\delta)}{\varepsilon^2}\right) samples from the unit 2\ell_2 ball in Rd\mathbb{R}^d and outputs an estimate r^\widehat r satisfying r^θ2ε|\widehat r-\|\theta\|_2|\le \varepsilon with probability at least 1δ1-\delta, where θ\theta is the unknown reward vector.

Keywords

Cite

@article{arxiv.2605.15663,
  title  = {On the Power of Adaptivity for $\varepsilon$-Best Arm Identification in Linear Bandits},
  author = {Arnab Maiti and Yunbei Xu and Kevin Jamieson},
  journal= {arXiv preprint arXiv:2605.15663},
  year   = {2026}
}

Comments

Accepted at COLT 2026