Sparser, Faster, Lighter Transformer Language Models
提出一种新的稀疏打包格式和 CUDA kernel 集合,专门针对 Transformer FFN 层的非结构化稀疏进行高效计算。通过简单的 L1 正则化即可在 FFN 层诱导 99% 以上的稀疏度,且对下游性能几乎无损。配合定制 kernel,稀疏化在推理和训练中均带来显著的吞吐量提升、能效改善和内存节省,且收益随模型规模增长而增大。
INSIGHT该工作的核心发现令人意外:简单的 L1 正则化就能在 FFN 层实现极端稀疏(99%+)而不损失性能,关键是配套的 CUDA kernel 能将理论稀疏转化为实际加速。对搜索/推荐系统中部署的 LLM 而言,这提供了一条不需要改变模型架构、仅通过正则化+定制 kernel 即可降低推理成本的实用路径。Llion Jones(Transformer 原作者之一)参与使本文可信度较高。