arXiv:2601.00679cs.NEcs.AI2026-01中稿 · the Design, Automa…被引 3

提出自动量化框架,让脉冲语言模型在低资源设备上高效运行

QSLM: A Performance- and Memory-aware Quantization Framework with Tiered Search Strategy for Spike-driven Language Models

  • 分层搜索策略结合多目标优化,自动找到最佳量化配置
  • 内存压缩达86.5%,功耗降低20%,性能损失极小
  • 适合部署于嵌入式设备的轻量级脉冲语言模型

大型语言模型虽性能优异,但计算成本高、内存占用大,难以在资源受限的嵌入式设备上部署。近期提出的脉冲驱动语言模型(SLMs)虽显著降低能耗,但内存仍过大。手动量化虽可压缩内存,但耗时长、不可扩展。为此,本文提出QSLM框架,通过分析网络层级与层敏感性,采用分层量化策略(全局、块、模块级),结合性能-内存多目标权衡函数,自动选择最优量化方案。实验表明,QSLM可将内存压缩86.5%,功耗降低20%,在不同任务中保持高性能:如在SST-2数据集上情感分类准确率达原模型的84.4%,在WikiText-2上文本生成困惑度为23.2,满足性能与内存约束。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have been emerging as prominent AI models for solving many natural language tasks due to their high performance (e.g., accuracy) and capabilities in generating high-quality responses to the given inputs. However, their large computational cost, huge memory footprints, and high processing power/energy make it challenging for their embedded deployments. Amid several tinyLLMs, recent works have proposed spike-driven language models (SLMs) for significantly reducing the processing power/energy of LLMs. However, their memory footprints still remain too large for low-cost and resource-constrained embedded devices. Manual quantization approach may effectively compress SLM memory footprints, but it requires a huge design time and compute power to find the quantization setting for each network, hence making this approach not-scalable for handling different networks, performance requirements, and memory budgets. To bridge this gap, we propose QSLM, a novel framework that performs automated quantization for compressing pre-trained SLMs, while meeting the performance and memory constraints. To achieve this, QSLM first identifies the hierarchy of the given network architecture and the sensitivity of network layers under quantization, then employs a tiered quantization strategy (e.g., global-, block-, and module-level quantization) while leveraging a multi-objective performance-and-memory trade-off function to select the final quantization setting. Experimental results indicate that our QSLM reduces memory footprint by up to 86.5%, reduces power consumption by up to 20%, maintains high performance across different tasks (i.e., by up to 84.4% accuracy of sentiment classification on the SST-2 dataset and perplexity score of 23.2 for text generation on the WikiText-2 dataset) close to the original non-quantized model while meeting the performance and memory constraints.

脉冲模型量化压缩嵌入式部署

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