arXiv:2409.04813cs.LGcs.AI2024-09被引 2

提出新方法让图卷积网络更精准地用滤波函数建模节点关系。

Generalized Learning of Coefficients in Spectral Graph Convolutional Networks

  • 基于阿诺尔迪正交化,高效逼近任意滤波函数为多项式形式。
  • 在10个不同拓扑数据集上,多分类任务性能超越现有最佳方法。
  • 适合想自定义滤波器设计的图学习研究者或工程应用开发者。

谱图卷积网络(GCN)因其传播规则可灵活设定而广受青睐。传统方法通过学习多项式滤波系数来实现传播,但这类方法难以捕捉图结构与标签分布间的关系。相比之下,显式滤波函数虽更具表达力,却面临求解多项式逼近时线性系统病态的问题。为此,本文提出一种基于阿诺尔迪正交化的新型算法,以及统一框架G-Arnoldi-GCN,能高效准确地将任意滤波函数近似为多项式。我们在10个具有不同拓扑特征的数据集上评估该方法,用于多分类节点预测。实验表明,在采用合适滤波函数时,G-Arnoldi-GCN始终优于当前最先进方法。该工作为图机器学习中显式设计和应用多样化滤波函数开辟了新方向。代码链接:https://github.com/mustafaCoskunAgu/GArnoldi-GCN

原文摘要 · Abstract (English)

Spectral Graph Convolutional Networks (GCNs) have gained popularity in graph machine learning applications due, in part, to their flexibility in specification of network propagation rules. These propagation rules are often constructed as polynomial filters whose coefficients are learned using label information during training. In contrast to learned polynomial filters, explicit filter functions are useful in capturing relationships between network topology and distribution of labels across the network. A number of algorithms incorporating either approach have been proposed; however the relationship between filter functions and polynomial approximations is not fully resolved. This is largely due to the ill-conditioned nature of the linear systems that must be solved to derive polynomial approximations of filter functions. To address this challenge, we propose a novel Arnoldi orthonormalization-based algorithm, along with a unifying approach, called G-Arnoldi-GCN that can efficiently and effectively approximate a given filter function with a polynomial. We evaluate G-Arnoldi-GCN in the context of multi-class node classification across ten datasets with diverse topological characteristics. Our experiments show that G-Arnoldi-GCN consistently outperforms state-of-the-art methods when suitable filter functions are employed. Overall, G-Arnoldi-GCN opens important new directions in graph machine learning by enabling the explicit design and application of diverse filter functions. Code link: https://github.com/mustafaCoskunAgu/GArnoldi-GCN

图神经网络谱图卷积滤波器设计

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