arXiv:2609.06073cs.LGcs.AI2026-09

通过结构化子空间压缩更新量,显著降低联邦微调通信开销。

FedSubMuon: Communication-Efficient Federated LLM Fine-Tuning via Structured Subspace Muon

论文配图:FedSubMuon: Communication-Efficient Federated LLM Fine-Tuning via Structured Subspace Muon
图 1 · 摘自论文原文
  • 在共享子空间中优化紧凑系数矩阵,保持矩阵感知更新结构。
  • 在五组数据-模型对中,四组达到最高精度;通信量仅为基线的1/5.5至1/1.4。
  • 适合资源受限设备上的高效联邦微调,尤其关注通信效率与精度平衡。

联邦微调可适应大语言模型(LLMs)到分散客户端数据,但跨设备训练中的可扩展性常受高通信成本限制。Muon是一种优化器,通过正交化矩阵参数的动量来提升优化性能。现有联邦Muon方法虽证明了矩阵感知优化的优势,但仍需传输完整层尺寸的更新和优化器状态。直接将Muon应用于LoRA因子虽可减少通信,但改变了优化对象并削弱其矩阵感知更新几何。我们提出FedSubMuon,一种通信高效的联邦Muon微调方法,通过在共享结构化子空间内优化紧凑系数矩阵实现,保持Muon作用于单一矩阵可训练对象的同时,将客户端上传量降至紧凑系数矩阵。进一步提出FedSubMuon-GT,采用投影梯度使追踪的子空间基向任务相关梯度方向自适应调整。在指令微调和数学推理任务上的实验表明,FedSubMuon-GT在五组数据-模型对中的四组达到最优准确率;而FedSubMuon在所有匹配通信预算下表现最佳。在Dolly-15K数据集上,最接近的通信基线分别需要Llama-1B和Qwen-4B 5.5倍和1.4倍的总通信量。

原文摘要 · Abstract (English)

Federated fine-tuning adapts large language models (LLMs) to decentralized client data, but its scalability in cross-device training is often limited by the high communication cost. Muon is an optimizer that improves optimization performance by orthogonalizing momentum for matrix-valued parameters. Existing federated Muon methods demonstrate the benefit of matrix-aware optimization in federated learning, but still require transmitting full layer-size updates and optimizer state. A natural way to reduce communication is to directly apply Muon to LoRA factors, but this changes the optimized object and weakens Muon's matrix-aware update geometry. We propose FedSubMuon, a communication-efficient federated Muon fine-tuning method that optimizes compact coefficient matrices within shared structured subspaces. This design keeps Muon on a single matrix-valued trainable object, while reducing the client upload to compact coefficient matrices. We further introduce FedSubMuon-GT, an accuracy-oriented extension that uses projected gradients to adapt tracked subspace bases toward task-relevant gradient directions. Experiments on instruction tuning and mathematical reasoning show that FedSubMuon-GT achieves the best overall accuracy on four of five dataset-model pairs, while FedSubMuon performs best under all matched communication budgets. On Dolly-15K, the closest communication baseline requires 5.5 times and 1.4 times more total communication on Llama-1B and Qwen-4B, respectively.

联邦学习通信效率大模型微调子空间优化

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