arXiv:2611.08990v1 Announce Type: new
Abstract: Web platforms select among costly interventions for each user to maximize incremental outcomes under a global budget. We formulate this problem as a budget-constrained contextual bandit with heterogeneous treatment effects. Although an action-independent outcome baseline cancels from treatment comparisons, modeling a nonlinear baseline can increase statistical complexity.
We propose DR-CATE-CBwK, an action-centered causal bandit that directly learns treatment contrasts. Its online update remains valid under baseline-model misspecification without inverse-propensity weighting. It combines an offline doubly robust learner to warm-start treatment-effect estimates, an online action-centered estimator for sequential contrast learning, and a primal-dual controller with an affordable-action rule to enforce the effective-cost budget pathwise.
For linear conditional average treatment effects (CATEs) in a stationary i.i.d. setting, we prove a high-probability (\widetilde{O}(Kd\sqrt{T})) regret bound, where K is the number of priced actions, d the context dimension, and T the horizon. We benchmark against the fluid linear program (LP) under the coverage-guaranteed action rule. We further show that the action-centered estimating moment remains unbiased under any predictable baseline estimate, whereas a specified separate-arm reward regression can exhibit asymptotic contrast bias under action-dependent logging and shared-baseline misspecification.
On a semi-synthetic job-marketplace benchmark with real logged features and simulated outcomes, our practical Thompson-restricted variant achieves the lowest regret among evaluated budget-feasible methods in the operating regime. It reduces regret by approximately 20% versus the strongest reward-based baseline, outperforming raw-reward contextual bandits, BwK baselines, and a static offline policy, while never exceeding the effective-cost budget.
Trending
- Reasoning to Model and Solve General Optimization Problems with Limited Resources
- Fine-Tune MMS Adapter Models for low-resource ASR
- An AI-Driven Decision Framework for Skin Production Planning via Multi-Modal Transformer Sales Forecasting
- Response to the U.S. NTIA’s Request for Comment on AI Accountability
- Advancing science & engineering with Antigravity’s /teamwork
- Panel on Hugging Face
- [2605.26078] Global Convergence of Wasserstein Policy Gradient for Entropy-Regularized Reinforcement Learning
- What’s going on with the Open LLM Leaderboard?
