English

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

Machine Learning 2026-04-29 v2 Artificial Intelligence

Abstract

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.

Keywords

Cite

@article{arxiv.2603.15954,
  title  = {MobileLLM-Flash: Latency-Guided On-Device LLM Design for Industry Scale Deployment},
  author = {Hanxian Huang and Igor Fedorov and Andrey Gromov and Bernard Beckerman and Naveen Suda and David Eriksson and Maximilian Balandat and Rylan Conway and Patrick Huber and Chinnadhurai Sankar and Ayushi Dalmia and Zechun Liu and Lemeng Wu and Tarek Elgamal and Adithya Sagar and Vikas Chandra and Raghuraman Krishnamoorthi},
  journal= {arXiv preprint arXiv:2603.15954},
  year   = {2026}
}

Comments

Accepted to ACL Industry Track 2026

R2 v1 2026-07-01T11:23:17.231Z