跳到正文
原文
Transformer Circuits(Anthropic 可解释性)·· 2024-04-26

Anthropic 可解释性团队 4 月更新:SAE 训练改进与字典学习缩放规律

Circuits Updates — April 2024

AI 导读

Anthropic 可解释性团队发布 2024 年 4 月更新,汇总多项尚在推进中的研究。在字典学习缩放规律方面,团队以四层 transformer 第三层后的残差流为案例,发现损失随 FLOPS 近似按幂律下降,最优特征数与训练步数也各自按幂律随算力增长,且最优特征数增长更快;对 L0 类损失函数需要更多训练步。在 SAE 训练方面,团队改进了自 Towards Monosemanticity 以来的训练设置,主要变化是解码器列向量不再约束为 L2 范数 1,稀疏惩罚项改为包含解码器列向量的 L2 范数,并给出学习率 5e-5、训练步数 200k、batch size 2048 或 4096、λ 默认 5 等默认值。

阅读原文transformer-circuits.pub