🤖 AI Summary
To address three critical challenges in edge deployment of large language models (LLMs)—high energy consumption under quantization, substantial dequantization overhead, and significant value overflow—this paper proposes SpikeQuant, the first method to couple the integrate-and-fire dynamics of spiking neural networks (SNNs) with quantization scaling. Its key contributions are: (1) implicit embedding of quantization scaling via binary spike coding, eliminating explicit dequantization; (2) channel-wise dynamic mixed-precision quantization driven by spike firing rates, mitigating severe truncation of salient values caused by uniform low-bit quantization; and (3) replacement of conventional multiply-accumulate (MAC) operations with threshold-modulated linear integration. Under a W4A4 configuration, SpikeQuant achieves language modeling perplexity comparable to FP16, while delivering up to 4.6× higher energy efficiency than state-of-the-art quantization methods—significantly improving the accuracy–efficiency trade-off.
📝 Abstract
In the era of large language models (LLMs), weight-activation quantization helps fit models on edge device by reducing memory and compute bit-widths. However, three challenges persist for energy constrained hardware: (1) even after quantization, multiply-accumulate (MAC) operations remain unavoidable and continue to dominate energy consumption; (2) dequantization (or per-tensor/channel rescaling) introduces extra arithmetic and data movement, increasing latency and energy; (3) uniform parameters bit widths clip salient values-while intra-channel mixed precision is generally impractical on current matrix hardware and memory. In contrast, brain-inspired Spiking Neural Networks (SNNs), owing to their binary spike-based information representation and the Integrate-and-Fire (IF) paradigm, naturally support mixed-precision storage and energy-efficient computation by replacing complex MACs with temporal Accumulate (ACCs). Motivated by this property, we propose SpikeQuant, which selectively applies mixed-precision quantization to activations with salient values and re-encodes them into binary spike counts, thereby enabling dynamic mixed storage of different bitwidths. Furthermore, by embedding the quantization scale into the threshold of the IF mechanism, our approach performs energy-efficient linear transformations on weights and activations while avoiding explicit dequantization. Experimental results demonstrate that SpikeQuant consistently achieves near-FP16 perplexity under W4A4 quantization while reducing energy cost by up to 4.6 times compared to existing methods, highlighting its effectiveness for accurate and energy-efficient LLM deployment.