cs.LGMar 2, 2026

Stochastic Multi-Armed Bandits with Limited Control Variates

Authors: Arun Verma, Manjesh Kumar Hanawal, Arun Rajkumar

Organizations: Singapore-MIT Alliance for Research and Technology Centre · Department of IEOR, Indian Institute of Technology Bombay · Department of DSAI, Indian Institute of Technology Madras

Abstract

Motivated by wireless networks where interference or channel state estimates provide partial insight into throughput, we study a variant of the classical stochastic multi-armed bandit problem in which the learner has limited access to auxiliary information. Recent work has shown that such auxiliary information, when available as control variates, can be used to get tighter confidence bounds, leading to lower regret. However, existing works assume that control variates are available in every round, which may not be realistic in several real-life scenarios. To address this, we propose UCB-LCV, an upper confidence bound (UCB) based algorithm that effectively combines the estimators obtained from rewards and control variates. When there is no control variate, UCB-LCV leads to a novel algorithm that we call UCB-NORMAL, outperforming its existing algorithms for the standard MAB setting with normally distributed rewards. Finally, we discuss variants of the proposed UCB-LCV that apply to general distributions and experimentally demonstrate that UCB-LCV outperforms existing bandit algorithms.

Figures & tables

Explore similar work

Aug 18, 2026

Toward the Optimal Regret-Instability Trade-off in Multi-Armed Bandits

Multi-armed bandit algorithms are evaluated by regret, yet comparable regret can coexist with different allocations across independent runs. We study the trade-off between worst-case regret RK,T\mathcal{R}_{K,T} and instability SK,T\mathcal S_{K,T}, defined as the largest standard deviation of a terminal pull count, for KK arms and TT rounds. We prove the finite-time lower bound RK,TSK,T≥CT3/2\mathcal R_{K,T}\mathcal S_{K,T}\ge C T^{3/2}, where CC is independent of KK and TT, under a finite-time regret condition and without the regularity assumptions imposed in the prior asymptotic analysis. We also introduce Stabilized Lower-Envelope UCB (\textup{SLE-UCB}), a new tunable algorithm combining a running lower-envelope index with a decreasing pull-count stabilizer. \textup{SLE-UCB} satisfies RK,TSK,T=O(T3/2log⁡K)\mathcal R_{K,T}\mathcal S_{K,T}=O(T^{3/2}\log K), with an implicit constant independent of KK and TT, matching the lower bound exactly in TT and within a logarithmic factor in KK. To prove the instability bound, we develop a new offline top-prefix representation that removes path dependence from online decisions. Together with single-reward perturbations and the Efron--Stein inequality, this representation controls pull-count variance. Thus, regret and instability depend reciprocally on KK, while their product has no polynomial dependence on KK. These results resolve the open question raised in the literature concerning the sharp arm-dependent regret--instability frontier.
Apr 21, 2026

Replicable Bandits with UCB based Exploration

We study replicable algorithms for stochastic multi-armed bandits (MAB) and linear bandits with UCB (Upper Confidence Bound) 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 1−ρ1-ρ. Prior approaches to this problem are elimination-based and, in linear bandits with infinitely many actions, rely on discretization, leading to suboptimal dependence on the dimension dd and ρρ. We develop optimistic alternatives for both settings. For stochastic multi-armed bandits, we propose RepUCB, a replicable batched UCB algorithm and show that it attains a regret O ⁣(K2log⁡2Tρ2∑a:Δa>0(Δa+log⁡(KTlog⁡T)Δa))O\!\left(\frac{K^2\log^2 T}{ρ^2}\sum_{a:Δ_a>0}\left(Δ_a+\frac{\log(KT\log T)}{Δ_a}\right)\right). For stochastic linear bandits, we first introduce RepRidge, a replicable ridge regression estimator that satisfies both a confidence guarantee and a ρρ-replicability guarantee. Beyond its role in our bandit algorithm, this may also be of independent interest in other statistical estimation settings. We then use RepRidge to design RepLinUCB, a replicable optimistic algorithm for stochastic linear bandits, and show that its regret is bounded by O~ ⁣((d+d3ρ)T)\widetilde{O}\!\big(\big(d+\frac{d^3}ρ\big)\sqrt{T}\big). This improves the best prior regret guarantee by a factor of O(d/ρ)O(d/ρ), showing that our optimistic algorithm can substantially reduce the price of replicability. This is the first linear-bandit algorithm with an optimal dependence on ρρ for large number of arms. Finally, we extend our framework to stochastic generalized linear bandits by developing RepGLM, a replicable penalized GLM estimator, and RepGLMUCB, a replicable optimistic algorithm for this setting.
Feb 16, 2026

Truly Adapting to Adversarial Constraints in Constrained MABs

We study the constrained variant of the multi-armed bandit (MAB) problem, in which the learner aims not only at minimizing the total loss incurred during the learning dynamic, but also at controlling the violation of multiple unknown constraints, under both full and bandit feedback. We consider a non-stationary environment that subsumes both stochastic and adversarial models and where, at each round, both losses and constraints are drawn from distributions that may change arbitrarily over time. In such a setting, it is provably not possible to guarantee both sublinear regret and sublinear violation. Accordingly, prior work has mainly focused either on settings with stochastic constraints or on relaxing the benchmark with fully adversarial constraints (e.g., via competitive ratios with respect to the optimum). We provide the first algorithms that achieve optimal rates of regret and positive constraint violation when the constraints are stochastic while the losses may vary arbitrarily, and that simultaneously yield guarantees that degrade smoothly with the degree of adversariality of the constraints. Specifically, under full feedback we propose an algorithm attaining O~(T+C)\widetilde{\mathcal{O}}(\sqrt{T}+C) regret and O~(T+C)\widetilde{\mathcal{O}}(\sqrt{T}+C) {positive} violation, where CC quantifies the amount of non-stationarity in the constraints. We then show how to extend these guarantees when only bandit feedback is available for the losses. Finally, when bandit feedback is available for the constraints, we design an algorithm achieving O~(T+C)\widetilde{\mathcal{O}}(\sqrt{T}+C) {positive} violation and O~(T+CT)\widetilde{\mathcal{O}}(\sqrt{T}+C\sqrt{T}) regret.