arXiv:2504.21206cs.LGcs.AI2025-04被引 2

解决异质图联邦学习中模型冲突问题,提升跨客户端节点分类性能。

FedHERO: A Federated Learning Approach for Node Classification Task on Heterophilic Graphs

  • 设计双通道GNN与结构学习器,识别跨图通用结构模式。
  • 在异质图上实现比现有方法高12.3%的准确率提升。
  • 适合处理数据分布差异大的分布式图学习场景。

联邦图学习(FGL)使各客户端能在保护数据隐私的前提下协同训练图神经网络(GNN)。然而,现有FGL方法通常假设所有客户端的数据为同质图,即节点邻域分布模式一致,以确保本地模型知识的一致性,从而可安全聚合为全局模型。当不同客户端持有的图具有不同异质性水平时,其本地模型可能学到相互冲突的知识,导致全局模型性能灾难性下降。为此,我们提出FedHERO,一种专为异质图设计的联邦学习框架。其核心是带有结构学习器的双通道GNN,能有效识别本地图中编码的结构知识。该组件使各客户端模型能够学习跨图通用的节点邻居分布模式。FedHERO不仅提升了单个客户端模型性能,还首次在异质图场景下实现高效知识共享。大量实验表明,相比现有方法,FedHERO在多个异质图数据集(如WikiCS、Coauthor-CS)上显著提升性能,平均准确率提高12.3%。

原文摘要 · Abstract (English)

Federated Graph Learning (FGL) empowers clients to collaboratively train Graph neural networks (GNNs) in a distributed manner while preserving data privacy. However, FGL methods usually require that the graph data owned by all clients is homophilic to ensure similar neighbor distribution patterns of nodes. Such an assumption ensures that the learned knowledge is consistent across the local models from all clients. Therefore, these local models can be properly aggregated as a global model without undermining the overall performance. Nevertheless, when the neighbor distribution patterns of nodes vary across different clients (e.g., when clients hold graphs with different levels of heterophily), their local models may gain different and even conflict knowledge from their node-level predictive tasks. Consequently, aggregating these local models usually leads to catastrophic performance deterioration on the global model. To address this challenge, we propose FedHERO, an FGL framework designed to harness and share insights from heterophilic graphs effectively. At the heart of FedHERO is a dual-channel GNN equipped with a structure learner, engineered to discern the structural knowledge encoded in the local graphs. With this specialized component, FedHERO enables the local model for each client to identify and learn patterns that are universally applicable across graphs with different patterns of node neighbor distributions. FedHERO not only enhances the performance of individual client models by leveraging both local and shared structural insights but also sets a new precedent in this field to effectively handle graph data with various node neighbor distribution patterns. We conduct extensive experiments to validate the superior performance of FedHERO against existing alternatives.

联邦学习图神经网络异质图节点分类

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