mlsanityData to 7 Oct 2026

Budget-Constrained Causal Bandits: Bridging Uplift Modeling and Sequential Decision-Making

arXiv 2604.26169 ↗First posted 28 Apr 2026cs.LG econ.EM stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Treatment allocation under budget constraints is a central challenge in digital advertising. The standard approach trains an offline uplift model on historical data, then solves a constrained optimization to allocate budget. This fails in cold-start settings where little historical data exists. We propose Budget-Constrained Causal Bandits (BCCB), an online framework that learns which users respond to ads while simultaneously spending the budget. BCCB unifies three components: learning individual-level treatment effects, exploring users whose response is uncertain, and pacing the budget over time. We derive the per-arrival decision rule as the KKT condition of a Lagrangian relaxation of the budgeted causal-allocation objective, providing a principled foundation for the algorithm. We evaluate on the Criteo Uplift dataset using 20 random seeds with paired statistical tests. Our central finding is a data-efficiency crossover at n = 7,500 historical observations (paired one-sided t-test, p = 0.043): below this threshold, offline pipelines either fail or produce unreliable allocations, while BCCB operates from the first user. BCCB exhibits 2-4x lower run-to-run variance than offline methods and outperforms all four online baselines (Thompson Sampling, budgeted Thompson Sampling, HTE Greedy, and Uplifting Bandits) at every budget level tested (p < 0.001). These results give practitioners a concrete decision rule for choosing between offline and online paradigms.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 21 of its 36 references that are in the map. First posted on arXiv 28 Apr 2026.

Builds on

21 of 36 references.

References · 21

  1. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
  2. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
  3. 2017A Tutorial on Thompson Samplingcited by 420
  4. 2017Smart "Predict, then Optimize"cited by 332
  5. 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 254
  6. 2019Uplift Modeling for Multiple Treatments with Cost Optimizationcited by 25
  7. 2015Thompson Sampling for Budgeted Multi-armed Banditscited by 23
  8. 2021E-Commerce Promotions Personalization via Online Multiple-Choice Knapsack with Uplift Modelingcited by 21
  9. 2022Direct Heterogeneous Causal Learning for Resource Allocation Problems in Marketingcited by 21
  10. 2020Free Lunch! Retrospective Uplift Modeling for Dynamic Promotions Recommendation within ROI Constraintscited by 20
  11. 2018Contextual Multi-Armed Bandits for Causal Marketingcited by 17
  12. 2022LBCF: A Large-Scale Budget-Constrained Causal Forest Algorithmcited by 17
9 more
  1. 2024End-to-End Cost-Effective Incentive Recommendation under Budget Constraint with Uplift Modelingcited by 10
  2. 2020Multi-armed Bandits with Cost Subsidycited by 7
  3. 2024Benchmarking for Deep Uplift Modeling in Online Marketingcited by 3
  4. 2024DISCO: An End-to-End Bandit Framework for Personalised Discount Allocationcited by 3
  5. 2026DARA: Few-shot Budget Allocation in Online Advertising via In-Context Decision Making with RL-Finetuned LLMscited by 3
  6. 2022Uplifting Banditscited by 2
  7. 2025Optimizing Online Advertising with Multi-Armed Bandits: Mitigating the Cold Start Problem under Auction Dynamicscited by 2
  8. 2025Bi-Level Decision-Focused Causal Learning for Large-Scale Marketing Optimization: Bridging Observational and Experimental Datacited by 2
  9. 2024Improving Thompson Sampling via Information Relaxation for Budgeted Multi-armed Banditscited by 1

Built on by

Not cited yet.