2 个月前

结合标签传播和简单模型优于图神经网络

Qian Huang; Horace He; Abhay Singh; Ser-Nam Lim; Austin R. Benson
结合标签传播和简单模型优于图神经网络
摘要

图神经网络(GNNs)是当前学习图结构数据的主要技术。然而,对于GNNs在实际应用中为何成功以及是否为实现良好性能所必需,目前的理解相对较少。本文展示了在许多标准的半监督节点分类基准测试中,通过结合忽略图结构的浅层模型和两种简单的利用标签相关性的后处理步骤,可以超过或匹配最先进的GNNs的性能:(i) 一种“误差相关”方法,该方法将训练数据中的残差误差传播到测试数据以纠正错误;(ii) 一种“预测相关”方法,该方法平滑测试数据上的预测结果。我们将这一整体过程称为校正和平滑(Correct and Smooth, C&S),后处理步骤是通过对早期基于图的半监督学习方法中的标准标签传播技术进行简单修改来实现的。我们的方法在广泛的基准测试中超过了或几乎匹配了最先进的GNNs的性能,同时仅使用了少量参数且运行速度提高了几个数量级。例如,在OGB-Products数据集上,我们以137倍少的参数和超过100倍少的训练时间超过了已知的最佳GNN性能。我们的方法的表现突显了直接将标签信息融入学习算法(如传统技术所做的那样)可以轻松获得显著的性能提升。此外,我们还可以将这些技术应用于大型GNN模型中,提供适度的增益。我们在GitHub上提供了OGB结果的相关代码:https://github.com/Chillee/CorrectAndSmooth。

结合标签传播和简单模型优于图神经网络 | 最新论文 | HyperAI超神经