May, 2023

大型图中节点分类的带邻域增强的标记图变换器

TL;DR本研究提出了一种称为邻域聚合图变换器(NAGphormer)的方法,用于处理大规模图数据,它将每个节点视为一个序列,每个序列包含一系列通过 Hop2Token 模块构建的标记。NAGphormer 通过从不同跳的邻域特征进行聚合,生成一系列标记向量作为输入,能够以 mini-batch 的方式进行训练,并能够比图神经网络更好地学习多跳邻域的节点表示。研究同时提出了一种名为 Neighborhood Augmentation 的新型数据增强方法,以同时增强全局邻域和局部邻域的特征,进一步提高 NAGphormer 的训练效果。实验证明,NAGphormer 优于现有的图变换器和主流图神经网络,并且 NrAug 对 NAGphormer 进一步提升作用显著。