arXiv:2411.14094cs.LG2024-11被引 3

针对多标签节点分类中GNN表现不佳的问题,提出融合特征、标签和位置信息的新方法。

GNN-MultiFix: Addressing the pitfalls for GNNs for multi-label node classification

  • 通过融合节点特征、标签和位置信息,提升GNN在多标签场景下的学习能力。
  • 在多个多标签数据集上,新方法显著优于现有GNN模型,尤其在无属性节点时仍有效。
  • 适用于多标签节点分类任务,尤其适合缺乏节点属性或标签信息不完整的场景。

图神经网络(GNN)在图数据表示学习中表现出强大性能,广泛应用于各类任务并取得领先结果。然而,其优越性通常基于对小规模基准数据集的评估,或通过图同构测试分析表达能力。本文在归纳设置下对节点分类任务进行深入分析,重点关注被多数研究忽视的多标签节点分类场景。通过分析GNN训练动态,发现即使在充足训练数据下,传统GNN仍难以在多标签图数据上有效学习。进一步表明,在归纳设置中,即使最富表达力的GNN也可能因缺少节点属性且未显式输入标签信息而失败。为此,我们提出GNN-MultiFix方法,整合节点的特征、标签及位置信息。实验显示,该方法在所有多标签数据集上均有显著提升。代码已公开于https://anonymous.4open.science/r/Graph-MultiFix-4121。

原文摘要 · Abstract (English)

Graph neural networks (GNNs) have emerged as powerful models for learning representations of graph data showing state of the art results in various tasks. Nevertheless, the superiority of these methods is usually supported by either evaluating their performance on small subset of benchmark datasets or by reasoning about their expressive power in terms of certain graph isomorphism tests. In this paper we critically analyse both these aspects through a transductive setting for the task of node classification. First, we delve deeper into the case of multi-label node classification which offers a more realistic scenario and has been ignored in most of the related works. Through analysing the training dynamics for GNN methods we highlight the failure of GNNs to learn over multi-label graph datasets even for the case of abundant training data. Second, we show that specifically for transductive node classification, even the most expressive GNN may fail to learn in absence of node attributes and without using explicit label information as input. To overcome this deficit, we propose a straightforward approach, referred to as GNN-MultiFix, that integrates the feature, label, and positional information of a node. GNN-MultiFix demonstrates significant improvement across all the multi-label datasets. We release our code at https://anonymous.4open.science/r/Graph-MultiFix-4121.

图神经网络多标签分类节点分类GNN改进

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。