arXiv:2409.20247cs.DCcs.AI2024-09被引 25

边云协同训练大模型,省电降延时还更稳定

Resource Allocation for Stable LLM Training in Mobile Edge Computing

  • 分层计算:手机调前端,边缘服务器处理后端,降低负载
  • 联合优化能耗与延迟,实测能效和响应速度双提升
  • 引入稳定性约束,适合资源受限的移动场景

随着移动设备在先进应用中的地位日益重要,边缘计算成为克服其固有算力瓶颈的有效方案,尤其在部署大语言模型(LLMs)方面。然而,由于模型计算需求高且涉及数据隐私问题,高效训练与部署仍面临挑战。本文提出一种融合移动用户与边缘服务器的协同训练框架,通过参数高效微调(PEFT)技术,使移动设备仅调整模型前几层,而边缘服务器负责计算量更大的后几层。我们构建了一个多目标优化问题,旨在最小化训练过程中的总能耗与延迟,并通过引入稳定性增强机制解决模型性能波动问题。利用新颖的分数规划方法,求得该问题的平稳点。仿真结果表明,该方法显著降低了能耗与延迟,提升了不同移动环境下的模型可靠性。

原文摘要 · Abstract (English)

As mobile devices increasingly become focal points for advanced applications, edge computing presents a viable solution to their inherent computational limitations, particularly in deploying large language models (LLMs). However, despite the advancements in edge computing, significant challenges remain in efficient training and deploying LLMs due to the computational demands and data privacy concerns associated with these models. This paper explores a collaborative training framework that integrates mobile users with edge servers to optimize resource allocation, thereby enhancing both performance and efficiency. Our approach leverages parameter-efficient fine-tuning (PEFT) methods, allowing mobile users to adjust the initial layers of the LLM while edge servers handle the more demanding latter layers. Specifically, we formulate a multi-objective optimization problem to minimize the total energy consumption and delay during training. We also address the common issue of instability in model performance by incorporating stability enhancements into our objective function. Through novel fractional programming technique, we achieve a stationary point for the formulated problem. Simulations demonstrate that our method reduces the energy consumption as well as the latency, and increases the reliability of LLMs across various mobile settings.

边缘计算大模型资源分配能耗优化

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