菜单

关于 🐙 GitHub
arXiv 提交日期: 2026-06-09
📄 Abstract - SpenseGPT: Practical One-shot Pruning Enabling Sparse and Dense GEMMs for LLM Inference

Semi-structured 2:4 sparsity is widely supported by modern accelerators, providing up to a 2x theoretical speedup. However, its strict 50% sparsity constraint often causes non-negligible accuracy degradation under post-training pruning. Meanwhile, existing relaxed sparsity formats either require specialized compiler support or introduce runtime overheads that limit end-to-end speedup. We propose Spense, a practical hybrid sparse-dense format that splits each weight matrix into a 2:4 sparse region and a dense region. This design relaxes the effective sparsity constraint while remaining compatible with existing high-performance sparse and dense GEMM libraries, avoiding both custom compiler support and input activation expansion. Building on this format, we introduce SpenseGPT, a one-shot post-training pruning method that produces sparse and dense regions. Notably, we show that selecting the right dense regions is important, and we devise two different strategies to choose them. Experiments on Qwen3-32B and Seed-OSS-36B demonstrate that our method achieves up to 1.2x end-to-end decoding speedup on B200 GPUs with FP8 precision, while preserving accuracy. To the best of our knowledge, this is the first one-shot pruning demonstration of real-world end-to-end LLM decoding speedup from semi-structured sparse tensor cores on recent GPUs such as B200s, while maintaining model quality.

顶级标签: llm model training systems
详细标签: sparse gemm post-training pruning 2:4 sparsity llm inference gpu acceleration 或 搜索:

SpenseGPT:一种实用的一次性剪枝方法,实现大语言模型推理中的稀疏与稠密矩阵乘 / SpenseGPT: Practical One-shot Pruning Enabling Sparse and Dense GEMMs for LLM Inference


1️⃣ 一句话总结

本文提出了一种新方法SpenseGPT,通过将权重矩阵拆分为稀疏和稠密两部分,在保持与现有高效计算库兼容的前提下,放宽了半结构化稀疏的限制,从而在不牺牲模型精度的情况下,在B200 GPU上实现了最高1.2倍的大语言模型推理加速。

源自 arXiv: 2606.10445