跳到正文
原文
arXiv 机器学习· Tianxiang Xia, Kai Lion, Niao He, Michael Muehlebach, Melanie Zeilinger, Hao Ma·· 5 小时前AI 评分38

大模型结构化稀疏学习:连接模型可压缩性与高效推理

Learning Structured Sparsity for Large Models: Bridging Model Compressibility and Efficient Inference

AI 导读

研究提出统一框架 S-SALAAD,在训练中联合学习模型权重及其低秩与结构化稀疏组件,结合专用 CPU 和 GPU 内核以加速推理。该框架支持将单一检查点弹性压缩至多种内存预算,解决非结构化稀疏的索引开销难题。在 Llama2 与 Qwen3 的预训练及知识蒸馏实验中,S-SALAAD 在困惑度、内存占用和推理延迟上推进了帕累托前沿。

来源:arXiv 机器学习 · arxiv.org