The Final-Stage Bottleneck: A Systematic Dissection of the R-Learner for Network Causal Inference
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
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 970
- 2020On the Bottleneck of Graph Neural Networks and its Practical Implicationscited by 624
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
- 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
- 2017Quasi-Oracle Estimation of Heterogeneous Treatment Effectscited by 328
- 2013Robust Inference on Average Treatment Effects with Possibly More Covariates than Observationscited by 38
- 2022Learning Causal Effects on Hypergraphscited by 31
- 2024Graph Machine Learning based Doubly Robust Estimator for Network Causal Effectscited by 4
- 2024Uplift Modeling Under Limited Supervisioncited by 3
Built on by
Not cited yet.