stat.MLApr 5, 2026

Nearly Optimal Best Arm Identification for Semiparametric Bandits

Authors: Seok-Jin Kim

Organizations: Columbia University

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.

Explore similar work

CardsList
  1. On The Complexity of Best-Arm Identification in Non-Stationary Linear Bandits

    Mar 11, 2026Leo Maynard-Zhang, Zhihan Xiong, Kevin Jamieson +1Multi-Armed BanditsBest-Arm Identification

  2. On the Power of Adaptivity for ε\varepsilon-Best Arm Identification in Linear Bandits

    May 15, 2026Arnab Maiti, Yunbei Xu, Kevin JamiesonMulti-Armed BanditsBest-Arm Identification