Best Arm Identification for Bandits with Shifting Means
stat.ML, cs.LG
Submitted: 2026-10-07
Updated: 2026-10-07
Comments: Accepted at NeurIPS 2026
Code: https://github.com/LukasZierahn/Shifting-Means-Experiments
License: http://creativecommons.org/licenses/by/4.0/
The gist: We study the best arm identification problem in a stochastic environment with a novel form of adversarial perturbations, which we coin Shifting Means.
Terminology
Abstract
We study the best arm identification problem in a stochastic environment with a novel form of adversarial perturbations, which we coin Shifting Means. While classically the mean rewards of the K arms are stable in time, in Shifting Means only the gaps Δ between mean rewards are stable, while their common shift may be determined adversarially in each round. The objective of the learner is to identify the best arm with high probability while minimizing sample complexity (the fixed confidence setting). Handling shifts requires new tools: we show that algorithms employing a Generalized Likelihood Ratio Test (GLRT) stopping rule, including the popular Track-and-Stop, fail under time-varying shifts. Instead, we propose Importance Weights for Shifting Means. Assuming means bounded by U and σ squared-sub-Gaussian rewards, we show to be δ-correct and to enjoy a sample complexity bound of order K (σ squared + U 2) Δ-2 δ. We also present a matching (up to constant factors) worst-case lower bound and evaluate our results empirically.
Sources
Related papers
- Behavior of prediction performance metrics with rare events
- Optimal Estimation of Generic Dynamics by Path-Dependent Neural Jump ODEs
- A Posterior-Dynamics Framework for Imaging Inverse Problems with Pretrained Diffusion Priors
- Efficient quadratic entropy with distance sketches
- Embedding-Bias in Conditional Independence Testing
- From Log-Odds to Shapley Values: An Explanatory Geometry for the Weighted Naive Bayes Classifier