LLM2D
基于因果消息传递的异嗜性图神经网络优化
Heterophilic Graph Neural Networks Optimization with Causal Message-passing
作者: Botao Wang, Jia Li, Heng Chang, Keli Zhang, Fugee Tsung
发布日期: 11/28/2024
arXiv ID: oai:arXiv.org:2411.13821v2

摘要

这项工作发现因果推理为捕捉图神经网络 (GNN) 中的异嗜性消息传递提供了一种有前景的方法。通过利用因果关系分析,我们可以根据节点间不对称的依赖关系识别异嗜性边。学习到的因果结构提供了节点之间更准确的关系。为了降低计算复杂度,我们在图学习中引入了基于干预的因果推理。我们首先通过将其公式化为结构学习模型来简化图上的因果分析,并在贝叶斯框架内定义优化问题。然后,我们提出了一种基于因果关系分解优化目标为一致性惩罚和结构修改的分析方法。然后,我们通过条件熵估计该目标,并深入探讨条件熵如何量化异嗜性。相应地,我们提出了 CausalMP,这是一种用于异嗜性图学习的因果消息传递发现网络,它迭代地学习输入图的显式因果结构。我们在异嗜性和同嗜性图设置中进行了大量的实验。结果表明,我们的模型实现了优越的链路预测性能。在不同基础模型的分类任务中,对因果结构的训练还可以增强节点表示。