arXiv:2511.11046cs.LGcs.AI2025-11

让图神经网络更懂邻居之间的关系,提升表达能力。

Enhancing Graph Representations with Neighborhood-Contextualized Message-Passing

  • 引入邻域上下文信息,改进传统消息传递机制。
  • 在多个数据集上显著优于基线模型,且计算效率高。
  • 适合需要高效强表达力的图学习任务研究者。

图神经网络(GNN)已成为分析关系数据的重要工具。经典GNN分为卷积型、注意力型和消息传递型三类。标准消息传递型虽具表达力,但其成对消息仅考虑中心节点与单个邻居的特征,忽略更广泛邻域中的上下文信息,可能限制对整体邻域关系的学习。为此,本文首先精炼了邻域上下文化的概念,借鉴集合聚合方法与注意力机制的关键性质,构建了邻域上下文化消息传递(NCMP)框架。进一步提出一种数学严谨的参数化方法,发展出软同构邻域上下文化图卷积网络(SINC-GCN)。在多样化的合成与基准数据集上,SINC-GCN在表达力与效率间取得良好平衡。尽管更复杂模型有显著计算开销,SINC-GCN仍以极低渐近时间复杂度实现显著性能提升,效应量大,凸显邻域上下文化的独特价值。通过整合多集合邻域上下文,该框架为提升经典GNN的图表示能力提供了一条实用且可扩展的路径。

原文摘要 · Abstract (English)

Graph neural networks (GNNs) have become an indispensable tool for analyzing relational data. Classical GNNs are broadly classified into three variants: convolutional, attentional, and message-passing. While the standard message-passing variant is expressive, its typical pair-wise messages only consider the features of the center node and each neighboring node individually. This design fails to incorporate contextual information contained within the broader local neighborhood, potentially hindering its ability to learn meaningful relationships within the entire set of neighboring nodes. To address this, the paper first refines the concept of neighborhood-contextualization within GNNs, leveraging ideas from set-based aggregation methods and a key property of the attentional variant. This then serves as the basis for generalizing the message-passing variant to the proposed neighborhood-contextualized message-passing (NCMP) framework. To demonstrate its utility, a simple, mathematically grounded method to parametrize and operationalize NCMP is presented, leading to the development of the proposed Soft-Isomorphic Neighborhood-Contextualized Graph Convolution Network (SINC-GCN). Across a diverse set of synthetic and benchmark datasets, SINC-GCN strikes a highly favorable balance between expressivity and efficiency. Notably, while more complex models incur significant computational overhead, SINC-GCN delivers substantial performance gains with considerable effect sizes over baseline GNN models while maintaining a highly efficient asymptotic runtime complexity, further underscoring the distinctive utility of neighborhood-contextualization. Overall, by integrating multiset neighborhood context, the proposed NCMP framework serves as a practical and scalable path toward enhancing the graph representational power of classical GNNs.

图神经网络消息传递邻域上下文

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