LLM2D
同质图上可扩展且表达能力强的图神经网络的图采样
Graph Sampling for Scalable and Expressive Graph Neural Networks on Homophilic Graphs
作者: Haolin Li, Luana Ruiz
发布日期: 10/23/2024
arXiv ID: oai:arXiv.org:2410.16593v1

摘要

图神经网络 (GNN) 在许多图机器学习任务中表现出色,但在扩展到大型网络时面临挑战。GNN的可迁移性允许在较小的图上进行训练并将模型应用于较大的图,但现有方法通常依赖于随机子采样,这会导致子图断开连接并降低模型的表达能力。我们提出了一种新颖的图采样算法,该算法利用特征同质性来保留图结构。通过最小化数据相关矩阵的迹,我们的方法比随机采样更好地保留了图拉普拉斯算子的秩,同时实现了比谱方法更低的复杂度。在引文网络上的实验表明,与随机采样相比,该方法在保留图秩和GNN可迁移性方面具有更好的性能。