arXiv:2509.20784cs.CLcs.AI2025-09

提出原子理论,揭示大模型内部表征的最小单位及其评估标准。

Towards Atoms of Large Language Models

  • 用非欧几何度量AIP定义原子,通过保真度与稳定性双指标评估
  • 发现神经元和特征均不理想:神经元保真但不稳定,特征稳定却失真
  • 在多个大模型中识别出高保真(99.9%)高稳定(99.8%)的理想原子

大语言模型的基础表征单元(FRUs)尚未明确,限制了对模型机制的理解。本文提出原子理论,系统定义、评估并识别此类基本单元——原子。基于原子内积(AIP)这一非欧几何度量,我们正式定义原子,并提出理想原子的两个关键标准:保真度(R²)与稳定性(q*)。证明原子在阈值激活稀疏自编码器(TSAE)下可识别。实证发现大模型普遍存在表征偏移,而AIP能有效纠正该偏移以捕捉底层几何结构。研究显示,广泛使用的神经元和特征均不符合理想原子:神经元保真度高(R²=1)但稳定性差(q*=0.5%),特征更稳定(q*=68.2%)但保真度低(R²=48.8%)。通过大规模实验,我们发现只有当TSAE容量匹配数据规模时,才能可靠识别原子。据此,我们在Gemma2-2B、Gemma2-9B和Llama3.1-8B各层中识别出满足理想原子标准的表征单元,保真度达99.9%,稳定性达99.8%。进一步分析表明,这些原子符合理论预期且具有显著更高的单义性。整体上,本文提出并验证了原子理论作为理解大模型内部表示的基础。

原文摘要 · Abstract (English)

The fundamental representational units (FRUs) of large language models (LLMs) remain undefined, limiting further understanding of their underlying mechanisms. In this paper, we introduce Atom Theory to systematically define, evaluate, and identify such FRUs, which we term atoms. Building on the atomic inner product (AIP), a non-Euclidean metric that captures the underlying geometry of LLM representations, we formally define atoms and propose two key criteria for ideal atoms: faithfulness ($R^2$) and stability ($q^*$). We further prove that atoms are identifiable under threshold-activated sparse autoencoders (TSAEs). Empirically, we uncover a pervasive representation shift in LLMs and demonstrate that the AIP corrects this shift to capture the underlying representational geometry. We find that two widely used units, neurons and features, fail to qualify as ideal atoms: neurons are faithful ($R^2=1$) but unstable ($q^*=0.5\%$), while features are more stable ($q^*=68.2\%$) but unfaithful ($R^2=48.8\%$). To find atoms of LLMs, leveraging atom identifiability under TSAEs, we show via large-scale experiments that reliable atom identification occurs only when the TSAE capacity matches the data scale. Guided by this insight, we identify FRUs with near-perfect faithfulness ($R^2=99.9\%$) and stability ($q^*=99.8\%$) across layers of Gemma2-2B, Gemma2-9B, and Llama3.1-8B, satisfying the criteria of ideal atoms statistically. Further analysis confirms that these atoms align with theoretical expectations and exhibit substantially higher monosemanticity. Overall, we propose and validate Atom Theory as a foundation for understanding the internal representations of LLMs. Code available at https://github.com/ChenhuiHu/towards_atoms.

大模型表征原子理论可解释性稀疏编码

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。