Budget-Constrained Causal Bandits: Bridging Uplift Modeling and Sequential Decision-Making
cs.LG, econ.EM, stat.ML
Submitted: 2026-04-28
Updated: 2026-08-28
Comments: 17 pages, 5 tables. v2: Expanded to 20 random seeds with paired statistical tests; formal crossover test at n=7,500 (p=0.043); added Uplifting Bandits (Hsieh et al., NeurIPS 2022) as fifth baseline; Lagrangian derivation of decision rule with Proposition 1; hyperparameter sensitivity analysis; corrected mathematical claims
License: http://creativecommons.org/licenses/by/4.0/
The gist: Treatment allocation under budget constraints is a central challenge in digital advertising.
Terminology
Abstract
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.
Sources
- Sparse superposition codes with rotational invariant coding matrices for memoryless channels
- Optimizing Online Advertising with Multi-Armed Bandits: Mitigating the Cold Start Problem under Auction Dynamics
- Leveraging Offline Data in Linear Latent Contextual Bandits
- THP: Topological Hawkes Processes for Learning Causal Structure on Event Sequences
- Landau levels in a gravitational field: The Schwarzschild spacetime case
- Contextual Multi-Armed Bandits for Causal Marketing
- Super-Macdonald polynomials: Orthogonality and Hilbert space interpretation
- Scalable spin squeezing from critical slowing down in short-range interacting systems
- LBCF: A Large-Scale Budget-Constrained Causal Forest Algorithm
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks