arXiv:2607.08786cs.LGcs.AI2026-07

用中等稀疏权重矩阵加速大模型推理,首次在现代GPU上超越密集计算。

Accelerating GPU Inference of Large Language Models with Moderately Unstructured Sparse Weight Matrices

  • 设计三层稀疏存储格式,融合稀疏张量核与CUDA核心
  • 在50%左右稀疏度下实现1.64倍内核加速和1.41倍端到端提速
  • 适合部署大模型且关注推理效率的工程团队使用

随着大语言模型(LLMs)部署增多,其推理成本成为关键挑战。通过剪枝引入权重重矩阵稀疏性可加速推理,但为保持模型质量,通常仅限于约50%的中等非结构化稀疏度。在此稀疏水平下,现有GPU稀疏矩阵乘法(SpMM)内核均无法超越稠密实现。本文提出一种面向中等稀疏度的高效GPU推理方法。设计三层次矩阵存储格式:(i) Sparse-TC层利用稀疏张量核心加速SpMM;(ii) Slot-Filling层通过并行差分距离实现压缩,支持低成本片上解码;(iii) 轻量级残差层确保SpMM计算正确性。基于此格式,构建联合使用稀疏张量核心与CUDA核心的SpMM内核,实现高效执行流水线,并重叠片上计算与内存访问。实验表明,本工作首次在配备高带宽内存(HBM)的现代GPU上超越稠密矩阵乘法。相比SpInfer(EuroSys'25,最佳论文),内核层面最高提速1.64倍;相比FlashLLM(VLDB'24),端到端最高提速1.41倍。源码见:https://github.com/moui0/cudac。

原文摘要 · Abstract (English)

With the growing deployment of large language models (LLMs), LLM inference cost has become a key challenge. Pruning techniques that introduce sparsity into weight matrices can accelerate inference. However, maintaining model quality typically limits pruning to moderate unstructured sparsity (around 50\%). At these sparsity levels, none of the existing GPU kernels for sparse matrix multiplication (SpMM) can outperform their dense counterparts. This paper proposes an efficient GPU inference method for LLMs with moderate sparsity. We propose a three-layer matrix storage format comprising: (i) a Sparse-TC layer enabling sparse tensor cores to accelerate SpMM; (ii) a Slot-Filling layer using parallel differential distance for matrix compression while supporting low-cost on-chip decoding; (iii) a lightweight Residual Layer ensuring correct SpMM computation. Building on this format, we design a SpMM kernel that jointly utilizes sparse tensor cores and CUDA cores. This design enables an efficient execution pipeline and overlaps on-chip computation with memory access. Evaluations show that our work is the first to outperform dense matrix multiplication on modern GPUs equipped with high-bandwidth memory (HBM). It achieves up to 1.64x kernel-level speedup over SpInfer (EuroSys'25, Best paper) and up to 1.41x end-to-end speedups over FlashLLM (VLDB'24). Our source code: https://github.com/moui0/cudac.

大模型推理稀疏加速GPU优化张量核

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。