arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.05232cs.LG
收藏

SlideSparse: 快速且灵活的(2N-2):2N结构稀疏

SlideSparse: Fast and Flexible (2N-2):2N Structured Sparsity

Peking University(北京大学);South China University of Technology(南方科技大学);Microsoft Research(微软研究院);Shanghai Jiao Tong University(上海交通大学);The Hong Kong University of Science and Technology(香港科技大学)

Hanyong Shao, Yingbo Hao, Ting Song, Yan Xia, Di Zhang, Shaohan Huang, Xun Wu, Songchen Xu, Le Xu, Li Dong, Zewen Chi, Yi Zou, Furu Wei

AI总结:

SlideSparse通过滑动窗口分解和激活提升技术,在通用GPU上实现(2N-2):2N结构稀疏的高效加速,提升LLM推理效率的同时保持准确性。

摘要:

NVIDIA的2:4稀疏张量核心提供2倍的吞吐量,但要求严格的50%剪枝——这种比率会破坏LLM推理准确性(Qwen3: 54%到15%)。更温和的(2N-2):2N模式(例如6:8,25%剪枝)保持了准确性,但没有硬件支持,不得不回退到密集执行,没有任何稀疏性的优势。我们提出了SlideSparse,这是第一个系统,能够在通用GPU上解锁(2N-2):2N模型家族的稀疏张量核心加速。我们的滑动窗口分解将任何(2N-2):2N权重块分解为N-1个重叠的2:4兼容窗口,而不会有任何精度损失;激活提升将相应的激活重排融合到每token量化中,成本接近于零。集成到vLLM中,SlideSparse在各种GPU(A100,H100,B200,RTX 4090,RTX 5080,DGX-spark)、精度(FP4,INT8,FP8,BF16,FP16)和模型家族(Llama,Qwen,BitNet)上进行评估。在计算受限的工作负载中,测得的加速比(1.33x)接近理论上限N/(N-1)=4/3,在Qwen2.5-7B中的6:8权重稀疏性下,确立了(2N-2):2N作为准确保持LLM加速的实用路径。代码可在https://github.com/bcacdwk/vllmbench获取。

英文摘要:

NVIDIA's 2:4 Sparse Tensor Cores deliver 2x throughput but demand strict 50% pruning -- a ratio that collapses LLM reasoning accuracy (Qwen3: 54% to 15%). Milder $(2N-2):2N$ patterns (e.g., 6:8, 25% pruning) preserve accuracy yet receive no hardware support, falling back to dense execution without any benefit from sparsity. We present SlideSparse, the first system to unlock Sparse Tensor Core acceleration for the $(2N-2):2N$ model family on commodity GPUs. Our Sliding Window Decomposition reconstructs any $(2N-2):2N$ weight block into $N-1$ overlapping 2:4-compliant windows without any accuracy loss; Activation Lifting fuses the corresponding activation rearrangement into per-token quantization at near-zero cost. Integrated into vLLM, SlideSparse is evaluated across various GPUs (A100, H100, B200, RTX 4090, RTX 5080, DGX-spark), precisions (FP4, INT8, FP8, BF16, FP16), and model families (Llama, Qwen, BitNet). On compute-bound workloads, the measured speedup ratio (1.33x) approaches the theoretical upper-bound $N/(N-1)=4/3$ at 6:8 weight sparsity in Qwen2.5-7B, establishing $(2N-2):2N$ as a practical path to accuracy-preserving LLM acceleration. Code available at https://github.com/bcacdwk/vllmbench.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑