cs.LGSep 27, 2026

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

Authors: Idan Attias, Orin Levy, Alexander Ryabchenko, Yishay Mansour, Uri Stemmer

Organizations: University of Illinois Chicago · Toyota Technological Institute at Chicago · Tel Aviv University · University of Toronto · Vector Institute · Google Research

Abstract

Can vanilla policy optimization explore enough to achieve near-optimal regret in stochastic contextual bandits? We show that standard exponential policy updates driven by offline regression do so under realizability, without exploration bonuses or importance weighting. For AA actions, TT rounds, and a finite prediction class FF, vanilla PO achieves O~(ATlog⁡(∣F∣))\widetilde O(\sqrt{AT\log(|F|)}) regret with high probability. Our analysis reveals an implicit exploration mechanism of independent interest: gradual policy updates prevent actions from losing probability too quickly, allowing the regression oracle to learn their expected losses. We further develop a batched version using only O(log⁡T)O(\log T) regression calls and policy switches, and show how private regression oracles yield differentially private contextual bandit algorithms without composition across batches. For a finite class, this gives pure εpriv\varepsilon_{\rm priv}-DP and regret O~(ATlog⁡(∣F∣/δ)(1+εpriv−1/2))\widetilde O\left( \sqrt{AT \log(|F|/δ)}(1+\varepsilon_{\rm priv}^{-1/2}) \right). Finally, experiments across oracle-based contextual bandit algorithms, with and without privacy, demonstrate the practical effectiveness of policy optimization and the value of explicit exploration under stronger privacy constraints.

Figures & tables

Appendix figures & tables6 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. AdaPrivate-TS: Private Thompson Sampling for Contextual Bandits with Privacy Amplification

    Jun 19, 2026Mohammadreza Riyazat, Eranga UkwattaContextual Bandit FrameworkThompson Sampling

  2. Decision-Focused On-Policy Learning for Contextual Linear Optimization with Partial Feedback

    May 31, 2026Wyame Benslimane, Tinghan Ye, Pascal Van Hentenryck +1Contextual Bandit FrameworkOn-Policy