arXiv:2508.20122cs.NEcs.ET2025-08被引 1

通过时空剪枝压缩脉冲语言模型,实现低功耗实时推理。

Spatio-Temporal Pruning for Compressed Spiking Large Language Models

  • 同时剪枝神经元和时间步数,降低计算开销。
  • 在GLUE基准上减少计算量并缩短推理延迟。
  • 适合边缘设备部署的低功耗自然语言处理场景。

大型语言模型(LLMs)因模型规模大、推理延迟高,在能源受限环境中的部署面临挑战。脉冲神经网络(SNNs)受大脑稀疏事件驱动和节能信息传输机制启发,为实现低功耗计算提供了可行路径。将脉冲神经元的事件驱动高效性与LLMs的先进能力结合,是实现节能型LLMs的重要方向。本文聚焦压缩型脉冲语言模型的设计,从SNN视角重新审视空间与时间剪枝,提出一种新的时空剪枝框架,以优化计算效率并保持高性能。空间剪枝减少活跃神经元和注意力头数量,有效降低模型计算复杂度;时间剪枝则通过动态调整各层所需时间步数,最小化推理延迟。结合其他压缩技术,首次在脉冲语言模型领域联合探索了空间剪枝、时间剪枝、极量化及知识蒸馏策略。在大规模GLUE基准上对SpikingBERT的实验评估表明,该方法在计算操作数和推理延迟方面均具显著优势。本方案为实时、低功耗自然语言处理应用提供了有力解决方案,使脉冲语言模型更适用于边缘设备与能源受限场景。

原文摘要 · Abstract (English)

Large Language Models (LLMs) present significant challenges for deployment in energy-constrained environments due to their large model sizes and high inference latency. Spiking Neural Networks (SNNs), inspired by the sparse event-driven neural processing and energy-efficient information transmission in the brain, offer a promising alternative for achieving low-power computing. Integrating the event-driven efficiency of spiking neurons with the advanced capabilities of LLMs represents a promising direction for power-efficient LLMs. This work specifically delves into the design of compressed spiking LLMs. Here, we revisit spatial and temporal pruning from the perspective of SNNs and propose a novel spatio-temporal pruning framework for Spiking LLMs to optimize computational efficiency while preserving high performance. Our spatial pruning technique reduces the number of active neurons and attention heads, effectively lowering the computational complexity of the model. Meanwhile, temporal pruning minimizes inference latency by dynamically adjusting the number of timesteps required for different layers. By combining these approaches with other compression techniques, we present the first work in the domain of Spiking LLMs to jointly explore spatial pruning, temporal pruning, extreme quantization and knowledge distillation strategies. Extensive experimental evaluation of our proposed framework for SpikingBERT on the large-scale GLUE benchmark demonstrates the efficacy of our approach in terms of computational operations and inference latency. Our approach offers a compelling solution for real-time, low-power natural language processing applications, making Spiking LLMs more practical for deployment on edge devices and in power-constrained settings.

脉冲神经网络模型压缩低功耗语言模型

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