arXiv:2507.06482cs.LG2025-07ICCV被引 5

用扩散模型缓解联邦学习中的数据异构问题,提升模型性能。

FedDifRC: Unlocking the Potential of Text-to-Image Diffusion Models in Heterogeneous Federated Learning

  • 引入扩散模型生成语义引导信号,构建文本对比与噪声正则化机制。
  • 在多个数据异构场景下显著提升模型收敛速度与准确率。
  • 无需标签数据即可自监督运行,适合隐私保护场景应用。

联邦学习旨在跨多方协作训练模型的同时保护隐私,但数据异构问题严重影响模型收敛与性能。本文首次将强大的扩散模型引入联邦学习框架,证明扩散表示在训练过程中具有有效引导作用。为此提出一种基于扩散表示协作的新型联邦学习范式FedDifRC,通过构建文本驱动的对比学习和噪声驱动的一致性正则化,提供丰富的类别语义信息与稳定的收敛信号。一方面,利用扩散模型对不同文本提示的条件反馈,建立文本驱动的对比学习策略;另一方面,引入噪声驱动的正则化,使本地样本与扩散去噪表示对齐,约束特征空间中的优化区域。此外,FedDifRC可扩展为无需标签数据的自监督方案。我们还提供了针对非凸目标的理论分析以保证收敛性。在多种场景下的实验验证了其有效性及关键组件的高效性。

原文摘要 · Abstract (English)

Federated learning aims at training models collaboratively across participants while protecting privacy. However, one major challenge for this paradigm is the data heterogeneity issue, where biased data preferences across multiple clients, harming the model's convergence and performance. In this paper, we first introduce powerful diffusion models into the federated learning paradigm and show that diffusion representations are effective steers during federated training. To explore the possibility of using diffusion representations in handling data heterogeneity, we propose a novel diffusion-inspired Federated paradigm with Diffusion Representation Collaboration, termed FedDifRC, leveraging meaningful guidance of diffusion models to mitigate data heterogeneity. The key idea is to construct text-driven diffusion contrasting and noise-driven diffusion regularization, aiming to provide abundant class-related semantic information and consistent convergence signals. On the one hand, we exploit the conditional feedback from the diffusion model for different text prompts to build a text-driven contrastive learning strategy. On the other hand, we introduce a noise-driven consistency regularization to align local instances with diffusion denoising representations, constraining the optimization region in the feature space. In addition, FedDifRC can be extended to a self-supervised scheme without relying on any labeled data. We also provide a theoretical analysis for FedDifRC to ensure convergence under non-convex objectives. The experiments on different scenarios validate the effectiveness of FedDifRC and the efficiency of crucial components.

联邦学习扩散模型数据异构自监督

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