端到端稀疏字典学习:让 SAE 只学对网络真正重要的特征

burny_tech · x · 2026-10-20

作者介绍 arXiv 论文《Identifying Functionally Important Features with End-to-End Sparse Dictionary Learning》。标准稀疏自编码器(SAE)重建网络内部激活,但可能学到的更多是数据集结构而非网络本身的计算结构。论文提出端到端(e2e)稀疏字典学习:以「插入 SAE 激活后模型与原模型输出分布的 KL 散度」为训练目标,保证学到的特征对网络功能确实重要。结果显示 e2e SAE 相比标准 SAE 是帕累托改进:解释更多网络性能、所需总特征数更少、每个样本同时激活的特征更少,且不损失可解释性。训练库与复现代码已开源。

所属事件:端到端稀疏字典学习让SAE聚焦网络重要特征(2 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →