mlsanityData to 14 Oct 2026

Significance-First Splitting: Aligning Treatment Heterogeneity Detection with Honest Estimation

arXiv 2607.03999 ↗First posted 4 Jul 2026stat.ME cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Estimating heterogeneous treatment effects (CATE) requires simultaneously detecting effect modification and quantifying estimation uncertainty. Existing tree-based methods make an uneasy trade-off: significance-based approaches (Radcliffe and Surry 2011) identify subgroup interactions directly but lack valid inference; honest causal trees (Athey and Imbens 2016) deliver nominal confidence interval coverage but use outcome-agnostic splitting criteria that sacrifice interaction sensitivity. We introduce a hybrid algorithm that fuses significance-based splitting with honest sample-splitting and cross-validation. Our splitting criterion uses the squared t-statistic for the treatment × side interaction (t2), which is shown to be directly aligned with the honest EMSEτ criterion when the interaction is strong. Post-hoc honest cross-validation selects the cost-complexity penalty, giving a single principled estimator with nominal CI coverage at the leaf level. For forests, we retain bootstrap count vectors to enable an infinitesimal jackknife (IJ) variance estimate of Monte-Carlo convergence rather than formal pointwise inference. On the three synthetic designs from (Athey and Imbens 2016) the single tree achieves approximately 90% leaf-average CI coverage at the 90% nominal level across all three designs (200 replications each); on the Criteo, Hillstrom and Starbucks uplift datasets we match Qini coefficient performance of S-, T-learner and GRF baselines. An open-source Python package with reproducible seeds, sklearn-compatible API, and full test coverage accompanies this work (https://codeberg.org/hadjipantelis/rattus).

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 2 of its 10 references that are in the map. First posted on arXiv 4 Jul 2026.

Builds on

2 of 10 references.

References · 2

  1. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
  2. 2014A regression tree approach to identifying subgroups with differential treatment effectscited by 6

Built on by

Not cited yet.