WriteSAE:面向循环状态的稀疏自编码器
摘要
WriteSAE 引入了第一个稀疏自编码器,能够分解状态空间模型和混合循环语言模型中的矩阵缓存写入,相比现有方法实现了更优的令牌级干预。
查看缓存全文
缓存时间: 2026/05/14 04:17
论文页面 - WriteSAE:面向循环状态的稀疏自编码器
来源:https://huggingface.co/papers/2605.12770
摘要
WriteSAE 实现了对状态空间和混合循环语言模型中矩阵缓存写入的稀疏自编码器分解与编辑,在词元级干预中取得了优于现有方法的性能。
我们介绍了 WriteSAE,这是首个对状态空间和混合循环语言模型的矩阵缓存写入进行分解与编辑的稀疏自编码器,而残差 SAEs 无法达到这些位置。现有的 SAEs 读取残差流,但 Gated DeltaNet、Mamba-2 和 RWKV-7 通过秩为1的更新 k_t v_t^top 写入一个 d_k × d_v 的缓存,任何向量原子都无法替代。WriteSAE 将每个解码器原子分解为原生写入形状,暴露了每个词元对数几率偏移的闭式解,并在匹配的弗罗贝尼乌斯范数下训练,使得原子每次交换一个缓存槽。
原子替换在 Qwen3.5-0.8B L9 H4 的 n=4,851 次触发中,以 92.4% 的比例优于匹配范数消融,87 原子群体测试保持在 89.8%,闭式解以 R²=0.98 预测测量效果,Mamba-2-370M 在 2,500 次触发中替换率为 88.1%。在贪婪解码下,持续的三位置安装将中位目标续写从 33.3% 提升至 100%(提升 3 倍),这是首次在矩阵循环写入位点实现行为安装。
查看 arXiv 页面查看 PDF项目页面GitHub添加到收藏
在您的智能体中获取本文:
hf papers read 2605\.12770
没有最新的 CLI?curl -LsSf https://hf.co/cli/install.sh | bash
引用本文的模型 1
JackYoung27/writesae-ckpts 特征提取 • 更新于大约 2 小时前 (https://huggingface.co/JackYoung27/writesae-ckpts)
引用本文的数据集 0
暂无链接本文的数据集
在数据集 README.md 中引用 arxiv.org/abs/2605.12770 即可从本页链接。
引用本文的 Spaces 0
暂无链接本文的 Space
在 Space README.md 中引用 arxiv.org/abs/2605.12770 即可从本页链接。
包含本文的收藏集 0
暂无包含本文的收藏集
将本文添加至收藏集即可从本页链接。
相似文章
Rational Sparse Autoencoder
介绍理性稀疏自编码器(RSAE),该模型用可训练的有理函数替换固定的编码器激活,在多个基线族开放权重语言模型的残差流激活上改善重建与稀疏性权衡。
稀疏自编码器实现CLIP模型的鲁棒且可解释的微调
SAE-FT提出了一种新颖的CLIP模型微调方法,利用稀疏自编码器约束来正则化视觉表示,在保持性能的同时提高对分布变化的鲁棒性,并实现可解释性。
用于特征发现与长上下文归因的回合平均SAEs
本文介绍了基于回合平均的稀疏自编码器(SAEs),该编码器基于对话回合的平均激活值运行,能够实现长上下文的高效特征发现与归因图。此外,本文还提出了一种嵌套架构,用于与每个词元的特征联合训练。
在应稀疏分解时稀疏分解,在应密集吸收时勿密集吸收
论文假设语言模型激活包含一个低秩密集分量,该分量被稀疏自编码器(SAEs)低效表示。通过添加一个线性瓶颈来吸收密集结构,作者减少了密集潜变量,并改进了在Gemma-2-2B上的稀疏探针性能。
VASAE:使用词汇对齐锚定命名SAE字典方向
本文介绍了词汇对齐稀疏自编码器(VASAE),该方法在词汇对齐锚定下训练SAE特征,基于最近令牌嵌入为每个特征分配内在令牌名称,在早期层实现高对齐而不降低重建质量。