arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.17088cs.LGcs.AIcs.CL

Mamba-Shedder:面向高效选择性结构化状态空间模型的Transformer后压缩

Mamba-Shedder: Post-Transformer Compression for Efficient Selective Structured State Space Models

Intel Labs(英特尔实验室);Intel Corporation(英特尔公司)

J. Pablo Muñoz, Jinjie Yuan, Nilesh Jain

AI总结:

本文提出Mamba-Shedder压缩方法,通过研究不同粒度移除Mamba等SSM模型组件的敏感性,在保持精度的同时减小规模与开销,实现高达1.4倍推理加速。

摘要:

大型预训练模型在序列建模中取得了卓越成果。Transformer块及其注意力机制是这些模型成功的主要驱动力。近期,选择性结构化状态空间模型(SSMs)等替代架构被提出,以解决Transformer的效率低下问题。本文探讨了基于SSM模型的压缩,特别是Mamba及其混合模型。我们研究了这些模型在不同粒度上移除选定组件的敏感性,以减小模型规模和计算开销,从而在保持准确性的同时提升效率。提出的解决方案统称为Mamba-Shedder,在推理期间实现了高达1.4倍的加速,证明可通过消除多种冗余来提升模型效率,且对整体性能影响极小。代码见https://github.com/IntelLabs/Hardware-Aware-Automated-Machine-Learning。

英文摘要:

Large pre-trained models have achieved outstanding results in sequence modeling. The Transformer block and its attention mechanism have been the main drivers of the success of these models. Recently, alternative architectures, such as Selective Structured State Space Models (SSMs), have been proposed to address the inefficiencies of Transformers. This paper explores the compression of SSM-based models, particularly Mamba and its hybrids. We study the sensitivity of these models to the removal of selected components at different granularities to reduce the model size and computational overhead, thus improving their efficiency while maintaining accuracy. The proposed solutions, collectively referred to as Mamba-Shedder, achieve a speedup of up to 1.4x during inference, demonstrating that model efficiency can be improved by eliminating several redundancies with minimal impact on the overall model performance. The code is available at https://github.com/IntelLabs/Hardware-Aware-Automated-Machine-Learning.

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

↑