arXiv:2412.00241cs.LG2024-12中稿 · publication in TML…被引 2

新模型Mega-GNN能更好处理多重边带属性的图数据

MEGA: Message Passing Neural Networks for Multigraphs with EdGe Attributes

  • 先合并同一邻居的多条边特征,再跨邻居聚合
  • 在社交与金融网络上性能超越现有方法
  • 保持数学对称性且计算效率不降

边属性多重图中,同一节点对间存在多条带不同属性的边,真实系统中常见。有效学习需保留重复交互信息,同时区分不同邻居贡献。现有神经网络方法受限:部分丢失重复交互信息,部分破坏置换等变性。本文提出邻近感知聚合,先融合每个邻居的多条边特征,再跨邻居聚合,捕捉标准单阶段聚合无法表示的邻居级统计特征。基于此,构建Mega-GNN——一种面向边属性多重图的模型无关消息传递框架。证明其具有置换等变性,渐近复杂度与标准带边更新的GNN相当。在社交网络与金融交易网络数据集上评估,邻近感知聚合持续提升GNN性能,达到或超过当前最优方法。

原文摘要 · Abstract (English)

Edge-attributed multigraphs, in which multiple edges with distinct attributes connect the same pair of nodes, arise naturally in many real-world systems. In these graphs, effective learning requires preserving information from repeated interactions while distinguishing contributions from different neighbors. Existing neural network solutions for edge-attributed multigraphs remain limited: some lose information from repeated interactions, while others break permutation equivariance. To address this, we introduce \emph{neighbor-aware aggregation}, an operator that first combines multi-edge features for each neighbor and then aggregates across neighbors. This operator captures per-neighbor statistics that standard single-stage aggregation cannot represent. Building on this operator, we present MEGA-GNN, a model-agnostic message-passing framework for edge-attributed multigraphs. We show that MEGA-GNN is permutation equivariant and has the same asymptotic complexity as standard GNNs with edge updates. We evaluate our approach on datasets from social networks and financial transaction networks. Neighbor-aware aggregation consistently improves GNN performance and matches or surpasses state-of-the-art methods.

图神经网络多重图消息传递

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