State Space Models (SSMs) such as Mamba and Mamba-2 achieve linear-time autoregressive inference, making them attractive for latency-sensitive and resource-constrained deployment. Yet every generated token still reads the full input and output projection weights of every layer from off-chip memory, and on an FPGA these two projections take the majority of a layer's cycles. Existing accelerators lower this cost by quantizing the weights or by speeding up the scan, but none reduce the number of weight elements a token must read, and a low-rank factorization on its own does not help: without a datapath built for it, the two-stage product takes as long as the dense one. We present \textbf{LowRank-SSM}, a hardware-software co-design framework that jointly addresses three constraints of rank-reduced Mamba inference: where rank should be spent, how the resulting factors should execute, and how reduced weight traffic should be delivered by the FPGA memory system. A sensitivity-guided per-projection search allocates rank under a perplexity ceiling while restricting compression to Mamba-2 operands that tolerate approximation. A rank-configurable projection datapath maps both factor stages onto the existing GEMV engines, keeping the intermediate vector on chip and changing rank through loop bounds and weight images rather than bitstream recompilation. A platform-aware NoC/AXI memory subsystem then balances and prefetches weight streams to sustain 42\,GB/s. Together, these software and hardware choices convert structural rank reduction into measured latency reduction: in our ablation, 93\% of the projection traffic removed by rank reduction appears as saved token latency. With a rank schedule found under a 5\% perplexity ceiling, which costs 1.6 points of average zero-shot accuracy, the complete accelerator decodes Mamba2-2.7B at 16.79 tokens/s and 0.41\,J per token.


翻译:暂无翻译

0
下载
关闭预览

相关内容

ICML 2026 Oral | MuonSSM:用正交化记忆更新稳定SSM
专知会员服务
13+阅读 · 6月30日
非Transformer不可?最新《状态空间模型(SSM)》综述
专知会员服务
75+阅读 · 2024年4月16日
超全汇总 | ORB-SLAM2 / ORB-SLAM3 相关改进代码!
计算机视觉life
35+阅读 · 2020年11月22日
基于 Carsim 2016 和 Simulink的无人车运动控制联合仿真(四)
meta learning 17年:MAML SNAIL
CreateAMind
11+阅读 · 2019年1月2日
用 LDA 和 LSA 两种方法来降维和做 Topic 建模
AI研习社
13+阅读 · 2018年8月24日
超全总结:神经网络加速之量化模型 | 附带代码
IJCAI | Cascade Dynamics Modeling with Attention-based RNN
KingsGarden
13+阅读 · 2017年7月16日
From Softmax to Sparsemax-ICML16(1)
KingsGarden
74+阅读 · 2016年11月26日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
VIP会员
最新内容
俄罗斯如何拦截乌克兰无人机与巡航导弹
专知会员服务
8+阅读 · 10月15日
《武器-目标分配问题:精确与近似求解算法》
专知会员服务
11+阅读 · 10月15日
综述|谁来改进机器人?迈向物理递归自我改进
专知会员服务
10+阅读 · 10月14日
软件定义无人机蜂群给防空带来全新压力
专知会员服务
10+阅读 · 10月13日
《半自主无人地面战车的互操作性》66页
专知会员服务
14+阅读 · 10月13日
相关资讯
超全汇总 | ORB-SLAM2 / ORB-SLAM3 相关改进代码!
计算机视觉life
35+阅读 · 2020年11月22日
基于 Carsim 2016 和 Simulink的无人车运动控制联合仿真(四)
meta learning 17年:MAML SNAIL
CreateAMind
11+阅读 · 2019年1月2日
用 LDA 和 LSA 两种方法来降维和做 Topic 建模
AI研习社
13+阅读 · 2018年8月24日
超全总结:神经网络加速之量化模型 | 附带代码
IJCAI | Cascade Dynamics Modeling with Attention-based RNN
KingsGarden
13+阅读 · 2017年7月16日
From Softmax to Sparsemax-ICML16(1)
KingsGarden
74+阅读 · 2016年11月26日
相关基金
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
Top
微信扫码咨询专知VIP会员