Graph Neural Networks for Causal Inference Under Network Confounding

📅 2022-11-15
📈 Citations: 4
Influential: 1
📄 PDF

career value

205K/year
🤖 AI Summary
This paper addresses causal inference from a single large-scale network observation, confronting the high-dimensional network confounding challenge arising from simultaneous network interference and treatment selection endogeneity. To overcome the limitation of conventional methods—which rely on low-dimensional summaries of confounders—we propose two key innovations: (i) the first integration of graph neural networks (GNNs) into a causal inference framework, enabling end-to-end modeling of high-dimensional network confounding; and (ii) a “network-distance-decaying interference” assumption, imposing a low-dimensional structural constraint that ensures theoretical interpretability of shallow GNNs. Combining nonparametric causal modeling with simultaneous equations estimation, our approach delivers consistent estimation of the local average treatment effect under endogenous peer effects. Extensive experiments on synthetic and real-world network data demonstrate superior confounding control and estimation accuracy compared to existing methods.
📝 Abstract
This paper studies causal inference with observational data from a single large network. We consider a nonparametric model with interference in potential outcomes and selection into treatment. Both stages may be the outcomes of simultaneous equation models, which allow for endogenous peer effects. This results in high-dimensional network confounding where the network and covariates of all units constitute sources of selection bias. In contrast, the existing literature assumes that confounding can be summarized by a known, low-dimensional function of these objects. We propose to use graph neural networks (GNNs) to adjust for network confounding. When interference decays with network distance, we argue that the model has low-dimensional structure that makes estimation feasible and justifies the use of shallow GNN architectures.
Problem

Research questions and friction points this paper is trying to address.

Addressing high-dimensional network confounding in causal inference
Modeling interference and treatment selection in networked data
Using GNNs to adjust for bias from network and covariates
Innovation

Methods, ideas, or system contributions that make the work stand out.

Uses Graph Neural Networks for network confounding
Adjusts for high-dimensional network confounding
Employs shallow GNN architectures for feasibility
🔎 Similar Papers
No similar papers found.