arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.03464cs.LG

图神经网络中的节点级滤波:一种专家混合方法

Node-wise Filtering in Graph Neural Networks: A Mixture of Experts Approach

Michigan State University(密歇根州立大学);RIKEN AIP(理化学研究所人工智能项目);Amazon(亚马逊公司)

Haoyu Han, Juanhui Li, Wei Huang, Xianfeng Tang, Hanqing Lu, Chen Luo, Hui Liu, Jiliang Tang

AI总结:

本文提出Node-MoE框架,通过专家混合机制为不同节点自适应选择滤波器,解决单一全局滤波器在同配与异配混合图中性能不佳的问题,并在两类图上验证了有效性。

摘要:

图神经网络(GNNs)已被证明在多种图结构模式下的节点分类任务中非常有效。传统上,GNN采用统一的全局滤波器,通常对同配图使用低通滤波器,对异配图使用高通滤波器。然而,现实世界中的图往往呈现出同配和异配模式的复杂混合,使得单一全局滤波器方法不够理想。在这项工作中,我们从理论上证明,针对一种模式优化的全局滤波器可能会对具有不同模式的节点上的性能产生不利影响。为了解决这个问题,我们引入了一种新颖的GNN框架Node-MoE,该框架利用专家混合来为不同节点自适应地选择合适的滤波器。大量实验证明了Node-MoE在同配图和异配图上的有效性。

英文摘要:

Graph Neural Networks (GNNs) have proven to be highly effective for node classification tasks across diverse graph structural patterns. Traditionally, GNNs employ a uniform global filter, typically a low-pass filter for homophilic graphs and a high-pass filter for heterophilic graphs. However, real-world graphs often exhibit a complex mix of homophilic and heterophilic patterns, rendering a single global filter approach suboptimal. In this work, we theoretically demonstrate that a global filter optimized for one pattern can adversely affect performance on nodes with differing patterns. To address this, we introduce a novel GNN framework Node-MoE that utilizes a mixture of experts to adaptively select the appropriate filters for different nodes. Extensive experiments demonstrate the effectiveness of Node-MoE on both homophilic and heterophilic graphs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑