Off-Policy Evaluation via Adaptive Weighting with Data from Contextual Bandits
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
- 2014OpenML: networked science in machine learningcited by 523
- 2017A Tutorial on Thompson Samplingcited by 420
- 2015Multi-armed Bandit Models for the Optimal Design of Clinical Trials: Benefits and Challengescited by 113
- 2016Optimal and Adaptive Off-policy Evaluation in Contextual Banditscited by 111
- 2019Confidence Intervals for Policy Evaluation in Adaptive Experimentscited by 95
- 2019Doubly robust off-policy evaluation with shrinkagecited by 87
- 2020Inference for Batched Banditscited by 63
- 2016Double/Debiased Machine Learning for Treatment and Causal Parameterscited by 41
- 2017Why Adaptively Collected Data Have Negative Bias and How to Correct for Itcited by 36
- 2020Optimal Off-Policy Evaluation from Multiple Logging Policiescited by 30
- 2019Are sample means in multi-armed bandits positively or negatively biased?cited by 24
- 2017Estimation Considerations in Contextual Banditscited by 22
3 more
Built on by
Citing papers per year: 2021: 3; 2022: 10; 2023: 8; 2024: 7; 2025: 5; 2026: 7.
40 papers.
Sort:
By area:
- 2022Anytime-valid off-policy inference for contextual banditscited by 37
- 2021Offline Neural Contextual Bandits: Pessimism, Optimization and Generalizationcited by 25
- 2022On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximationcited by 22
- 2024UCB algorithms for multi-armed bandits: Precise regret and adaptive inferencecited by 16
- 2022Counterfactual inference in sequential experimentscited by 13
- 2024Active Adaptive Experimental Design for Treatment Effect Estimation with Covariate Choicescited by 12
- 2022Off-policy estimation of linear functionals: Non-asymptotic theory for semi-parametric efficiencycited by 10
- 2024Inference with the Upper Confidence Bound Algorithmcited by 9
- 2022Online Statistical Inference for Contextual Bandits via Stochastic Gradient Descentcited by 7
- 2023Proportional Response: Contextual Bandits for Simple and Cumulative Regret Minimizationcited by 7
- 2022Contextual Bandits in a Survey Experiment on Charitable Giving: Within-Experiment Outcomes versus Policy Learningcited by 6
- 2023Semi-parametric inference based on adaptively collected datacited by 6
28 more
- 2023Adaptive Linear Estimating Equationscited by 6
- 2023Using Adaptive Bandit Experiments to Increase and Investigate Engagement in Mental Healthcited by 6
- 2025Stable Thompson Sampling: Valid Inference via Variance Inflationcited by 6
- 2021Doubly Robust Interval Estimation for Optimal Policy Evaluation in Online Learningcited by 4
- 2022Optimal Best Arm Identification in Two-Armed Bandits with a Fixed Budget under a Small Gapcited by 4
- 2023Online learning in bandits with predicted contextcited by 4
- 2023Online Estimation and Inference for Robust Policy Evaluation in Reinforcement Learningcited by 4
- 2024Allocation Stability and Wald Inference under Variance-Aware UCBcited by 4
- 2025Choosing the Better Bandit Algorithm under Data Sharing: When Do A/B Experiments Work?cited by 4
- 2022Best Arm Identification with Contextual Information under a Small Gapcited by 3
- 2023Uncertainty-Aware Instance Reweighting for Off-Policy Learningcited by 3
- 2023Statistical Limits of Adaptive Linear Models: Low-Dimensional Estimation and Inferencecited by 3
- 2024Online Policy Learning and Inference by Matrix Completioncited by 3
- 2024Off-policy estimation with adaptively collected data: the power of online learningcited by 3
- 2025Simulation-Based Inference for Adaptive Experimentscited by 3
- 2026Valid Best-Model Identification for LLM Evaluation via Low-Rank Factorizationcited by 3
- 2022Adversarial Estimatorscited by 2
- 2022Catoni-style Confidence Sequences under Infinite Variancecited by 2
- 2025Clustering Context in Off-Policy Evaluationcited by 2
- 2026Decision Support for Marketplace Policies under Incomplete Evidence: From Replay to Launch Readinesscited by 2
- 2021Dynamic Selection in Algorithmic Decision-makingcited by 1
- 2024Demistifying Inference after Adaptive Experimentscited by 1
- 2026Design Experiments to Compare Multi-armed Bandit Algorithmscited by 1
- 2025SNPL: Simultaneous Policy Learning and Evaluation for Safe Multi-Objective Policy Improvementcited by 0
- 2026Performative Learning Theorycited by 0
- 2026Fixed-Horizon Self-Normalized Inference for Adaptive Experiments via Martingale AIPW/DML with Logged Propensitiescited by 0
- 2026Bandit Simulation for Average Reward Inferencecited by 0
- 2026Fast A/B/n Testing: Exact Multi-Policy Comparison via Tree-Coupled Feedback Sharingcited by 0
- 2026Fast A/B/n Testing: Exact Multi-Policy Comparison via Tree-Coupled Feedback Sharing13 Aug 2026
- 2026Bandit Simulation for Average Reward Inference30 May 2026
- 2026Decision Support for Marketplace Policies under Incomplete Evidence: From Replay to Launch Readiness13 May 2026
- 2026Valid Best-Model Identification for LLM Evaluation via Low-Rank Factorization11 May 2026
- 2026Design Experiments to Compare Multi-armed Bandit Algorithms6 Mar 2026
- 2026Fixed-Horizon Self-Normalized Inference for Adaptive Experiments via Martingale AIPW/DML with Logged Propensities17 Feb 2026
- 2026Performative Learning Theory4 Feb 2026
- 2025Choosing the Better Bandit Algorithm under Data Sharing: When Do A/B Experiments Work?16 Jul 2025
- 2025Simulation-Based Inference for Adaptive Experiments3 Jun 2025
- 2025Stable Thompson Sampling: Valid Inference via Variance Inflation29 May 2025
- 2025SNPL: Simultaneous Policy Learning and Evaluation for Safe Multi-Objective Policy Improvement17 Mar 2025
- 2025Clustering Context in Off-Policy Evaluation28 Feb 2025
28 more
- 2024Allocation Stability and Wald Inference under Variance-Aware UCB12 Dec 2024
- 2024UCB algorithms for multi-armed bandits: Precise regret and adaptive inference9 Dec 2024
- 2024Off-policy estimation with adaptively collected data: the power of online learning19 Nov 2024
- 2024Inference with the Upper Confidence Bound Algorithm8 Aug 2024
- 2024Demistifying Inference after Adaptive Experiments2 May 2024
- 2024Online Policy Learning and Inference by Matrix Completion26 Apr 2024
- 2024Active Adaptive Experimental Design for Treatment Effect Estimation with Covariate Choices6 Mar 2024
- 2023Using Adaptive Bandit Experiments to Increase and Investigate Engagement in Mental Health13 Oct 2023
- 2023Online Estimation and Inference for Robust Policy Evaluation in Reinforcement Learning4 Oct 2023
- 2023Statistical Limits of Adaptive Linear Models: Low-Dimensional Estimation and Inference1 Oct 2023
- 2023Online learning in bandits with predicted context26 Jul 2023
- 2023Adaptive Linear Estimating Equations14 Jul 2023
- 2023Proportional Response: Contextual Bandits for Simple and Cumulative Regret Minimization5 Jul 2023
- 2023Uncertainty-Aware Instance Reweighting for Off-Policy Learning11 Mar 2023
- 2023Semi-parametric inference based on adaptively collected data5 Mar 2023
- 2022Online Statistical Inference for Contextual Bandits via Stochastic Gradient Descent30 Dec 2022
- 2022On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximation23 Nov 2022
- 2022Contextual Bandits in a Survey Experiment on Charitable Giving: Within-Experiment Outcomes versus Policy Learning22 Nov 2022
- 2022Anytime-valid off-policy inference for contextual bandits19 Oct 2022
- 2022Off-policy estimation of linear functionals: Non-asymptotic theory for semi-parametric efficiency26 Sep 2022
- 2022Best Arm Identification with Contextual Information under a Small Gap15 Sep 2022
- 2022Catoni-style Confidence Sequences under Infinite Variance5 Aug 2022
- 2022Adversarial Estimators22 Apr 2022
- 2022Counterfactual inference in sequential experiments14 Feb 2022
- 2022Optimal Best Arm Identification in Two-Armed Bandits with a Fixed Budget under a Small Gap12 Jan 2022
- 2021Offline Neural Contextual Bandits: Pessimism, Optimization and Generalization27 Nov 2021
- 2021Doubly Robust Interval Estimation for Optimal Policy Evaluation in Online Learning29 Oct 2021
- 2021Dynamic Selection in Algorithmic Decision-making28 Aug 2021
- 2022Anytime-valid off-policy inference for contextual banditscited by 37
- 2024UCB algorithms for multi-armed bandits: Precise regret and adaptive inferencecited by 16
- 2022Counterfactual inference in sequential experimentscited by 13
- 2024Active Adaptive Experimental Design for Treatment Effect Estimation with Covariate Choicescited by 12
- 2022Off-policy estimation of linear functionals: Non-asymptotic theory for semi-parametric efficiencycited by 10
- 2024Inference with the Upper Confidence Bound Algorithmcited by 9
- 2022Online Statistical Inference for Contextual Bandits via Stochastic Gradient Descentcited by 7
- 2023Proportional Response: Contextual Bandits for Simple and Cumulative Regret Minimizationcited by 7
- 2022Contextual Bandits in a Survey Experiment on Charitable Giving: Within-Experiment Outcomes versus Policy Learningcited by 6
- 2023Semi-parametric inference based on adaptively collected datacited by 6
- 2023Adaptive Linear Estimating Equationscited by 6
- 2023Using Adaptive Bandit Experiments to Increase and Investigate Engagement in Mental Healthcited by 6
12 most cited of 36 All 36 (JSON)
- 2026Bandit Simulation for Average Reward Inference30 May 2026
- 2026Decision Support for Marketplace Policies under Incomplete Evidence: From Replay to Launch Readiness13 May 2026
- 2026Valid Best-Model Identification for LLM Evaluation via Low-Rank Factorization11 May 2026
- 2026Design Experiments to Compare Multi-armed Bandit Algorithms6 Mar 2026
- 2026Fixed-Horizon Self-Normalized Inference for Adaptive Experiments via Martingale AIPW/DML with Logged Propensities17 Feb 2026
- 2026Performative Learning Theory4 Feb 2026
- 2025Choosing the Better Bandit Algorithm under Data Sharing: When Do A/B Experiments Work?16 Jul 2025
- 2025Simulation-Based Inference for Adaptive Experiments3 Jun 2025
- 2025Stable Thompson Sampling: Valid Inference via Variance Inflation29 May 2025
- 2025Clustering Context in Off-Policy Evaluation28 Feb 2025
- 2024Allocation Stability and Wald Inference under Variance-Aware UCB12 Dec 2024
- 2024UCB algorithms for multi-armed bandits: Precise regret and adaptive inference9 Dec 2024
12 newest of 36 All 36 (JSON)