arXiv:2409.13931cs.LGcs.CL2024-09ICML被引 8

让手机上的大模型私密协作,自动匹配算力与数据差异。

On-Device Collaborative Language Modeling via a Mixture of Generalists and Specialists

  • 用通用专家+专用专家混合架构,动态适配不同设备算力。
  • 实验表明能有效平衡通用知识与个性化生成,抗过拟合。
  • 适合隐私敏感场景下的移动端大模型协同训练。

本地设备上的大语言模型因其提升隐私保护和个性化体验的能力而受到关注。为在数据稀缺情况下实现私密学习,联邦学习已成为标准方法。然而,其面临终端用户间计算资源异构性和数据异构性的挑战。本文提出 CoMiGS(协作式混合通用与专用专家学习),是首个同时应对这两类挑战的方法。核心创新在于将专家混合学习目标设计为双层优化形式,其中路由器通过独立验证集优化,以确保与目标分布对齐。我们采用交替最小化求解该目标,并提供理论分析。方法在用户间共享通用专家,同时本地化数量可变的专用专家,从而适应不同计算资源并保障隐私。大量实验表明,CoMiGS 能有效平衡每个标记生成的通用与个性化知识。结果还显示,由于通用专家的正则化作用,模型对过拟合具有鲁棒性,同时通过专用专家适应本地数据。我们已开源协作大模型代码库。

原文摘要 · Abstract (English)

On-device LLMs have gained increasing attention for their ability to enhance privacy and provide a personalized user experience. To facilitate private learning with scarce data, Federated Learning has become a standard approach. However, it faces challenges such as computational resource heterogeneity and data heterogeneity among end users. We propose CoMiGS ($\textbf{Co}$llaborative learning with a $\textbf{Mi}$xture of $\textbf{G}$eneralists and $\textbf{S}$pecialists), the first approach to address both challenges. A key innovation of our method is the bi-level optimization formulation of the Mixture-of-Experts learning objective, where the router is optimized using a separate validation set to ensure alignment with the target distribution. We solve our objective with alternating minimization, for which we provide a theoretical analysis. Our method shares generalist experts across users while localizing a varying number of specialist experts, thereby adapting to users' computational resources and preserving privacy. Through extensive experiments, we show CoMiGS effectively balances general and personalized knowledge for each token generation. We demonstrate that CoMiGS remains robust against overfitting-due to the generalists' regularizing effect-while adapting to local data through specialist expertise. We open source our codebase for collaborative LLMs.

联邦学习大模型移动推理专家混合

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