Using Subgraph GNNs for Node Classification:an Overlooked Potential Approach
作者: Qian Zeng, Xin Lin, Jingyi Gao, Yang Yu
发布时间: 2025-03-11
来源: arxiv
研究方向: 图神经网络(GNN)在节点分类中的应用
主要内容
该研究提出了一种名为SubGND(Subgraph GNN for NoDe)的新型框架,将节点分类任务转化为子图分类问题,以解决传统GNN方法在可扩展性和分类精度之间的权衡问题。
主要贡献
1. 首次将节点分类任务完全从子图的角度进行探索,通过问题重构引入了一种新的方法来应用GNN于节点分类。
2. 识别了子图转换过程中常见的标签冲突问题,并提出了一种包含差异化零填充策略、独特的Ego-Alter子图表示方法和自适应特征缩放机制的子图框架。
3. 在多个公共数据集上的广泛实验表明,SubGND在异构图设置中显著优于现有方法,验证了其有效性和可扩展性。
研究方法
1. 使用ISRW算法生成节点诱导子图。
2. 采用差异化零填充策略来增强节点表示的可表达性。
3. 引入Ego-Alter子图表示方法来解决标签冲突问题。
4. 应用自适应特征缩放机制来动态调整特征贡献。
5. 使用GIN作为骨干网络进行节点表示学习。
6. 通过MaxPooling聚合节点表示以增强节点特征。
7. 使用Feedforward神经网络(FFNN)进行最终的预测。
实验结果
SubGND在异构图设置中显著优于现有方法,尤其是在Chameleon和Squirrel数据集上。在同构图设置中,SubGND的性能与全局消息传递GNN相当,甚至略优于一些模型。
未来工作
进一步研究SubGND在更大数据集上的性能,并探索其在其他图学习任务中的应用。