arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.09613cs.LGcs.AIstat.ML

重新思考图神经网络中的因果关系学习

Rethinking Causal Relationships Learning in Graph Neural Networks

Science and Technology on Integrated Information System Laboratory, Institute of Software Chinese Academy of Sciences(中国科学院软件研究所综合信息系统技术国家级重点实验室);University of Chinese Academy of Sciences(中国科学院大学);State Key Laboratory of Intelligent Game(智能博弈国家重点实验室);Department of Computer Science and Technology, Tsinghua University(清华大学计算机科学与技术系)

Hang Gao, Chengyu Yao, Jiangmeng Li, Lingyu Si, Yifan Jin, Fengge Wu, Changwen Zheng, Huaping Liu

AI总结:

针对图神经网络因果建模能力缺乏深度分析的问题,构建可控因果关系的合成数据集,提出轻量高适配的GNN模块以增强其因果学习能力,经多数据集实验验证有效。

摘要:

图神经网络(GNNs)通过有效建模图结构数据中的复杂关联,展现出其重要价值。为提升GNNs的可信度与鲁棒性,增强其捕捉因果关系的能力变得至关重要。然而,尽管近期研究已从因果学习视角强化了GNNs,但针对GNNs因果建模能力的深度分析仍是一个未解决的问题。为从因果学习视角全面分析各类GNN模型,我们构建了一个人工合成数据集,其中数据与标签间的因果关系已知且可控,理论基础进一步保障了生成数据的合理性。基于使用该数据集开展分析得到的洞见,我们提出了一个轻量级且高适配性的GNN模块,旨在增强GNNs在各类任务中的因果学习能力。通过在合成数据集与其他真实世界数据集上开展的一系列实验,我们通过实验验证了所提模块的有效性。

英文摘要:

Graph Neural Networks (GNNs) demonstrate their significance by effectively modeling complex interrelationships within graph-structured data. To enhance the credibility and robustness of GNNs, it becomes exceptionally crucial to bolster their ability to capture causal relationships. However, despite recent advancements that have indeed strengthened GNNs from a causal learning perspective, conducting an in-depth analysis specifically targeting the causal modeling prowess of GNNs remains an unresolved issue. In order to comprehensively analyze various GNN models from a causal learning perspective, we constructed an artificially synthesized dataset with known and controllable causal relationships between data and labels. The rationality of the generated data is further ensured through theoretical foundations. Drawing insights from analyses conducted using our dataset, we introduce a lightweight and highly adaptable GNN module designed to strengthen GNNs' causal learning capabilities across a diverse range of tasks. Through a series of experiments conducted on both synthetic datasets and other real-world datasets, we empirically validate the effectiveness of the proposed module.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑