arXiv:2501.09410cs.NIcs.AI2025-01被引 9

让边缘端大模型协作推理更省电更快,提升资源利用率。

MoE$^2$: Optimizing Collaborative Inference for Edge Large Language Models

  • 设计两级专家选择机制,解耦训练与推理,降低复杂度。
  • 在不同延迟和能耗预算下,实现性能最优的平衡。
  • 适合边缘部署大模型、追求低功耗高效率的场景。

大型语言模型(LLMs)在自然语言处理任务中展现出卓越能力。利用边缘端LLM的异构特性对多样新兴应用至关重要,可提升成本效益并降低延迟。本文提出一种新型协同推理框架——边缘专家混合(MoE²),将联合门控与专家选择问题形式化,以在能量和延迟约束下优化推理性能。不同于传统MoE,边缘LLM的专家选择更具挑战性,因其组合性及属性异质性。为此,我们提出两级专家选择机制,并发现门控参数在专家选择中的保优性质,使训练与选择过程可解耦,显著降低复杂度。此外,利用目标函数的单调性,设计了离散单调优化算法实现最优专家选择。我们在搭载NVIDIA Jetson AGX Orin和RTX 4090 GPU的边缘服务器上进行了大量实验,结果验证了多种LLM模型的性能提升,表明MoE²能在不同延迟与能耗预算下实现最优权衡,且在各类系统资源约束下优于基线方法。

原文摘要 · Abstract (English)

Large language models (LLMs) have demonstrated remarkable capabilities across a wide range of natural language processing tasks. Exploiting the heterogeneous capabilities of edge LLMs is crucial for diverse emerging applications, as it enables greater cost-effectiveness and reduced latency. In this work, we introduce \textit{Mixture-of-Edge-Experts (MoE$^2$)}, a novel collaborative inference framework for edge LLMs. We formulate the joint gating and expert selection problem to optimize inference performance under energy and latency constraints. Unlike conventional MoE problems, LLM expert selection is significantly more challenging due to the combinatorial nature and the heterogeneity of edge LLMs across various attributes. To this end, we propose a two-level expert selection mechanism through which we uncover an optimality-preserving property of gating parameters across expert selections. This property enables the decomposition of the training and selection processes, significantly reducing complexity. Furthermore, we leverage the objective's monotonicity and design a discrete monotonic optimization algorithm for optimal expert selection. We implement edge servers with NVIDIA Jetson AGX Orins and NVIDIA RTX 4090 GPUs, and perform extensive experiments. Our results validate that performance improvements of various LLM models and show that our MoE$^2$ method can achieve optimal trade-offs among different delay and energy budgets, and outperforms baselines under various system resource constraints.

边缘计算大模型推理MoE节能优化

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