Replicable Linear Bandits with UCB based Exploration
Abstract
We study replicable algorithms for stochastic linear bandits with optimism-based exploration. A bandit algorithm is -replicable if two executions using shared internal randomness but independent reward realizations produce the same action sequence with probability at least . For linear bandits with infinitely many actions, existing replicable algorithms are elimination-based and rely on discretizing the action space, leading to suboptimal dependence on the dimension and on . We first introduce RepRidge, a replicable ridge regression estimator that satisfies a self-normalized confidence bound whose radius is inflated by a factor of . We believe this estimator is of independent interest. Building on it, we design RepLinUCB, a replicable optimistic algorithm that combines determinant-triggered batching with RepRidge, and show that its regret is . This improves the state-of-the-art by a factor of and gives the first linear bandit algorithm with optimal dependence on for large action sets. Because our approach does not rely on action-space discretization, it also applies to time-varying action sets, which existing results could not handle. Finally, we extend our approach to generalized linear bandits via a replicable penalized GLM estimator, RepGLM, and a corresponding optimistic algorithm, RepGLMUCB. In synthetic experiments, RepLinUCB achieves substantially lower regret than existing elimination-based algorithms.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.