arXiv:2508.10471cs.LG2025-08被引 1

解决联邦图学习中少数类数据不足的问题

GraphFedMIG: Tackling Class Imbalance in Federated Graph Learning via Mutual Information-Guided Generation

  • 通过生成对抗网络合成少数类节点特征
  • 利用互信息机制引导生成重点提升稀有类表现
  • 适合处理数据分布不均的分布式图学习场景

联邦图学习(FGL)允许多个客户端在不共享私有图数据的前提下协同训练强大的图神经网络。与通用联邦学习类似,FGL受统计异质性挑战,各客户端数据分布非独立同分布(non-IID)会严重损害模型性能。其中,类别不平衡尤为严重,导致全局模型偏向多数类,难以识别罕见但关键事件。这一问题在FGL中更突出,因为少数类节点常被偏倚的邻域信息包围,阻碍表达性嵌入的学习。为此,我们提出GraphFedMIG,一种新型FGL框架,将问题重构为联邦生成式数据增强任务。该框架采用分层生成对抗网络,每个客户端训练本地生成器以合成高保真特征表示,并按客户聚类分组,每组共享专用判别器。关键在于,框架设计了互信息引导机制,计算每个客户端的独特信息价值,动态修正本地生成器参数,确保后续互信息引导生成聚焦于生成高价值、少数类特征。我们在四个真实世界数据集上进行了广泛实验,结果表明GraphFedMIG优于其他基线方法。

原文摘要 · Abstract (English)

Federated graph learning (FGL) enables multiple clients to collaboratively train powerful graph neural networks without sharing their private, decentralized graph data. Inherited from generic federated learning, FGL is critically challenged by statistical heterogeneity, where non-IID data distributions across clients can severely impair model performance. A particularly destructive form of this is class imbalance, which causes the global model to become biased towards majority classes and fail at identifying rare but critical events. This issue is exacerbated in FGL, as nodes from a minority class are often surrounded by biased neighborhood information, hindering the learning of expressive embeddings. To grapple with this challenge, we propose GraphFedMIG, a novel FGL framework that reframes the problem as a federated generative data augmentation task. GraphFedMIG employs a hierarchical generative adversarial network where each client trains a local generator to synthesize high-fidelity feature representations. To provide tailored supervision, clients are grouped into clusters, each sharing a dedicated discriminator. Crucially, the framework designs a mutual information-guided mechanism to steer the evolution of these client generators. By calculating each client's unique informational value, this mechanism corrects the local generator parameters, ensuring that subsequent rounds of mutual information-guided generation are focused on producing high-value, minority-class features. We conduct extensive experiments on four real-world datasets, and the results demonstrate the superiority of the proposed GraphFedMIG compared with other baselines.

联邦学习图神经网络数据不平衡生成模型

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