中文

GIPA:用于图学习的通用信息传播算法

机器学习 2021-08-11 v2

摘要

图神经网络(GNNs)已广泛用于分析图结构数据,在节点分类、链接预测与网络推荐等多种应用中展现出良好结果。本文提出一种新的图注意力神经网络,即 GIPA,用于属性图数据学习。GIPA 由三个关键组件构成:注意力、特征传播与聚合。具体而言,注意力组件引入了一种新的基于多层感知机的多头机制,以生成优于点积等传统实现的更好的非线性特征映射与表示。传播组件不仅考虑节点特征,还考虑边特征,这与现有仅考虑节点特征的 GNNs 不同。聚合组件使用残差连接生成最终嵌入。我们使用 Open Graph Benchmark 蛋白质数据集(简称 ogbn-proteins)评估 GIPA 的性能。实验结果表明,GIPA 在预测精度上可击败最先进模型,例如 GIPA 取得了 0.8700±0.00100.8700\pm 0.0010 的平均测试 ROC-AUC,并优于 ogbn-proteins 排行榜中列出的所有先前方法。

关键词

引用

@article{arxiv.2105.06035,
  title  = {GIPA: General Information Propagation Algorithm for Graph Learning},
  author = {Qinkai Zheng and Houyi Li and Peng Zhang and Zhixiong Yang and Guowei Zhang and Xintan Zeng and Yongchao Liu},
  journal= {arXiv preprint arXiv:2105.06035},
  year   = {2021}
}

备注

5 pages, 2 figures, 2 tables; Technical report