cs.LGOct 15, 2026

Explore Once, Evaluate Many: Multiple Policy Evaluation for Stochastic Linear Bandits

Authors: Spyros Dragazis, Yilei Chen, Aldo Pacchiano

Organizations: Boston University · Broad Institute of MIT and Harvard

Abstract

We study how to evaluate KK target policies simultaneously in stochastic contextual bandits with linear reward structure. Given a known dd-dimensional feature map and reward-free access to policy queries, we propose PRISMA (Policy evaluation through Regularized Information Sharing and Mixture Allocation), an algorithm that estimates every target policy's value to accuracy ε\varepsilon with probability at least 1−δ1-δ. The algorithm uses O~ ⁣((L2S2+β2)(1∨ρ⋆)ε2⋅log⁡(K))\widetilde{\mathcal{O}}\!\left( \frac{(L^2S^2+β^2)(1\veeρ^\star)}{\varepsilon^2} \cdot \log(K) \right) context draws and reward observations, where ρ⋆≤dρ^\star\le d is an instance-dependent measure of the difficulty of designing a shared sampling distribution. Here, LL, SS, and ββ bound the feature norm, reward parameter norm, and reward noise, respectively, and the suppressed factors are logarithmic in KK, dd, and 1/δ1/δ. Our method combines reward-free planning, a regularized sampling design, and a single ridge regression estimator shared across all target policies. An adaptive doubling procedure selects the regularization scale without knowledge of ρ⋆ρ^\star. The guarantees require no numerical lower bound on second-moment eigenvalues and have no explicit dependence on the cardinalities of the context and action spaces. Policy-query costs are accounted for separately, with KK queries per shared planning context.

Figures & tables

Appendix figures & tables1 asset

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Active Learning for Stochastic Contextual Linear Bandits

    May 24, 2026Emma Brunskill, Ishani Karmarkar, Zhaoqi LiActive LearningAdaptive Sampling

  2. The Sample Complexity of Multiclass and Sparse Contextual Bandits

    May 28, 2026Liad Erez, Fan Chen, Alon Cohen +4Combinatorial Semi-BanditsContextual Bandits

  3. Vanilla Policy Optimization Is Both Optimal and Differentially Private for Stochastic Contextual Bandits

    Sep 27, 2026Idan Attias, Orin Levy, Alexander Ryabchenko +2Policy OptimizationDifferential Privacy