跳到正文
arXiv · cs.CL· Shang Tingyu·· 1 天前AI 评分47

Balanced State Subspaces:面向 Mamba-2 状态压缩的平衡状态子空间方法

Balanced State Subspaces for Mamba-2 State Compression

来源与原文

arXiv · cs.CL

作者:Shang Tingyu

发布时间:(北京时间)

阅读原始来源 · arxiv.org ↗
完整链接与内容说明

https://arxiv.org/abs/2610.36617v1

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

研究者提出一种针对 Mamba-2 的后训练状态压缩变换,发现其状态能量集中在偏置类方向,而输出实际读取的内容位于低维子空间,因此应压缩“写入-读取”子空间而非逐坐标处理。配合按层分配秩并保护检索敏感层的 HARP,该方法将状态维度减半,在 130M–2.7B 模型上 WikiText-2 困惑度平均仅升 0.3 点(1.8%),优于 GHOST 的 1.6 点(9.3%);校准只需一次前向传播,速度比 GHOST 快一个数量级,峰值内存更低。

来源:arXiv · cs.CL · arxiv.org