跳到正文
arXiv cs.AI· Hendrik Stumpf, Francesco Quinzan·· 6 天前AI 评分22

CFGNP:无需因果图估计的摊销反事实预测

Amortized Counterfactual Prediction with Graph Neural Networks

AI 导读

研究者提出 Counterfactual Graph Neural Process(CFGNP),一种端到端方法,可直接从观测数据回答反事实查询,无需先估计因果图。

正文 · 原文

View PDF HTML (experimental)

Abstract:Counterfactual inference asks how an observed outcome might have differed under different circumstances, such as whether a patient would have recovered had they received a different treatment. Such questions arise whenever decisions depend on the outcomes of alternatives that were not observed. Answering counterfactual queries generally requires knowledge of the underlying causal structure, which is often unavailable in practice. A common approach is therefore to first estimate the causal graph from observational data and then perform counterfactual inference. However, errors in the estimated causal graph can propagate to downstream counterfactual predictions. To address this challenge, we introduce the Counterfactual Graph Neural Process (CFGNP), an end-to-end approach that answers counterfactual queries directly from observational data without first estimating a causal graph. CFGNP uses Graph Neural Networks (GNNs) to exploit the dual-graph structure by restricting cross-world message passing to corresponding factual--counterfactual variables. Across synthetic and image-based benchmarks, CFGNP achieves stronger predictive performance than existing approaches, including on the German Loan benchmark, where it reduces MSE from $4.51 \times 10^{-2}$ to $0.19 \times 10^{-2}$. Furthermore, on synthetic causal graphs, a model trained only on 10-node graphs maintains strong predictive performance on graphs with up to 30 nodes, without retraining.
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)
Cite as: arXiv:2610.13724 [cs.LG]
  (or arXiv:2610.13724v1 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2610.13724

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Hendrik Stumpf [view email]
[v1] Fri, 9 Oct 2026 15:22:18 UTC (5,106 KB)

来源:arXiv cs.AI · arxiv.org