用扩散模型提升跨模态联邦学习的语义一致性
Diffusion-Guided Semantic Consistency for Multimodal Heterogeneity
- 借预训练扩散模型的多层语义特征构建共享表示空间
- 在异构数据下比FedAvg最高提升5.49%准确率
- 适合需要跨设备鲁棒感知的多媒体系统
联邦学习在非独立同分布(non-IID)客户端数据面前面临严峻挑战,尤其在多模态感知场景中,传统方法常忽略客户端间的语义差异,导致性能下降。为此,我们提出SemanticFL框架,利用预训练扩散模型(Stable Diffusion)的丰富语义表示,为本地训练提供隐私保护的引导。该方法提取扩散模型中VAE编码的潜在表示与U-Net分层特征,构建对齐异构客户端的共享潜在空间,并通过高效的客户端-服务器架构将重计算任务移至服务器端。进一步采用跨模态对比学习的统一一致性机制,稳定模型收敛。我们在CIFAR-10、CIFAR-100和TinyImageNet上进行广泛实验,涵盖多种异构场景。结果表明,SemanticFL优于现有联邦学习方法,在多项设置中相较FedAvg最高提升5.49%准确率,验证了其在异构且多模态数据下的鲁棒表征学习能力。
原文摘要 · Abstract (English)
Federated learning (FL) is severely challenged by non-independent and identically distributed (non-IID) client data, a problem that degrades global model performance, especially in multimodal perception settings. Conventional methods often fail to address the underlying semantic discrepancies between clients, leading to suboptimal performance for multimedia systems requiring robust perception. To overcome this, we introduce SemanticFL, a novel framework that leverages the rich semantic representations of pre-trained diffusion models to provide privacy-preserving guidance for local training. Our approach leverages multi-layer semantic representations from a pre-trained Stable Diffusion model (including VAE-encoded latents and U-Net hierarchical features) to create a shared latent space that aligns heterogeneous clients, facilitated by an efficient client-server architecture that offloads heavy computation to the server. A unified consistency mechanism, employing cross-modal contrastive learning, further stabilizes convergence. We conduct extensive experiments on benchmarks including CIFAR-10, CIFAR-100, and TinyImageNet under diverse heterogeneity scenarios. Our results demonstrate that SemanticFL surpasses existing federated learning approaches, achieving accuracy gains of up to 5.49% over FedAvg, validating its effectiveness in learning robust representations for heterogeneous and multimodal data for perception tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。