arXiv:2508.13625cs.LGcs.AI2025-08中稿 · Globecom 2025

一拍即合的联邦学习,让不同设备模型联手打造更大更强的服务器模型。

Towards a Larger Model via One-Shot Federated Learning on Heterogeneous Client Models

  • 用预测结果替代参数传输,实现单轮通信构建大模型。
  • 在无标签公共数据集上通过伪标签迭代优化,提升模型精度。
  • 支持异构客户端架构,适合资源受限但数据私密的移动网络场景。

大规模模型虽性能优越,但移动端计算资源有限且数据隐私约束导致无法直接共享原始数据。联邦学习(FL)通过交换模型参数实现协同训练,但需统一架构与多轮通信,难以应对设备异构性,且计算和通信开销高。为此,我们提出FedOL,在单轮通信中构建更大更全面的服务器模型。该方法采用知识蒸馏,客户端仅传输对无标签公共数据集的预测输出,以紧凑的预测代替完整模型权重,降低通信负担,并支持异构模型架构。核心挑战在于本地数据分布偏移导致预测偏差,且公共数据无真实标签影响学习可靠性。为此,FedOL设计专用目标函数,迭代优化伪标签与服务器模型;同时引入定制化伪标签生成与知识融合策略,有效整合多样化知识。仿真结果表明,FedOL显著优于现有基线,为移动端提供低成本、高效率的大模型构建方案。

原文摘要 · Abstract (English)

Large models, renowned for superior performance, outperform smaller ones even without billion-parameter scales. While mobile network servers have ample computational resources to support larger models than client devices, privacy constraints prevent clients from directly sharing their raw data. Federated Learning (FL) enables decentralized clients to collaboratively train a shared model by exchanging model parameters instead of transmitting raw data. Yet, it requires a uniform model architecture and multiple communication rounds, which neglect resource heterogeneity, impose heavy computational demands on clients, and increase communication overhead. To address these challenges, we propose FedOL, to construct a larger and more comprehensive server model in one-shot settings (i.e., in a single communication round). Instead of model parameter sharing, FedOL employs knowledge distillation, where clients only exchange model prediction outputs on an unlabeled public dataset. This reduces communication overhead by transmitting compact predictions instead of full model weights and enables model customization by allowing heterogeneous model architectures. A key challenge in this setting is that client predictions may be biased due to skewed local data distributions, and the lack of ground-truth labels in the public dataset further complicates reliable learning. To mitigate these issues, FedOL introduces a specialized objective function that iteratively refines pseudo-labels and the server model, improving learning reliability. To complement this, FedOL incorporates a tailored pseudo-label generation and knowledge distillation strategy that effectively integrates diverse knowledge. Simulation results show that FedOL significantly outperforms existing baselines, offering a cost-effective solution for mobile networks where clients possess valuable private data but limited computational resources.

联邦学习知识蒸馏异构模型单轮通信

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