arXiv:2509.01750cs.LG2025-09中稿 · Globecom 2025被引 2

针对无线网络下大模型联邦微调通信开销大的问题,提出自适应稀疏化日志传输方案。

Communication-Aware Knowledge Distillation for Federated LLM Fine-Tuning over Wireless Networks

  • 根据实时通信条件动态选择重要日志,实现自适应稀疏化传输
  • 设计自适应聚合机制,解决稀疏化带来的维度不一致问题
  • 结合LoRA投影优化蒸馏损失,降低50%通信开销且提升性能

面向大语言模型的联邦学习提供了一种保护隐私的协作微调方案,使客户端可在不交换原始数据的情况下本地微调大型语言模型(LLMs)或小型语言模型(SLMs)。传统参数共享方法虽解决部分技术挑战,但仍面临高通信开销和异构模型架构适配困难的问题。联邦蒸馏通过共享日志实现知识互传,通常比参数共享通信开销更低。然而,大模型日志高维特性在带宽受限的客户端中仍具挑战性。本文聚焦于高效通信的联邦大模型蒸馏,提出自适应Top-k日志选择机制,根据实时通信状况动态稀疏化日志;为解决稀疏化引入的维度不一致问题,设计自适应日志聚合方案,有效缓解传统零填充带来的冗余信息;为进一步增强蒸馏效果,将大模型中的LoRA适配隐藏层投影融入蒸馏损失,进一步降低通信开销并提供更丰富表示。实验表明,该方案在性能上优于基线方法,同时通信开销降低约50%。

原文摘要 · Abstract (English)

Federated learning (FL) for large language models (LLMs) offers a privacy-preserving scheme, enabling clients to collaboratively fine-tune locally deployed LLMs or smaller language models (SLMs) without exchanging raw data. While parameter-sharing methods in traditional FL models solves number of technical challenges, they still incur high communication overhead and struggle with adapting to heterogeneous model architectures. Federated distillation, a framework for mutual knowledge transfer via shared logits, typically offers lower communication overhead than parameter-sharing methods. However, transmitting logits from LLMs remains challenging for bandwidth-limited clients due to their high dimensionality. In this work, we focus on a federated LLM distillation with efficient communication overhead. To achieve this, we first propose an adaptive Top-k logit selection mechanism, dynamically sparsifying logits according to real-time communication conditions. Then to tackle the dimensional inconsistency introduced by the adaptive sparsification, we design an adaptive logits aggregation scheme, effectively alleviating the artificial and uninformative inputs introduced by conventional zero-padding methods. Finally, to enhance the distillation effect, we incorporate LoRA-adapted hidden-layer projection from LLM into the distillation loss, reducing the communication overhead further while providing richer representation. Experimental results demonstrate that our scheme achieves superior performance compared to baseline methods while effectively reducing communication overhead by approximately 50%.

联邦学习知识蒸馏通信优化大模型

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