将大模型专家网络分布到手机和基站,降低推理延迟。
WDMoE: Wireless Distributed Mixture of Experts for Large Language Models
- 基站负责门控与前层计算,手机分担专家网络并行推理。
- 联合优化选专家与带宽,延迟降低超40%且不损失精度。
- 适用于资源受限的移动端大模型部署,如智能终端应用。
大型语言模型(LLMs)在自然语言处理任务中取得显著进展,但无线网络在支持LLM方面的潜力尚未充分探索。本文提出无线分布式混合专家(WDMoE)架构,实现大模型在基站边缘服务器与移动设备间的协同部署。具体地,将LLM中的混合专家(MoE)层拆分:门控网络与前序神经网络层置于基站,专家网络则分布于各移动设备。该部署方式利用移动设备上专家网络的并行推理能力,有效利用其有限的计算与缓存资源。为此,我们构建了融合模型能力与延迟的性能指标,并基于该指标联合优化专家选择与带宽分配,以最小化延迟同时保持精度。此外,我们使用NVIDIA Jetson套件搭建硬件测试平台验证方案有效性。理论仿真与实际硬件实验均表明,所提方法可显著降低延迟,且不影响大模型性能。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have achieved significant success in various natural language processing tasks, but the role of wireless networks in supporting LLMs has not been thoroughly explored. In this paper, we propose a wireless distributed Mixture of Experts (WDMoE) architecture to enable collaborative deployment of LLMs across edge servers at the base station (BS) and mobile devices in wireless networks. Specifically, we decompose the MoE layer in LLMs by placing the gating network and the preceding neural network layer at BS, while distributing the expert networks among the devices. This deployment leverages the parallel inference capabilities of expert networks on mobile devices, effectively utilizing the limited computing and caching resources of these devices. Accordingly, we develop a performance metric for WDMoE-based LLMs, which accounts for both model capability and latency. To minimize the latency while maintaining accuracy, we jointly optimize expert selection and bandwidth allocation based on the performance metric. Moreover, we build a hardware testbed using NVIDIA Jetson kits to validate the effectiveness of WDMoE. Both theoretical simulations and practical hardware experiments demonstrate that the proposed method can significantly reduce the latency without compromising LLM performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。