SQS: Bayesian DNN Compression through Sparse Quantized Sub-distributions
SQS:基于稀疏量化子分布的贝叶斯 DNN 压缩
🧭 模型压缩 · 贝叶斯变分推断
Purdue University · University of Texas at El Paso
用 spike-and-slab 先验与高斯混合模型统一贝叶斯剪枝与量化,压缩率更高且精度损失相当
📌 工作参考价值(4 点)
- 统一压缩替代两阶段:电商/广告排序塔及 LLM-based reranker 对推理延迟和内存敏感,可借鉴 SQS 把稀疏化和量化作为同一个变分学习问题,避免先剪枝再量化带来的误差累积;对 embedding 层、FFN 层可分别设置稀疏率和码本宽度。
- 可微的先验/近似方案:spike-and-slab 先验天然支持非结构化剪枝,GMM 后验提供量化码本均值,整体保持可微优化;工程落地时可用更轻量的近似替代复杂积分。
- 面向 LLM 部署可迁移:论文在 Llama3.2、Qwen2.5 上验证,说明适合本地部署的 embedding 模型、轻量生成式推荐模型;可按层或 token 粒度做联合压缩,降低 CPU/边缘推理成本。
- 注意:低 bit 量化依赖硬件算子和量化校准,需结合 TensorRT/ONNX 等推理框架;论文偏学术方法,业务收益还需线上延迟和精度实测。