English

Nearly Optimal Best Arm Identification for Semiparametric Bandits

Machine Learning 2026-04-07 v1 Machine Learning Methodology

Abstract

We study fixed-confidence Best Arm Identification (BAI) in semiparametric bandits, where rewards are linear in arm features plus an unknown additive baseline shift. Unlike linear-bandit BAI, this setting requires orthogonalized regression, and its instance-optimal sample complexity has remained open. For the transductive setting, we establish an attainable instance-dependent lower bound characterized by the corresponding linear-bandit complexity on shifted features. We then propose a computationally efficient phase-elimination algorithm based on a new XYXY-design for orthogonalized regression. Our analysis yields a nearly optimal high-probability sample-complexity upper bound, up to log factors and an additive d2d^2 term, and experiments on synthetic instances and the Jester dataset show clear gains over prior baselines.

Keywords

Cite

@article{arxiv.2604.03969,
  title  = {Nearly Optimal Best Arm Identification for Semiparametric Bandits},
  author = {Seok-Jin Kim},
  journal= {arXiv preprint arXiv:2604.03969},
  year   = {2026}
}

Comments

To appear at AISTATS 2026