AI消息速览

STEPQuant:Delta 规则循环状态量化中的错误时空影响

事件日期 2026-09-29 · 学术前沿 · 已接受

事件日期2026-09-29
信息日期2026-09-29
入库日期2026-09-30
通道学术前沿
状态已接受
来源arXiv 论文

知识卡片:STEPQuant:Delta 规则循环状态量化中的错误时空影响

一句话结论

STEPQuant 提出面向 Delta 规则循环状态的时空后训练量化框架,在名义 6-bit 预算下接近 FP32 状态精度,并在 SGLang 集成中实现超过 5 倍循环状态压缩、总服务内存最多降低 68.7%。

事件概述或研究问题

论文研究线性注意力中的内存问题:线性注意力用固定大小的循环状态替代不断增长的 KV cache,但这些持久状态在并发服务下可能成为显著内存瓶颈。直接将这些循环状态量化到低精度,常因量化误差在连续状态更新中传播而导致严重精度退化。作者进一步追问:这些误差在“何时”和“何处”影响更大。

方法/产品要点

  • 作者发现量化误差影响有两个互补维度:
    • 时间维度:长寿命记忆中的错误可跨许多解码步持续存在。
    • 空间维度:不同 key rows 的错误对模型输出的影响不同,同时状态幅度沿行和列都有显著变化。
  • STEPQuant 是一个针对 Delta 规则循环状态的空间-时间后训练量化框架。
  • 它根据误差幅度和记忆寿命分配精度。
  • 它联合拟合 key-row 与 value-column scales,依据是状态分布以及 key-row 对输出误差的影响。
  • 材料还提到该方法集成到 SGLang,并使用优化过的 GPU kernels。

主要结果或产业意义

  • 在 Qwen3.8-27B 和 Kimi-Linear-48B-A3B-Instruct 上,覆盖长生成与短生成 benchmark:
    • 在名义 6-bit 预算下,STEPQuant 接近 FP32-state 精度。
    • 在其 4-bit 配置下,性能超过 uniform INT8。
  • 集成到 SGLang 后,6-bit STEPQuant 实现:
    • 超过 5 倍循环状态压缩。
    • 总服务内存最多降低 68.7%。
  • 代码地址:https://github.com/Dreamer-Toby/STEPQuant

为什么重要

线性注意力试图用固定大小循环状态缓解 KV cache 增长问题,但固定状态本身在并发推理服务中仍可能成为内存瓶颈。STEPQuant 的意义在于把循环状态量化从“统一低比特”推进到“按误差的时空影响分配精度”,并给出与 SGLang 结合的工程结果。与已有相关卡片相比,本条不是多智能体、知识图谱问答或 4D 感知方向,而是 foundation-model 推理内存与量化脉络中的增量信息:它补上了“循环状态量化需要同时考虑时间寿命与空间行/列差异”这一视角。

局限与不确定性

  • 材料未列出具体 benchmark 名称、评估指标、uniform INT8 基线设置、硬件环境和端到端延迟/吞吐数据,待核实。
  • “nominal 6-bit budget”的具体含义、4-bit 配置下与 FP32-state 的具体差距,待核实。
  • 未说明方法在更多模型规模、更长上下文、不同并发规模、不同线性注意力变体上的泛化能力,待核实。
  • 未提供校准成本、显存占用分解、能耗与部署复杂度信息,待核实。
  • 代码虽已给出链接,但其可复现性与实验结果一致性未在材料中验证,待核实。

可用于图书/PPT/简报的角度

  • 从 KV cache 到固定大小循环状态:内存瓶颈如何转移。
  • 量化误差传播:为什么均匀低比特可能不适合循环状态。
  • 时空敏感量化:按记忆寿命与 key-row 影响力分配精度。
  • 推理服务优化:SGLang 集成、GPU kernels 与总服务内存下降。
  • 后训练量化在新型序列模型状态上的新问题。
  • 学术论文与推理框架工程结合、推动部署成本下降的案例。

原始材料

  • 英文标题:STEPQuant: When and Where Errors Matter in Delta-Rule Recurrent State Quantization
  • 作者:Bingchen Yao, Haobo Xu, Haokun Lin, Yichen Wu, Ziyu Guo, Renrui Zhang, Zhichao Lu, Zhenan Sun, Ying Wei
  • 发表/更新:2026-09-29T17:59:40Z
  • 主分类:cs.CL;分类:cs.CL, cs.AI, cs.LG
  • arXiv ID:2609.38169v1
  • Abstract URL:https://arxiv.org/abs/2609.38169v1
  • PDF URL:https://arxiv.org/pdf/2609.38169v1
  • 代码:https://github.com/Dreamer-Toby/STEPQuant
  • 英文关键词:材料未单独列出关键词;据标题与摘要保留为:Delta-Rule Recurrent State Quantization; Linear Attention; Post-Training Quantization; Spatial-Temporal; SGLang。

发现内容有误?

免登录提交,管理员审核并完成修正后才会公开修订说明。

值得关注

线性注意力试图用固定大小循环状态缓解 KV cache 增长问题,但固定状态本身在并发推理服务中仍可能成为内存瓶颈。STEPQuant 的意义在于把循环状态量化从“统一低比特”推进到“按误差的时空影响分配精度”,并给出与 SGLang 结合的工程结果。与已有相关卡片相比,本条不是多智能体、知识图谱问答或 4D 感知方向,而是 foundation-mode…

相关主题

对应的教材知识点

教材知识点匹配尚未完成。