mlsanityData to 14 Oct 2026

Co-Supervised Tree Synthesis for Interpretable Subgroup Identification and Honest Treatment Effect Inference in Randomized Trials

arXiv 2610.03927 ↗First posted 2 Oct 2026stat.ME stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Identifying patient subgroups with heterogeneous treatment effects is central to precision medicine, yet existing approaches face a tension. Black-box methods estimate individualized effects well but yield no interpretable subgroups, while tree-based methods produce explicit partitions but are unstable and less accurate in moderate-sample trials. We propose CausalSynthTree, a co-supervised method in which black-box causal estimators guide construction of an interpretable tree. It partitions the covariate space into cells, fits cell-wise conditional average treatment effect models from both observed data and synthetic data labeled by black-box causal teachers, and synthesizes them into a tree through a treatment-effect disparity criterion. Each leaf carries a sparse linear model, so the tree path defines a subgroup and the leaf model shows which covariates modify the effect. Honest inference for leaf-wise effects remains valid despite co-supervised augmentation and data-driven subgroup selection. In simulations it closes much of the gap between the two families. From moderate sample sizes onward it is more accurate than the black-box learners whenever the effect is linear or has a single threshold, and trails them only on a finer partition than the tree it grows. It controls spurious splits and attains nominal coverage throughout. In the ACTG175 HIV trial, where standard interaction tests detect no effect modification, the method reports no subgroups while the competing tree methods report several. In an adjuvant colon cancer trial it reports a borderline covariate in part of the resamples, together with a within-region age gradient that is consistent with an independent test.

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

Builds on

13 of 47 references.

References · 13

  1. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
  2. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
  3. 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 332
  4. 2016Generalized Random Forestscited by 312
  5. 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 255
  6. 2013Confidence Intervals and Hypothesis Testing for High-Dimensional Regressioncited by 79
  7. 2016Model-based Recursive Partitioning for Subgroup Analysescited by 22
  8. 2015A concave pairwise fusion approach to subgroup analysiscited by 16
  9. 2021Mixture of Linear Models Co-supervised by Deep Neural Networkscited by 5
  10. 2018A Semiparametric Approach to Model Effect Modificationcited by 4
  11. 2023Modern approaches for evaluating treatment effect heterogeneity from clinical trials and observational datacited by 4
  12. 2025Distilling heterogeneous treatment effects: Stable subgroup estimation in causal inferencecited by 3
1 more
  1. 2024SynthTree: Co-supervised Local Model Synthesis for Explainable Predictioncited by 2

Built on by

Not cited yet.