mlsanityData to 7 Oct 2026

Off-Policy Evaluation via Adaptive Weighting with Data from Contextual Bandits

arXiv 2106.02029 ↗First posted 3 Jun 2021stat.ML cs.LG econ.EMcited by 40

Not on the map. Why?

Abstract (arXiv)

It has become increasingly common for data to be collected adaptively, for example using contextual bandits. Historical data of this type can be used to evaluate other treatment assignment policies to guide future innovation or experiments. However, policy evaluation is challenging if the target policy differs from the one used to collect data, and popular estimators, including doubly robust (DR) estimators, can be plagued by bias, excessive variance, or both. In particular, when the pattern of treatment assignment in the collected data looks little like the pattern generated by the policy to be evaluated, the importance weights used in DR estimators explode, leading to excessive variance. In this paper, we improve the DR estimator by adaptively weighting observations to control its variance. We show that a t-statistic based on our improved estimator is asymptotically normal under certain conditions, allowing us to form confidence intervals and test hypotheses. Using synthetic data and public benchmarks, we provide empirical evidence for our estimator's improved accuracy and inferential properties relative to existing alternatives.

Key facts. Cited by 40 papers in the map (data through 7 Oct 2026). Builds on 15 of its 40 references that are in the map. First posted on arXiv 3 Jun 2021.

Builds on

15 of 40 references.

References · 15

  1. 2014OpenML: networked science in machine learningcited by 523
  2. 2017A Tutorial on Thompson Samplingcited by 420
  3. 2015Multi-armed Bandit Models for the Optimal Design of Clinical Trials: Benefits and Challengescited by 113
  4. 2016Optimal and Adaptive Off-policy Evaluation in Contextual Banditscited by 111
  5. 2019Confidence Intervals for Policy Evaluation in Adaptive Experimentscited by 95
  6. 2019Doubly robust off-policy evaluation with shrinkagecited by 87
  7. 2020Inference for Batched Banditscited by 63
  8. 2016Double/Debiased Machine Learning for Treatment and Causal Parameterscited by 41
  9. 2017Why Adaptively Collected Data Have Negative Bias and How to Correct for Itcited by 36
  10. 2020Optimal Off-Policy Evaluation from Multiple Logging Policiescited by 30
  11. 2019Are sample means in multi-armed bandits positively or negatively biased?cited by 24
  12. 2017Estimation Considerations in Contextual Banditscited by 22
3 more
  1. 2017Effective Evaluation using Logged Bandit Feedback from Multiple Loggerscited by 20
  2. 2020On conditional versus marginal bias in multi-armed banditscited by 9
  3. 2020Tractable contextual bandits beyond realizabilitycited by 8

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2021: 3; 2022: 10; 2023: 8; 2024: 7; 2025: 5; 2026: 7.

40 papers.

Sort:

By area:

  1. 2022Anytime-valid off-policy inference for contextual banditscited by 37
  2. 2021Offline Neural Contextual Bandits: Pessimism, Optimization and Generalizationcited by 25
  3. 2022On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximationcited by 22
  4. 2024UCB algorithms for multi-armed bandits: Precise regret and adaptive inferencecited by 16
  5. 2022Counterfactual inference in sequential experimentscited by 13
  6. 2024Active Adaptive Experimental Design for Treatment Effect Estimation with Covariate Choicescited by 12
  7. 2022Off-policy estimation of linear functionals: Non-asymptotic theory for semi-parametric efficiencycited by 10
  8. 2024Inference with the Upper Confidence Bound Algorithmcited by 9
  9. 2022Online Statistical Inference for Contextual Bandits via Stochastic Gradient Descentcited by 7
  10. 2023Proportional Response: Contextual Bandits for Simple and Cumulative Regret Minimizationcited by 7
  11. 2022Contextual Bandits in a Survey Experiment on Charitable Giving: Within-Experiment Outcomes versus Policy Learningcited by 6
  12. 2023Semi-parametric inference based on adaptively collected datacited by 6
28 more
  1. 2023Adaptive Linear Estimating Equationscited by 6
  2. 2023Using Adaptive Bandit Experiments to Increase and Investigate Engagement in Mental Healthcited by 6
  3. 2025Stable Thompson Sampling: Valid Inference via Variance Inflationcited by 6
  4. 2021Doubly Robust Interval Estimation for Optimal Policy Evaluation in Online Learningcited by 4
  5. 2022Optimal Best Arm Identification in Two-Armed Bandits with a Fixed Budget under a Small Gapcited by 4
  6. 2023Online learning in bandits with predicted contextcited by 4
  7. 2023Online Estimation and Inference for Robust Policy Evaluation in Reinforcement Learningcited by 4
  8. 2024Allocation Stability and Wald Inference under Variance-Aware UCBcited by 4
  9. 2025Choosing the Better Bandit Algorithm under Data Sharing: When Do A/B Experiments Work?cited by 4
  10. 2022Best Arm Identification with Contextual Information under a Small Gapcited by 3
  11. 2023Uncertainty-Aware Instance Reweighting for Off-Policy Learningcited by 3
  12. 2023Statistical Limits of Adaptive Linear Models: Low-Dimensional Estimation and Inferencecited by 3
  13. 2024Online Policy Learning and Inference by Matrix Completioncited by 3
  14. 2024Off-policy estimation with adaptively collected data: the power of online learningcited by 3
  15. 2025Simulation-Based Inference for Adaptive Experimentscited by 3
  16. 2026Valid Best-Model Identification for LLM Evaluation via Low-Rank Factorizationcited by 3
  17. 2022Adversarial Estimatorscited by 2
  18. 2022Catoni-style Confidence Sequences under Infinite Variancecited by 2
  19. 2025Clustering Context in Off-Policy Evaluationcited by 2
  20. 2026Decision Support for Marketplace Policies under Incomplete Evidence: From Replay to Launch Readinesscited by 2
  21. 2021Dynamic Selection in Algorithmic Decision-makingcited by 1
  22. 2024Demistifying Inference after Adaptive Experimentscited by 1
  23. 2026Design Experiments to Compare Multi-armed Bandit Algorithmscited by 1
  24. 2025SNPL: Simultaneous Policy Learning and Evaluation for Safe Multi-Objective Policy Improvementcited by 0
  25. 2026Performative Learning Theorycited by 0
  26. 2026Fixed-Horizon Self-Normalized Inference for Adaptive Experiments via Martingale AIPW/DML with Logged Propensitiescited by 0
  27. 2026Bandit Simulation for Average Reward Inferencecited by 0
  28. 2026Fast A/B/n Testing: Exact Multi-Policy Comparison via Tree-Coupled Feedback Sharingcited by 0