Causal Graph Transformer for Treatment Effect Estimation Under Unknown Interference
Anpeng Wu, Haiyi Qiu, Zhengming Chen, Zijian Li, Ruoxuan Xiong, Fei Wu, Kun Zhang
Abstract
Networked interference, also known as the peer effect in social science and spillover effect in economics, has drawn increasing interest across various domains. This phenomenon arises when a unit's treatment and outcome are influenced by the actions of its peers, posing significant challenges to causal inference, particularly in treatment assignment and effect estimation in real applications, due to the violation of the SUTVA assumption. While extensive graph models have been developed to identify treatment effects, these models often rely on structural assumptions about networked interference, assuming it to be identical to the social network, which can lead to misspecification issues in real applications. To address these challenges, we propose an Interference-Agnostic Causal Graph Transformer (CauGramer), which aggregates peers information via L-order Graph Transformer and employs cross-attention to infer aggregation function for learning interference representations. By integrating confounder balancing and minimax moment constraints, CauGramer fully incorporates peer information, enabling robust treatment effect estimation. Extensive experiments on two widely-used benchmarks demonstrate the effectiveness and superiority of CauGramer. The code is available at https://github.com/anpwu/CauGramer .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 92e71ae8-3011-40a2-9207-6dadfce31994Cited by top-tier papers3
- Journey to the Centre of Cluster: Harnessing Interior Nodes for A/B Testing under Network InterferenceQianyi Chen, Anpeng Wu, Bo Li, Lu Deng et al.ICLR 2026 · 1 citation
- Learning Exposure Mapping Functions for Inferring Heterogeneous Peer EffectsShishir Adhikari, Sourav Medya, Elena ZhelevaICLR 2026 · 1 citation
- Treatment Effect Estimation with Differentiated Networked Effect on Graph DataXiaofeng Lin, Han Bao, Hisashi KashimaKDD 2026
Builds on7
- Do Transformers Really Perform Badly for Graph Representation?Chengxuan Ying, Tianle Cai, Shengjie Luo, Shuxin Zheng et al.NeurIPS 2021 · 1,632 citations
- Recipe for a General, Powerful, Scalable Graph TransformerLadislav Rampásek, Michael Galkin, Vijay Prakash Dwivedi, Anh Tuan Luu et al.NeurIPS 2022 · 1,216 citations
- Structure-Aware Transformer for Graph Representation LearningDexiong Chen, Leslie O'Bray, Karsten M. BorgwardtICML 2022 · 349 citations
- Learning Disentangled Representations for CounterFactual RegressionNegar Hassanpour, Russell GreinerICLR 2020 · 176 citations
- Causal Transformer for Estimating Counterfactual OutcomesValentyn Melnychuk, Dennis Frauen, Stefan FeuerriegelICML 2022 · 146 citations
Related papers
- Modeling Interference for Treatment Effect Estimation in Network Dynamic EnvironmentQiang Huang, Jin TianICLR 2026
- Higher-Order Causal Message Passing for Experimentation with Complex InterferenceMohsen Bayati, Yuwei Luo, William Overman, Mohamad Sadegh Shirani Faradonbeh et al.NeurIPS 2024 · 9 citations
- Telling Peer Direct Effects from Indirect Effects in Observational Network DataXiaojing Du, Jiuyong Li, Debo Cheng, Lin Liu et al.ICML 2025
- Staggered Rollout Designs Enable Causal Inference Under Interference Without Network KnowledgeMayleen Cortez, Matthew Eichhorn, Christina Lee YuNeurIPS 2022 · 29 citations
- Invariant Graph Learning for Causal Effect EstimationYongduo Sui, Caizhi Tang, Zhixuan Chu, Junfeng Fang et al.WWW 2024 · 17 citations
