无需参数调整,显著提升稀疏自编码器的特征质量与稳定性
Aligned Training: A Parameter-Free Method to Improve Feature Quality and Stability of Sparse Autoencoders (SAE)

- 通过强制编码器与解码器方向对齐,消除训练中的退化问题
- 在多个模型和稀疏度下均实现重建精度、特征激活率与稳定性的全面提升
- 适用于机械可解释性方法,无额外计算开销,适合研究者直接使用
稀疏自编码器(SAE)是解析深度神经网络内部机制的重要工具,能将激活分解为高维特征。然而,现有方法存在大量特征从未被激活且不稳定的问题。尽管已有多种变体尝试缓解此问题,但需额外数据、重采样或训练。本文提出“对齐训练”(aligned training),一种无需参数的重参数化方法,可同时提升重建质量、消除死特征,并显著增强跨训练种子的稳定性。该方法基于一个被忽视的现象:现代架构中SAE特征质量(以编码器与解码器方向内积衡量,称为“对齐得分”)呈现双峰分布。通过对齐训练强制编码器与解码器内积恒为1,去除训练中的退化源,且不引入任何超参数。在多个模型、词典大小和稀疏度下,对齐训练在SAEBench基准上均实现帕累托改进。除改善死特征、稳定性和重建外,该方法可无缝集成至Top/BatchTop-K架构与p-Annealing等机械可解释性技术中。整体上,该方法在不增加计算复杂度或成本的前提下,显著提升SAE的特征质量与稳定性。
原文摘要 · Abstract (English)
Sparse autoencoders (SAEs) are one of the main methods to interpret the inner workings of deep neural networks (DNNs), decomposing activations into higher-dimensional features. However, they exhibit critical shortcomings where a large fraction of features are never activated and are unstable. Despite variants of SAEs that attempt to mitigate these issues, they require additional data, resampling, or training. We propose the \textbf{aligned training}, a parameter-free reparameterization of SAEs that simultaneously improves reconstruction quality, eliminates dead features, and significantly enhances stability across training seeds. Our approach is motivated by an overlooked observation that SAE feature quality, measured by the inner product between encoder and decoder directions (which we call the \textbf{alignment score}), follows a bimodal distribution across all modern architectures. The proposed aligned training enforces a geometric constraint between the encoder and decoder such that their inner product equals one for every feature, which removes a source of degeneracy in the SAE training without adding any hyperparameters. Across multiple models, dictionary sizes, and sparsity levels, the aligned training shows Pareto improvements on the SAEBench benchmarks. Beyond improving dead features, stability and reconstruction, our method readily integrates with techniques in mechanical interpretability such as Top/BatchTop-K architectures and p-Annealing. Overall, the aligned training substantially improves feature quality and stability of SAE without computational complexity or cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。