Estimating Conditional Average Treatment Effects (CATE) at the individual level is central to precision marketing, yet systematic benchmarking of uplift modeling methods at industrial scale remains limited. We present UpliftBench, an empirical evaluation of four CATE estimators: S-Learner, T-Learner, X-Learner (all with LightGBM base learners), and Causal Forest (EconML), applied to the Criteo Uplift v2.1 dataset comprising 13.98 million customer records. The near-random treatment assignment (propensity AUC = 0.509) provides strong internal validity for causal estimation. Evaluated via Qini coefficient and cumulative gain curves, the S-Learner achieves the highest Qini score of 0.376, with the top 20% of customers ranked by predicted CATE capturing 77.7% of all incremental conversions, a 3.9x improvement over random targeting. SHAP analysis identifies f8 as the dominant heterogeneous treatment effect (HTE) driver among the 12 anonymized covariates. Causal Forest uncertainty quantification reveals that 1.9% of customers are confident persuadables (lower 95% CI > 0) and 0.1% are confident sleeping dogs (upper 95% CI < 0). Our results provide practitioners with evidence-based guidance on method selection for large-scale uplift modeling pipelines.
Figures & tables
Property
Value
Records
13,979,592
Features
12 covariates (f0–f11)
Treatment
85% treated / 15% control
Primary Outcome
visit (4.7% positive rate)
Secondary Outcome
conversion (0.29% rate)
Propensity AUC
0.509
Table 1: Criteo Uplift v2.1 Dataset Summary
Figure 1: Qini curves for all four CATE estimators. S-Learner achieves the highest cumulative gain across all targeting thresholds.
Model
CATE Mean
CATE Std
Qini
Top 20%
Top 50%
S-Learner
0.0070
0.0223
0.3759
77.7%
95.8%
X-Learner
0.0074
0.0248
0.3615
76.0%
92.8%
T-Learner
0.0074
0.0267
0.3503
75.1%
92.4%
Causal Forest
0.0072
0.0377
0.2524
–
–
Random Baseline
–
–
≈ 0.007
20.0%
50.0%
Table 2: Model Performance on Criteo Uplift v2.1
Figure 2: CATE distributions for S-Learner (left) and T-Learner (right). S-Learner exhibits significantly lower variance, consistent with its implicit regularization toward zero treatment effect.
Strategy
Contacts
Incremental Conversions Captured
Efficiency vs. Random
Untargeted (100%)
1,000,000
100%
1.0 ×
Top 20% S-Learner
200,000
77.7%
3.9 ×
Random 20%
200,000
20.0%
1.0 ×
Table 3: Policy Simulation: Targeting Efficiency
Figure 3: Policy simulation: incremental conversions captured vs. fraction of population contacted. S-Learner captures 77.7% of all incremental conversions at a 20% contact rate.
Segment
Criterion
Share
Confident persuadables
Lower CI >0
1.9%
Confident sleeping dogs
Upper CI <0
0.1%
Uncertain
CI overlaps 0
98.0%
Table 4: Causal Forest Customer Segmentation
Figure 4: Causal Forest 95% CI distribution on the ≈ 280K evaluation subsample. Left: distribution of CI widths. Right: CATE point estimates vs. uncertainty. Most customers have uncertain effect estimates; 1.9% are confident persuadables.
Figure 5: Top: Mean absolute SHAP values for T-Learner CATE predictions; f8, f6, and f2 dominate. Bottom: SHAP beeswarm plot; each point is one customer, color indicates feature value magnitude.
Uplift modeling (conditional-average-treatment-effect estimation) drives personalized targeting, yet published uplift benchmarks frequently disagree on which estimator performs best; we show the disagreement is substantially about metrics, not models. UpliftBench evaluates 12 uplift estimators under an outer-test-isolated, multi-objective protocol across seven dataset families; its two findings are identified where a reference objective exists -- F1 on the standard continuous benchmark (IHDP), F2 in a within-sample case study on Jobs. On that benchmark, Qini shows no detectable alignment with effect accuracy -- across all 100 IHDP realizations its mean rank correlation with effect accuracy is +0.07, 95% CI [-0.03, +0.16] -- while AUUC is consistently more aligned (paired prefix-mean-AUUC-over-Qini gap +0.49 [+0.40, +0.59]; the shipped cumulative-gain AUUC aligns better still, +0.73). On Jobs, ranking metrics are structurally insufficient for a sign-threshold policy because they discard the score level; empirically, within the released split-rotation analysis direct policy-risk selection yields lower benchmark regret than random model selection while Qini, AUUC, and uplift-at-k do not (14-15% regret). Calibrating the decision threshold removes 81% of the Qini-selection regret. Both findings are bounded, not universal: F1 is not detected on either validation family (the ACIC and Revenue-Synthetic gaps are both indistinguishable from zero), and F2 vanishes under a budgeted-value objective where rank suffices. UpliftBench releases versioned loaders, fixed protocols, result artifacts, and a reproducible living leaderboard; the public repository accompanies the paper.
We study uplift estimation for combinatorial treatments. Uplift measures the pure incremental causal effect of an intervention (e.g., sending a coupon or a marketing message) on user behavior, modeled as a conditional individual treatment effect. Many real-world interventions are combinatorial: a treatment is a policy that specifies context-dependent action distributions rather than a single atomic label. Although recent work considers structured treatments, most methods rely on categorical or opaque encodings, limiting robustness and generalization to rare or newly deployed policies. We propose an uplift estimation framework that aligns treatment representation with causal semantics. Each policy is represented by the mixture it induces over contextaction components and embedded via a permutation-invariant aggregation. This representation is integrated into an orthogonalized low-rank uplift model, extending Robinson-style decompositions to learned, vector-valued treatments. We show that the resulting estimator is expressive for policy-induced causal effects, orthogonally robust to nuisance estimation errors, and stable under small policy perturbations. Experiments on large-scale randomized platform data demonstrate improved uplift accuracy and stability in long-tailed policy regimes
Xinyan Su, Jiacan Gao, Mingyuan Ma +6
Didi Chuxing, Beijing, China · School of Statistics, East China Normal University, Shanghai, China · School of Mathematics and Statistics, Beijing Jiaotong University, Beijing, China +2
Conditional Average Treatment Effect (CATE) estimation in practice demands three properties simultaneously: heterogeneous effects τ(x), calibrated uncertainty over them, and robustness to the heavy tails that contaminate real outcome data. Meta-learners (Künzel et al., 2019) give (i); causal forests and BART give (i)-(ii) with Gaussian-tail assumptions; no widely used tool gives all three. We present Bayesian X-Learner, an X-Learner built on cross-fitted doubly robust pseudo-outcomes (Kennedy, 2020) with a full MCMC posterior over τ(x) via a Welsch redescending pseudo-likelihood. On Hill's IHDP benchmark the default configuration attains mean εPEHE=0.56 on 5 replications (lowest mean; differences from S-/T-/X-learners, full-config Causal BART, and a causal forest baseline are not significant at α=0.05, and rank ordering is unstable at 10 replications -- IHDP comparisons are competitive rather than dominant). On contaminated "whale" DGPs with up to 20-25% tail density, a one-flag extension (contamination_severity) that selects a Huber-δ nuisance loss per Huber's minimax-δ relation recovers RMSE ≈0.13 with tight credible intervals (single-cross-fit 30-seed coverage 83% [Wilson 66%, 93%] at 20% density; modular-Bayes pooling with Bayesian-bootstrap nuisance draws restores nominal 95% coverage).