C^2R:跨样本一致性正则化缓解稀疏自编码器的特征分裂与吸收

C$^{2}$R: Cross-sample Consistency Regularization Mitigates Feature Splitting and Absorption in Sparse Autoencoders

精选理由

这篇论文发现了SAE解释模型时的两个大坑——特征分裂和吸收,然后用C^2R正则化给治好了,代码开源,直接上手试试。

AI 摘要

稀疏自编码器(SAE)常用于解释大语言模型,但扩展到大规模字典时会出现特征分裂(将连贯概念拆分为非原子潜在变量)和特征吸收(在通用特征中创建任意例外)问题。这些问题源于不同样本间的潜在变量分配不一致。C^2R方法通过惩罚批量中方向相似潜在变量的共激活,显式鼓励每个语义特征由统一潜在变量表示。评估显示C^2R有效缓解分裂和吸收,同时保持重建保真度,代码已在GitHub开源。

AI 翻译 · 中文

稀疏自编码器(SAE)常用于解释大语言模型,但扩展到大规模字典时会出现特征分裂(将连贯概念拆分为非原子潜在变量)和特征吸收(在通用特征中创建任意例外)问题。这些问题源于不同样本间的潜在变量分配不一致。C^2R方法通过惩罚批量中方向相似潜在变量的共激活,显式鼓励每个语义特征由统一潜在变量表示。评估显示C^2R有效缓解分裂和吸收,同时保持重建保真度,代码已在GitHub开源。

arXiv cs.AISparse Autoencoders (SAEs) are widely used to interpret large language models by decomposing activations into sparse, human-understandable features, but scaling to large dictionaries exposes fundamental challenges. Syste