arXiv:2608.27513cs.LGcs.AI2026-08

提出新量化方法,让大模型推理更省内存更快

DAMP: Decay-Aware Mixed-Precision Recurrent-State Quantization

论文配图:DAMP: Decay-Aware Mixed-Precision Recurrent-State Quantization
图 1 · 摘自论文原文
  • 根据误差能量和状态衰减强度识别高风险通道,分层量化
  • 9.9比特下保持接近原精度的推理准确率
  • 减少69%状态存储,提速近2倍,适合部署大模型

Softmax注意力机制需存储每个先前标记的键值向量,导致推理内存随序列长度增长。采用带门控的DeltaNet(GDN)或Kimi Delta注意力(KDA)的模型通过将多数层中的KV缓存替换为固定大小的循环状态,降低了这一开销。然而,这些循环状态通常以FP32存储,占用大量GPU内存;其更新受内存带宽限制,显著增加解码延迟。据我们所知,这是首个研究基于GDN与KDA的语言模型中循环状态后训练量化的工作。发现均匀量化在复杂推理任务上精度下降严重:INT8与FP8已造成性能损失,而INT4和NVFP4使精度近乎归零。进一步发现,大部分量化误差能量集中于少数通道,且状态通道的相对衰减速率在不同提示和任务间保持稳定。受此启发,DAMP在离线校准阶段结合量化误差能量与衰减持续性识别高风险通道,将其保留更高精度,其余通道采用INT8存储。我们在Qwen3.6-35B和Kimi-Linear-48B上评估了DAMP,覆盖数学推理、通用推理与代码生成六项基准。在每状态值9.9比特时,平均精度接近FP32基线。DAMP将循环状态存储减少69.1%,加速循环状态更新核函数最高达2.01倍,全模型TPOT降低最多10.9%。

原文摘要 · Abstract (English)

Softmax attention stores key and value vectors for every preceding token, causing inference memory to grow with sequence length. Recent language models incorporating Gated DeltaNet (GDN) or Kimi Delta Attention (KDA) reduce this cost by replacing the KV cache in most layers with fixed-size recurrent states. However, these recurrent states are commonly stored in FP32 and consume substantial GPU memory; their updates are memory-bandwidth bound and contribute significantly to decoding latency. To our knowledge, we are the first to study post-training quantization of recurrent states in GDN and KDA based language models. We find that uniform quantization provides a poor accuracy--storage trade-off: INT8 and FP8 already degrade accuracy on complex reasoning tasks, while INT4 and NVFP4 reduce it to near zero. We further find that most quantization-error energy is concentrated in a small subset of channels and that the relative decay strength of state channels remains stable across prompts and tasks. Motivated by these findings, DAMP uses both quantization-error energy and decay-based persistence to identify high-risk channels during offline calibration. It stores these channels at higher precision and the remainder in INT8. We evaluate DAMP on Qwen3.6-35B and Kimi-Linear-48B across six benchmarks covering mathematical reasoning, general reasoning, and code generation. At 9.9 bits per state value, DAMP maintains average accuracy close to the FP32 baseline. DAMP reduces recurrent-state storage by 69.1%, accelerates the recurrent-state update kernel by up to 2.01x, and lowers full-model TPOT by up to 10.9%.

量化大模型推理内存优化

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