SparseDecoding: Decoding-Aware Pruning for Accurate and Efficient LLM Inference
解码感知剪枝:面向 LLM 高效推理的 SparseDecoding
🧭 LLM 推理剪枝与稀疏计算优化
Westlake University · Ant Group
提出解码感知的剪枝框架,用生成过程激活校准并优化N:M稀疏SpMV内核,实现1.48x解码加速
📌 工作参考价值(3 点)
- 校准数据选择:剪枝/量化 LLM 时,让 dense 模型自回归生成并从各层收集激活作为校准集,而非离线静态文本,能缓解分布偏移导致的性能下降;在电商推荐理由生成、Agent 长回复等 decode 为主场景尤其适用。
- 系统内核优化:线上 LLM 以 decode 为主时,应重点优化 SpMV;采用 N:M 结构化稀疏配合 bitmask 索引与固定步长遍历,可有效降低 memory-bound 延迟,在商品描述生成、对话式推荐等长输出服务中提升吞吐。
- 训练后剪枝无需微调,工程成本低,适合对已有搜索推荐 LLM 服务做快速压缩迭代,保留任务性能的同时降低 A100/H100 推理成本。