Anytime-valid Optimal Policy Identification
Daniel Molitor
stat.ME, stat.ML
Submitted: 2026-06-16
Comments: 15 pages, 3 figures
Code: https://github.com/dmolitor/av-policy-selection
License: http://creativecommons.org/licenses/by/4.0/
The gist: We develop an anytime-valid framework for optimal policy identification from logged contextual bandit data.
Terminology
Abstract
We develop an anytime-valid framework for optimal policy identification from logged contextual bandit data. In many applied settings, the analyst wants to select the optimal policy from a candidate policy class, but data are generated by an externally determined logging policy that they do not control. The analyst may also wish to monitor evidence continuously, stopping once the optimal policy is clear rather than committing to a fixed sample size in advance. This paper addresses these challenges by constructing a time-indexed set S t that retains the true optimal policy set uniformly over time with high probability. The resulting procedure allows the analyst to monitor policy values, eliminate clearly suboptimal policies, and stop at data-dependent times without invalidating inference. When the optimal policy is unique, we define a stopping time for its identification and derive a sample-complexity bound scaling as O! (+ (1/) over squared), where is the gap between the best and second-best policy values. Simulations demonstrate that the anytime-valid approach can yield substantial sample savings relative to fixed- N designs. An application to a large adaptive experiment on reducing misinformation online illustrates how the method provides a dynamic view as evidence on the optimal policy accumulates.
Sources
- Post-Contextual-Bandit Inference
- Multiple Identifications in Multi-Armed Bandits
- Gamification of Pure Exploration for Linear Bandits
- Doubly Robust Policy Evaluation and Learning
- Sequential Experimental Design for Transductive Linear Bandits
- Optimal Best Arm Identification with Fixed Confidence
- Confidence Intervals for Policy Evaluation in Adaptive Experiments
- Time-uniform, nonparametric, nonasymptotic confidence sequences
- Optimal Best-arm Identification in Linear Bandits
- Off-policy Confidence Sequences
- Confident Off-Policy Evaluation and Selection through Self-Normalized Importance Weighting
- Adaptive Off-Policy Inference for M-Estimators Under Model Misspecification
- A Contextual-Bandit Approach to Personalized News Article Recommendation
- Instance-optimal PAC Algorithms for Contextual Bandits
- Anytime-Valid Linear Models and Regression Adjusted Causal Inference in Randomized Experiments
- An optimal algorithm for the Thresholding Bandit Problem
- Finding All {\epsilon}-Good Arms in Stochastic Bandits
- Anytime-Valid Inference in Adaptive Experiments: Covariate Adjustment and Balanced Power
- Best-Arm Identification in Linear Bandits
- On Elimination Strategies for Bandit Fixed-Confidence Identification
Related papers
- Doubly robust inference via calibration
- Bayesian Empirical Bayes: Simultaneous Inference from Probabilistic Symmetries
- Weak-Form Recovery of Stochastic Generators
- Optimal Watermark Localization in Mixed-Source Large Language Model Texts
- Optimized Conformal Selection: Powerful Selective Inference After Data-Dependent Score Construction
- Small area learning