基于自监督多尺度邻域预测的节点特征提取
机器学习
2022-03-14 v3
摘要
由于众多现实应用,图上的学习在学习界引起了显著关注。特别是,以数值节点特征和图结构为输入的图神经网络(GNNs)已被证明在各种图相关学习任务上达到最先进的性能。近期通过自监督学习探索数值节点特征与图结构之间关联的工作,为 GNNs 的进一步提升铺平了道路。然而,在标准 GNN 流程中,用于从原始数据提取数值节点特征的方法仍是图不可知的。这种做法次优,因为它阻碍了对图拓扑与节点属性之间潜在关联的充分利用。为缓解此问题,我们提出一个新的自监督学习框架——图信息辅助节点特征提取(GIANT)。GIANT 利用极端多标签分类(XMC)形式化方法,这对于基于图信息微调语言模型至关重要,并可扩展到大型数据集。我们还提供理论分析,证明使用 XMC 而非链接预测的合理性,并促使将用于求解 XMC 问题的强大方法 XR-Transformers 集成到 GIANT 框架中。我们在 Open Graph Benchmark 数据集上展示了 GIANT 相对于标准 GNN 流程的优越性能:例如,在 ogbn-papers100M 数据集上,通过利用 GIANT,我们将排名靠前的方法 GAMLP 的准确率从 提升至 ,SGC 从 提升至 ,MLP 从 提升至 。
引用
@article{arxiv.2111.00064,
title = {Node Feature Extraction by Self-Supervised Multi-scale Neighborhood Prediction},
author = {Eli Chien and Wei-Cheng Chang and Cho-Jui Hsieh and Hsiang-Fu Yu and Jiong Zhang and Olgica Milenkovic and Inderjit S Dhillon},
journal= {arXiv preprint arXiv:2111.00064},
year = {2022}
}
备注
Published in ICLR 2022