arXiv:2603.15954cs.LGcs.AI2026-03ACL被引 2

为手机端高效部署大模型,设计出低延迟、兼容性强的轻量级语言模型。

MobileLLM-Flash: Latency-Guided On-Device LLM Design for Industry Scale Deployment

  • 基于硬件反馈的架构搜索,联合优化模型结构与注意力模式。
  • 在移动端CPU上推理速度提升1.8倍,支持8k上下文长度。
  • 无需定制内核,适配标准运行时,适合工业级落地应用。

实时AI体验需要在资源受限设备上高效部署大语言模型(OD-LLMs)。理想的OD-LLMs应实现近实时响应并具备广泛硬件兼容性,以最大化用户覆盖。本文提出一种面向移动延迟约束的硬件闭环架构搜索方法,可生成无需定制内核、兼容标准移动端运行时(如Executorch)的模型。该方法避免使用专用注意力机制,转而采用注意力跳过策略加速长上下文处理,同时联合优化模型架构(层数、维度)与注意力模式。通过将候选模型视为预训练主干的剪枝版本,继承权重实现高精度且极少微调。利用延迟评估成本低的特点,分阶段学习准确的延迟模型,并搜索延迟与质量间的帕累托前沿。最终产出MobileLLM-Flash系列基础模型(350M、650M、1.4B),支持高达8k上下文长度,在移动端CPU上预填充和解码速度分别提升1.8倍和1.6倍,性能相当或更优。对帕累托前沿设计选择的分析提供了可操作的OD-LLM设计原则。

原文摘要 · Abstract (English)

Real-time AI experiences call for on-device large language models (OD-LLMs) optimized for efficient deployment on resource-constrained hardware. The most useful OD-LLMs produce near-real-time responses and exhibit broad hardware compatibility, maximizing user reach. We present a methodology for designing such models using hardware-in-the-loop architecture search under mobile latency constraints. This system is amenable to industry-scale deployment: it generates models deployable without custom kernels and compatible with standard mobile runtimes like Executorch. Our methodology avoids specialized attention mechanisms and instead uses attention skipping for long-context acceleration. Our approach jointly optimizes model architecture (layers, dimensions) and attention pattern. To efficiently evaluate candidates, we treat each as a pruned version of a pretrained backbone with inherited weights, thereby achieving high accuracy with minimal continued pretraining. We leverage the low cost of latency evaluation in a staged process: learning an accurate latency model first, then searching for the Pareto-frontier across latency and quality. This yields MobileLLM-Flash, a family of foundation models (350M, 650M, 1.4B) for efficient on-device use with strong capabilities, supporting up to 8k context length. MobileLLM-Flash delivers up to 1.8x and 1.6x faster prefill and decode on mobile CPUs with comparable or superior quality. Our analysis of Pareto-frontier design choices offers actionable principles for OD-LLM design.

手机端模型低延迟架构搜索轻量化

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