mlsanityData to 7 Oct 2026

Hardware Co-Design Scaling Laws via Roofline Modelling for On-Device LLMs

arXiv 2602.10377 ↗First posted 10 Feb 2026cs.LG cs.CLcited by 5

On the map: Vision-language-action models → Efficient VLA inference and adaptation

Abstract (arXiv)

Vision-Language-Action Models (VLAs) have emerged as a key paradigm of Physical AI and are increasingly deployed in autonomous vehicles, robots, and smart spaces. In these resource-constrained on-device settings, selecting an appropriate large language model (LLM) backbone is a critical challenge: models must balance accuracy with strict inference latency and hardware efficiency constraints. This makes hardware-software co-design a game-changing requirement for on-device LLM deployment, where each hardware platform demands a tailored architectural solution. We propose a hardware co-design law that jointly captures model accuracy and inference performance. Specifically, we model training loss as an explicit function of architectural hyperparameters and characterise inference latency via roofline modelling. We empirically evaluate 1,942 candidate architectures on NVIDIA Jetson Orin, training 170 selected models for 10B tokens each to fit a scaling law relating architecture to training loss. By coupling this scaling law with latency modelling, we establish a direct accuracy-latency correspondence and identify the Pareto frontier for hardware co-designed LLMs. We further formulate architecture search as a joint optimisation over precision and performance, deriving feasible design regions under industrial hardware and application budgets. Our approach reduces architecture selection from months to days. At the same latency as Qwen2.5-0.5B on the target hardware, our co-designed architecture achieves 19.42% lower perplexity on WikiText-2. To our knowledge, this is the first principled and operational framework for hardware co-design scaling laws in on-device LLM deployment. We will make the code and related checkpoints publicly available.

Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Placed in Efficient VLA inference and adaptation, one of 13 directions in Vision-language-action models. Builds on 42 of its 53 references that are in the map. First posted on arXiv 10 Feb 2026.

Builds on

42 of 53 references.

References · 42

  1. 2025Qwen3 Technical Reportcited by 8,195
  2. 2020Scaling Laws for Neural Language Modelscited by 6,411
  3. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  4. 2024Qwen2.5 Technical Reportcited by 4,713
  5. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  6. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  7. 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 2,993
  8. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  9. 2024OpenVLA: An Open-Source Vision-Language-Action Modelcited by 2,777
  10. 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,838
  11. 2018DARTS: Differentiable Architecture Searchcited by 1,782
  12. 2020GLU Variants Improve Transformercited by 1,724
30 more
  1. 2025Gemma 3 Technical Reportcited by 1,686
  2. 2016Neural Architecture Search with Reinforcement Learningcited by 1,489
  3. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
  4. 2023AWQ: Activation-aware Weight Quantization for LLM Compression and Accelerationcited by 1,234
  5. 2017Learning Transferable Architectures for Scalable Image Recognitioncited by 1,097
  6. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
  7. 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 266
  8. 2022Unified Scaling Laws for Routed Language Modelscited by 249
  9. 2024MobileLLM: Optimizing Sub-billion Parameter Language Models for On-Device Use Casescited by 216
  10. 2023PowerInfer: Fast Large Language Model Serving with a Consumer-grade GPUcited by 210
  11. 2025Kimi Linear: An Expressive, Efficient Attention Architecturecited by 169
  12. 2024Scaling Laws for Fine-Grained Mixture of Expertscited by 166
  13. 2023LLM4Drive: A Survey of Large Language Models for Autonomous Drivingcited by 136
  14. 2025LMCache: An Efficient KV Cache Layer for Enterprise-Scale LLM Inferencecited by 119
  15. 2021Scale Efficiently: Insights from Pre-training and Fine-tuning Transformerscited by 105
  16. 2025Vision-Language-Action (VLA) Models: Concepts, Progress, Applications and Challengescited by 72
  17. 2025KVLink: Accelerating Large Language Models via Efficient KV Cache Reusecited by 60
  18. 2025NVIDIA Nemotron 3: Efficient and Open Intelligencecited by 54
  19. 2025Parameters vs FLOPs: Scaling Laws for Optimal Sparsity for Mixture-of-Experts Language Modelscited by 48
  20. 2024"Give Me BF16 or Give Me Death"? Accuracy-Performance Trade-Offs in LLM Quantizationcited by 34
  21. 2025Bridging the Gap Between Promise and Performance for Microscaling FP4 Quantizationcited by 30
  22. 2024A Survey: Collaborative Hardware and Software Design in the Era of Large Language Modelscited by 21
  23. 2025Scaling Inference-Efficient Language Modelscited by 14
  24. 2018Searching Toward Pareto-Optimal Device-Aware Neural Architecturescited by 5
  25. 2025PLM: Efficient Peripheral Language Models Hardware-Co-Designed for Ubiquitous Computingcited by 4
  26. 2025SpecPipe: Accelerating Pipeline Parallelism-based LLM Inference with Speculative Decodingcited by 4
  27. 2025Scaling Laws Meet Model Architecture: Toward Inference-Efficient LLMscited by 4
  28. 2025GTA: Grouped-head latenT Attentioncited by 2
  29. 2025Characterizing Communication Patterns in Distributed Large Language Model Inferencecited by 2
  30. 2025A Hierarchical Test Platform for Vision Language Model (VLM)-Integrated Real-World Autonomous Drivingcited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 5.

5 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Beyond LLM Serving: Characterizing Vision-Language-Action Workloads for Embodied AI System Design
  2. 2026AI Hardware Accelerators for Large Language Models: Architectures and the Memory Wall ↗
  3. 2025Scaling Laws for Energy Efficiency of Local LLMs
  4. 2025Cross-Platform Scaling of Vision-Language-Action Models from Edge to Cloud GPUs
  5. 2025David vs. Goliath: Can Small Models Win Big with Agentic AI in Hardware Design?
  6. 2026Characterizing Vision-Language-Action Models across XPUs: Constraints and Acceleration for On-Robot Deployment
  7. 2026Characterizing VLA Models: Identifying the Action Generation Bottleneck for Edge AI Architectures
  8. 2026EcoVLA: Energy-Efficient Device-Edge Co-Inference for Vision-Language-Action Models under Real-Time Constraints