mlsanityData to 7 Oct 2026

The Final-Stage Bottleneck: A Systematic Dissection of the R-Learner for Network Causal Inference

arXiv 2511.13018 ↗First posted 17 Nov 2025cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

The R-Learner is a powerful, theoretically-grounded framework for estimating heterogeneous treatment effects, prized for its robustness to nuisance model errors. However, its application to network data, where causal heterogeneity is often graph-dependent, presents a critical challenge to its core assumption of a well-specified final-stage model. In this paper, we conduct a large-scale empirical study to systematically dissect the R-Learner framework on graphs. We provide the first rigorous evidence that the primary driver of performance is the inductive bias of the final-stage CATE estimator, an effect that dominates the choice of nuisance models. Our central finding is the quantification of a catastrophic "representation bottleneck": we prove with overwhelming statistical significance (p < 0.001) that R-Learners with a graph-blind final stage fail completely (MSE > 4.0), even when paired with powerful GNN nuisance models. Conversely, our proposed end-to-end Graph R-Learner succeeds and significantly outperforms a strong, non-DML GNN T-Learner baseline. Furthermore, we identify and provide a mechanistic explanation for a subtle, topology-dependent "nuisance bottleneck," linking it to GNN over-squashing via a targeted "Hub-Periphery Trade-off" analysis. Our findings are validated across diverse synthetic and semi-synthetic benchmarks. We release our code as a reproducible benchmark to facilitate future research on this critical "final-stage bottleneck."

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 11 of its 19 references that are in the map. First posted on arXiv 17 Nov 2025.

Builds on

11 of 19 references.

References · 11

  1. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
  2. 2017Graph Attention Networkscited by 8,854
  3. 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 970
  4. 2020On the Bottleneck of Graph Neural Networks and its Practical Implicationscited by 624
  5. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
  6. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
  7. 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
  8. 2013Robust Inference on Average Treatment Effects with Possibly More Covariates than Observationscited by 38
  9. 2022Learning Causal Effects on Hypergraphscited by 31
  10. 2024Graph Machine Learning based Doubly Robust Estimator for Network Causal Effectscited by 4
  11. 2024Uplift Modeling Under Limited Supervisioncited by 3

Built on by

Not cited yet.