SparseSSM: Efficient Selective Structured State Space Models Can Be Pruned in One-Shot

修剪 冗余(工程) 算法 计算机科学 离散化 灵敏度(控制系统) 国家(计算机科学) 基质(化学分析) 计算复杂性理论 人工智能 组分(热力学) 变压器 状态空间 时间复杂性 电流(流体) 数学 空格(标点符号) 理论计算机科学 状态空间表示 数据挖掘 高效算法 钥匙(锁)
作者
Kaiwen Tuo,Huan Wang
出处
期刊:Cornell University - arXiv [Cornell University]
标识
DOI:10.48550/arxiv.2506.09613
摘要

State-space language models such as Mamba match Transformer quality while permitting linear complexity inference, yet still comprise billions of parameters that hinder deployment. Existing one-shot pruning methods are tailored to attention blocks and fail to account for the time-shared and discretized state-transition matrix at the heart of the selective state-space module (SSM). In this paper, we introduce SparseSSM, the first training-free pruning framework that extends the classic optimal brain surgeon (OBS) framework to state space architectures. Our layer-wise algorithm (i) derives an approximate second-order saliency score that aggregates Hessian-trace information across time steps, (ii) incorporates a component sensitivity analysis to guide feed-forward network (FFN) pruning, which also sheds light on where redundancy resides in mamba architecture, (iii) can be easily extended to semi-structured and structured sparsity. Empirically, we prune 50% of SSM weights without fine-tuning and observe no zero-shot accuracy loss, achieving the current state-of-the-art pruning algorithm for Mamba-based LLMs.
最长约 10秒,即可获得该文献文件

科研通智能强力驱动
Strongly Powered by AbleSci AI
科研通是完全免费的文献互助平台,具备全网最快的应助速度,最高的求助完成率。 对每一个文献求助,科研通都将尽心尽力,给求助人一个满意的交代。
实时播报
peregrine发布了新的文献求助10
1秒前
richelle发布了新的文献求助10
2秒前
3秒前
小力啵啵完成签到 ,获得积分10
4秒前
00完成签到,获得积分10
5秒前
陈开颜发布了新的文献求助10
5秒前
大意的博完成签到,获得积分10
6秒前
不安初阳发布了新的文献求助10
8秒前
天天快乐应助hello采纳,获得10
8秒前
香蕉觅云应助鑫鑫采纳,获得10
8秒前
00发布了新的文献求助20
8秒前
10秒前
10秒前
领导范儿应助俭朴觅松采纳,获得10
10秒前
落后乘风完成签到 ,获得积分10
10秒前
Epilogue完成签到,获得积分10
10秒前
李健应助paperslicing采纳,获得10
10秒前
11秒前
weizhi完成签到,获得积分10
11秒前
义气的夏旋完成签到,获得积分10
12秒前
12秒前
兰亭序完成签到 ,获得积分10
13秒前
CRP完成签到,获得积分10
13秒前
SciGPT应助panpan采纳,获得10
14秒前
万能图书馆应助彭雄武采纳,获得10
15秒前
15秒前
16秒前
zclzclzcl完成签到 ,获得积分10
16秒前
高兴的灰狼完成签到,获得积分10
17秒前
科研通AI6.3应助wangli采纳,获得10
17秒前
三月发布了新的文献求助10
19秒前
恭敬完成签到 ,获得积分20
19秒前
hello发布了新的文献求助10
22秒前
23秒前
杨一天完成签到 ,获得积分10
23秒前
24秒前
24秒前
25秒前
bkagyin应助ghhhn采纳,获得10
26秒前
paperslicing发布了新的文献求助10
27秒前
高分求助中
Markov Chain Monte Carlo 10000
(应助此贴封号)【重要!!请各用户(尤其是新用户)详细阅读】【科研通的精品贴汇总】 10000
Common Foundations of American and East Asian Modernisation: From Alexander Hamilton to Junichero Koizumi 5000
Pediatric Dermoscopy Trichoscopy & Onychoscopy 2030
Matrix Methods in Data Mining and Pattern Recognition Second Edition 610
Handbuch Trainingswissenschaft – Trainingslehre 500
Additive Manufacturing Design and Applications (ASM Handbook, Volume 24A) 500
热门求助领域 (近24小时)
化学 材料科学 医学 生物 纳米技术 工程类 有机化学 化学工程 生物化学 计算机科学 内科学 物理 复合材料 催化作用 细胞生物学 无机化学 光电子学 物理化学 电极 基因
热门帖子
关注 科研通微信公众号,转发送积分 7576942
求助须知:如何正确求助?哪些是违规求助? 9156563
关于积分的说明 19589018
捐赠科研通 7160731
什么是DOI,文献DOI怎么找? 3265177
关于科研通互助平台的介绍 2430231
邀请新用户注册赠送积分活动 2255825