arXiv:2605.16690cs.LG2026-05

提出UB-SMoE,让不同设备在联邦微调中更省资源且表现更好。

UB-SMoE: Universally Balanced Sparse Mixture-of-Experts for Resource-adaptive Federated Fine-tuning of Foundation Models

论文配图:UB-SMoE: Universally Balanced Sparse Mixture-of-Experts for Resource-adaptive Federated Fine-tuning of Foundation Models
图 1 · 摘自论文原文
  • 动态路由+伪梯度设计,解决专家使用不均和梯度不可导问题
  • 低资源设备计算量减少45%,性能提升8.7倍
  • 适合算力差异大的设备协同训练大模型

异构LoRA秩方法通过根据客户端计算能力分配不同秩来应对联邦微调中的系统异构性,但计算节省有限,因密集前馈计算占主导。稀疏混合专家(SMoE)通过条件计算提供替代方案,然而我们发现其直接应用于异构联邦设置会引入两大问题:(i) 专家使用不均衡,(ii) Top-K路由非可微。收敛性分析表明,这些问题导致收敛性能下降,尤其对资源受限客户端影响显著。为此,我们提出通用平衡稀疏混合专家(UB-SMoE),引入动态调制路由(DMR)以重平衡专家使用,并采用通用伪梯度(PG)重构未激活专家的学习信号。这些机制形成自强化循环,保持各异构客户端上专家的持续可用性。基准实验显示,UB-SMoE在低资源客户端上实现最高达45.0%的计算量降低,同时性能较现有异构LoRA秩方法提升8.7倍。

原文摘要 · Abstract (English)

Heterogeneous LoRA-rank methods address system heterogeneity in federated fine-tuning of foundation models by assigning client-specific ranks based on computational capabilities. However, these methods achieve only marginal computational savings, as dense feed-forward computations dominate. Sparse Mixture-of-Experts (SMoE) provides a promising alternative through conditional computation, yet we identify that its naive application to heterogeneous federated settings introduces two critical discordances: (i) expert utilization imbalance and (ii) non-differentiability of Top-K routing. Our convergence analysis demonstrates that these discordances lead to degraded convergence, particularly for resource-constrained clients. To address these challenges, we propose Universally Balanced Sparse Mixture-of-Experts (UB-SMoE), which introduces Dynamic Modulated Routing (DMR) to rebalance expert utilization, and Universal Pseudo-Gradient (PG) to reconstruct learning signals for non-activated experts. These mechanisms form a self-reinforcing cycle that maintains expert viability across heterogeneous clients. Experiments on benchmarks show that UB-SMoE achieves up to $45.0\%$ computational reduction on low-resource clients while improving their performance by $8.7 \times$ compared to existing heterogeneous LoRA-rank methods.

联邦学习稀疏专家资源自适应模型压缩

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