Statistical Inference on Multi-armed Bandits with Delayed Feedback
Lei Shi, Jingshen Wang, Tianhao Wu
Abstract
Multi armed bandit (MAB) algorithms have been increasingly used to complement or integrate with A/B tests and randomized clinical trials in e-commerce, healthcare, and policymaking. Recent developments incorporate possible delayed feedback. While existing MAB literature often focuses on maximizing the expected cumulative reward outcomes (or, equivalently, regret minimization), few efforts have been devoted to establish valid statistical inference approaches to quantify the uncertainty of learned policies. We attempt to fill this gap by providing a unified statistical inference framework for policy evaluation where a target policy is allowed to differ from the data collecting policy, and our framework allows delay to be associated with the treatment arms. We present an adaptively weighted estimator that on one hand incorporates the arm-dependent delaying mechanism to achieve consistency, and on the other hand mitigates the variance inflation across stages due to vanishing sampling probability. In particular, our estimator does not critically depend on the ability to estimate the unknown delay mechanism. Under appropriate conditions, we prove that our estimator converges to a normal distribution as the number of time points goes to infinity, which provides guarantees for large-sample statistical inference. We illustrate the finite-sample performance of our approach through Monte Carlo experiments.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext e7da8419-9022-4f67-9b07-7b86edca4669Cited by top-tier papers3
- Delay as Payoff in MABOfir Schlisselberg, Ido Cohen, Tal Lancewicki, Yishay MansourAAAI 2025 · 5 citations
- Using Surrogates in Covariate-adjusted Response-adaptive Randomization Experiments with Delayed OutcomesLei Shi, Waverly Wei, Jingshen WangNeurIPS 2024 · 4 citations
- Leveraging semantic similarity for experimentation with AI-generated treatmentsLei Shi, David Arbour, Raghavendra Addanki, Ritwik Sinha et al.NeurIPS 2025 · 1 citation
Builds on10
- Linear bandits with Stochastic Delayed FeedbackClaire Vernade, Alexandra Carpentier, Tor Lattimore, Giovanni Zappella et al.ICML 2020 · 74 citations
- Post-Contextual-Bandit InferenceAurélien Bibaut, Maria Dimakopoulou, Nathan Kallus, Antoine Chambaz et al.NeurIPS 2021 · 58 citations
- Stochastic bandits with arm-dependent delaysAnne Gael Manegueu, Claire Vernade, Alexandra Carpentier, Michal ValkoICML 2020 · 49 citations
- Online Multi-Armed Bandits with Adaptive InferenceMaria Dimakopoulou, Zhimei Ren, Zhengyuan ZhouNeurIPS 2021 · 47 citations
- Stochastic Multi-Armed Bandits with Unrestricted Delay DistributionsTal Lancewicki, Shahar Segal, Tomer Koren, Yishay MansourICML 2021 · 45 citations
Related papers
- Adaptive Experimentation with Delayed Binary FeedbackZenan Wang, Carlos Carrion, Xiliang Lin, Fuhua Ji et al.WWW 2022 · 8 citations
- Mutli-Armed Bandits with Network InterferenceAbhineet Agarwal, Anish Agarwal, Lorenzo Masoero, Justin WhitehouseNeurIPS 2024 · 4 citations
- Design-Based Anytime-Valid Inference for Randomized Experiments with Delayed Outcomes and Staggered EntryMichael Lindon, Nathan KallusICML 2026 · 1 citation
- Off-Policy Evaluation via Adaptive Weighting with Data from Contextual BanditsRuohan Zhan, Vitor Hadad, David A. Hirshberg, Susan AtheyKDD 2021 · 22 citations
- A New Framework: Short-Term and Long-Term Returns in Stochastic Multi-Armed BanditAbdalaziz Sawwan, Jie WuINFOCOM 2023 · 11 citations
