arXiv:2607.26618cs.LGcs.CL2026-07

解决联邦学习中任务异构导致的模型干扰问题,提升多任务协作效果。

FedWeave: Rethinking the Unit of Specialization in Heterogeneous Federated MoE-LoRA

论文配图:FedWeave: Rethinking the Unit of Specialization in Heterogeneous Federated MoE-LoRA
图 1 · 摘自论文原文
  • 采用异步聚合:专家与路由分开优化,分别满足纯度与对比需求。
  • 原型发现实现跨客户端对齐,支持在原型层面聚合专家。
  • 稀疏推理仅激活一个专家,性能接近全路由,适合资源受限场景。

联邦PEFT使大模型能在不共享原始数据的前提下协同适应分散的私有数据。然而,客户端间任务异构会导致聚合时出现跨任务干扰和梯度冲突。联邦MoE-LoRA通过专用的LoRA专家和条件路由缓解此问题。但现有方法通常以客户端为粒度进行专业化,隐含假设客户端任务一致。本文核心洞察是:专家需保持更新模式一致性(纯度),而路由器需接触多任务混合样本(对比性)。为此提出FedWeave框架,采用异步聚合策略,分离专家聚合与路由器优化。该方法利用无监督原型发现构建本地桶并跨客户端对齐,实现原型级专家聚合,同时保留客户端的混合任务轨迹用于路由器训练。推理阶段仅激活一个专家,实现稀疏推理,性能接近软路由。理论分析表明:异步聚合可通过非模式污染控制专家收敛稳定性,揭示碎片化路由轨迹带来的共识误差,并界定稀疏推理风险。在主流大模型骨干的异构多任务基准上,FedWeave持续优于强基线,消融实验验证了设计有效性。

原文摘要 · Abstract (English)

Federated PEFT enables LLMs to collaboratively adapt to decentralized private data without sharing raw examples. However, task heterogeneity across clients can cause cross-task interference and gradient conflicts during aggregation. Federated MoE-LoRA addresses this challenge through specialized LoRA experts and conditional routing. Yet existing methods typically specialize at client granularity, implicitly assuming task-coherent clients. Our core insight is that experts need purity, namely pattern-coherent updates that preserve specialization, whereas routers need contrast, namely mixed-task observations that support expert comparison. We propose FedWeave, a framework that adopts asymmetric aggregation, separating expert aggregation from router optimization to meet these two requirements. FedWeave uses unsupervised prototype discovery to form local buckets and align them across clients, enabling prototype-level expert aggregation while retaining mixed-task client trajectories for router training. At inference, FedWeave performs sparse inference with one active expert while preserving nearly all soft-routing performance. Our theoretical analysis explains why asymmetric aggregation is advantageous: it controls expert convergence in stationarity through off-pattern contamination, identifies the consensus error induced by fragmented router trajectories, and bounds sparse-inference risk. On a heterogeneous multi-task benchmark with mainstream LLM backbones, FedWeave consistently outperforms strong baselines, while ablations verify the effectiveness of our design.

联邦学习MoE-LoRA稀疏推理任务异构

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