Aug, 2023

Transformer 和 GNN 能互助吗?

TL;DR在这篇论文中,我们提出了一种名为 TransGNN 的新模型,其中 Transformer 层和 GNN 层交替使用以相互改进。我们利用 Transformer 来聚合更相关的节点信息以改善 GNNs 的消息传递,并利用位置编码和 GNN 层将图结构融入节点属性中,改善 Transformer 在图数据中的表现。通过理论证明,我们证明了 TransGNN 相对于仅具有额外线性复杂度的 GNNs 更具表达力。在八个数据集上的实验证实了 TransGNN 在节点和图分类任务上的有效性。