arXiv:2411.12330cs.LG2024-11被引 1

用非神经网络方法融合图结构与特征,显著提升节点分类性能。

Graph as a feature: improving node classification with non-neural graph-aware logistic regression

  • 将节点关系编码为特征向量,与自身属性结合进行分类
  • 在多个数据集上超越主流GNN模型,计算速度提升百倍以上
  • 适合追求高效、可解释性的图学习应用

图神经网络(GNN)通过消息传递机制同时利用结构和特征信息,已成为图机器学习的标准方法。然而,这些方法在同质性较弱的数据集上泛化能力差,促使研究者设计复杂的神经架构,带来效率与可扩展性挑战。为此,本文提出非神经的图感知逻辑回归(GLR),不依赖消息传递,而是将每个节点的关联关系编码为额外特征向量,并与节点自身属性结合。实验表明,该方法在严格评估框架下优于基础及先进GNN模型,在多个数据集上实现更优泛化能力,且计算时间比最佳神经对手快两个数量级。

原文摘要 · Abstract (English)

Graph Neural Networks (GNNs) and their message passing framework that leverages both structural and feature information, have become a standard method for solving graph-based machine learning problems. However, these approaches still struggle to generalise well beyond datasets that exhibit strong homophily, where nodes of the same class tend to connect. This limitation has led to the development of complex neural architectures that pose challenges in terms of efficiency and scalability. In response to these limitations, we focus on simpler and more scalable approaches and introduce Graph-aware Logistic Regression (GLR), a non-neural model designed for node classification tasks. Unlike traditional graph algorithms that use only a fraction of the information accessible to GNNs, our proposed model simultaneously leverages both node features and the relationships between entities. However instead of relying on message passing, our approach encodes each node's relationships as an additional feature vector, which is then combined with the node's self attributes. Extensive experimental results, conducted within a rigorous evaluation framework, show that our proposed GLR approach outperforms both foundational and sophisticated state-of-the-art GNN models in node classification tasks. Going beyond the traditional limited benchmarks, our experiments indicate that GLR increases generalisation ability while reaching performance gains in computation time up to two orders of magnitude compared to it best neural competitor.

图学习逻辑回归非神经网络节点分类

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