arXiv:2505.19699cs.LGcs.AI2025-05中稿 · Knowledge-Based Sy…被引 4

无数据蒸馏框架Mosaic,解决异构分布式学习中的模型与数据异质性问题。

Mosaic: Data-Free Knowledge Distillation via Mixture-of-Experts for Heterogeneous Distributed Environments

  • 用本地生成模型合成隐私保护的虚拟数据,避免真实数据暴露。
  • 构建专家混合(MoE)架构,通过少量原型训练轻量元模型融合专家预测。
  • 在图像与多模态基准上超越现有方法,尤其适应复杂异构环境。

联邦学习(FL)是一种去中心化机器学习范式,使客户端在保护数据隐私的前提下协同训练模型。然而,模型与数据异质性的共存导致客户端间表征不一致和优化动态发散,最终影响全局性能。为此,我们提出Mosaic——一种专为异构分布式环境设计的无数据知识蒸馏框架。Mosaic首先训练本地生成模型以逼近各客户端的个性化分布,生成严格隔离真实数据的合成数据,保障隐私;随后,基于客户端模型的专业知识构建混合专家(MoE)结构,并利用生成数据将知识蒸馏至全局模型。为进一步优化MoE架构,Mosaic引入一个在少数代表性原型上训练的轻量级元模型,实现专家预测的融合。在标准图像与多模态基准上的大量实验表明,无论面对模型或数据异质性,Mosaic均持续优于当前最优方法。源代码已公开于https://github.com/Wings-Of-Disaster/Mosaic。

原文摘要 · Abstract (English)

Federated Learning (FL) is a decentralized machine learning paradigm that enables clients to collaboratively train models while preserving data privacy. However, the coexistence of model and data heterogeneity gives rise to inconsistent representations and divergent optimization dynamics across clients, ultimately hindering robust global performance. To transcend these challenges, we propose Mosaic, a novel data-free knowledge distillation framework tailored for heterogeneous distributed environments. Mosaic first trains local generative models to approximate each client's personalized distribution, enabling synthetic data generation that safeguards privacy through strict separation from real data. Subsequently, Mosaic forms a Mixture-of-Experts (MoE) from client models based on their specialized knowledge, and distills it into a global model using the generated data. To further enhance the MoE architecture, Mosaic integrates expert predictions via a lightweight meta model trained on a few representative prototypes. Extensive experiments on standard image and multimodal benchmarks demonstrate that Mosaic consistently outperforms state-of-the-art approaches under both model and data heterogeneity. The source code has been published at https://github.com/Wings-Of-Disaster/Mosaic.

联邦学习知识蒸馏异构环境生成模型

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