arXiv:2509.23324cs.DCcs.AI2025-09被引 14

用手机NPU的闲置算力,让小模型跑出大模型效果

Scaling LLM Test-Time Compute with Mobile NPU on Smartphones

  • 在手机NPU上实现并行测试时扩展,提升小模型性能
  • 小模型经扩展后可达到甚至超过大模型准确率
  • 适配移动端算力瓶颈,适合移动AI部署场景

将大语言模型(LLM)部署在移动设备面临小模型性能不足、大模型资源消耗过大的困境。本文指出,移动神经网络处理单元(NPU)在常规LLM推理中存在未充分利用的计算资源,尤其是矩阵乘法单元。为利用这部分闲置算力,我们提出在移动NPU上应用并行测试时扩展技术,以增强小模型性能。然而,该方法面临硬件限制:缺乏对细粒度量化支持,通用计算效率低。为此,我们设计了两种关键技术:一种与NPU内存访问模式对齐的硬件感知分块量化方案,以及基于查找表(LUT)的Softmax和反量化操作高效替代。我们构建了一个端到端推理系统,可在高通骁龙平台上实现测试时扩展。实验显示,该方法带来显著加速:混合精度GEMM最高达19.0倍,Softmax达2.2倍。更重要的是,使用测试时扩展的小模型可达到或超越大模型的准确率,开辟了新的性能-成本帕累托前沿。

原文摘要 · Abstract (English)

Deploying Large Language Models (LLMs) on mobile devices faces the challenge of insufficient performance in smaller models and excessive resource consumption in larger ones. This paper highlights that mobile Neural Processing Units (NPUs) have underutilized computational resources, particularly their matrix multiplication units, during typical LLM inference. To leverage this wasted compute capacity, we propose applying parallel test-time scaling techniques on mobile NPUs to enhance the performance of smaller LLMs. However, this approach confronts inherent NPU challenges, including inadequate hardware support for fine-grained quantization and low efficiency in general-purpose computations. To overcome these, we introduce two key techniques: a hardware-aware tile quantization scheme that aligns group quantization with NPU memory access patterns, and efficient LUT-based replacements for complex operations such as Softmax and dequantization. We design and implement an end-to-end inference system that leverages the NPU's compute capability to support test-time scaling on Qualcomm Snapdragon platforms. Experiments show our approach brings significant speedups: up to 19.0 for mixed-precision GEMM and 2.2 for Softmax. More importantly, we demonstrate that smaller models using test-time scaling can match or exceed the accuracy of larger models, achieving a new performance-cost Pareto frontier.

移动AINPU优化测试时扩展量化

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