arXiv:2412.04139cs.AI2024-12ICLR被引 14

让大模型每个神经元只学一个概念,提升可解释性且不丢性能。

Monet: Mixture of Monosemantic Experts for Transformers

  • 将稀疏字典学习嵌入专家混合架构,实现端到端训练。
  • 每层支持262,144个专家,参数仅随专家数平方根增长。
  • 可无损调整领域、语言和毒性,适合需要可控行为的场景。

理解大语言模型(LLM)内部计算过程对对其对齐人类价值观、防止生成有害内容至关重要。然而,多义性问题——即单个神经元响应多个无关概念——阻碍了机制可解释性。尽管稀疏自编码器(SAEs)尝试通过稀疏字典学习解耦特征,但依赖事后重建损失导致模型性能下降。为此,我们提出面向Transformer的单义专家混合架构(Monet),将稀疏字典学习直接融入端到端专家混合预训练中。新颖的专家分解方法使每层专家数量达262,144,总参数量仅随专家数平方根增长。分析表明专家间知识互斥,且个体专家封装了参数化知识。此外,Monet可在不损害通用性能的前提下,实现对领域、语言及毒性的知识操控。本研究展示了提升专家数量以增强机制可解释性的潜力,并可直接重构内部知识,从根本上调整模型行为。源代码与预训练检查点见https://github.com/dmis-lab/Monet。

原文摘要 · Abstract (English)

Understanding the internal computations of large language models (LLMs) is crucial for aligning them with human values and preventing undesirable behaviors like toxic content generation. However, mechanistic interpretability is hindered by polysemanticity -- where individual neurons respond to multiple, unrelated concepts. While Sparse Autoencoders (SAEs) have attempted to disentangle these features through sparse dictionary learning, they have compromised LLM performance due to reliance on post-hoc reconstruction loss. To address this issue, we introduce Mixture of Monosemantic Experts for Transformers (Monet) architecture, which incorporates sparse dictionary learning directly into end-to-end Mixture-of-Experts pretraining. Our novel expert decomposition method enables scaling the expert count to 262,144 per layer while total parameters scale proportionally to the square root of the number of experts. Our analyses demonstrate mutual exclusivity of knowledge across experts and showcase the parametric knowledge encapsulated within individual experts. Moreover, Monet allows knowledge manipulation over domains, languages, and toxicity mitigation without degrading general performance. Our pursuit of transparent LLMs highlights the potential of scaling expert counts to enhance mechanistic interpretability and directly resect the internal knowledge to fundamentally adjust model behavior. The source code and pretrained checkpoints are available at https://github.com/dmis-lab/Monet.

可解释性专家混合大模型知识操控

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