让专家模型天生可解释,比传统方法更直观。
Mixture of Experts Made Intrinsically Interpretable
- 将专家网络重写为稀疏大模型,天然支持可解释性
- 在棋类和语言任务上性能媲美GPT-2,困惑度更低
- 适合需要透明决策过程的AI应用开发者
大型语言模型中的神经元常表现出多义性,同时编码多个无关概念,导致可解释性差。本文提出MoE-X,一种设计为内在可解释的混合专家(MoE)语言模型。受启发于宽网络与稀疏激活更易捕捉可解释特征的现象,但直接训练大规模稀疏网络成本过高。MoE架构通过仅激活部分专家实现高效扩展,自然契合可解释目标。在MoE-X中,我们将MoE层重写为等效的稀疏大前馈网络,实现隐藏层规模高效扩展并保持稀疏性。进一步,我们强制每个专家内部激活稀疏,并重构路由机制以优先选择激活最稀疏的专家,确保仅有最显著特征被传递处理。在国际象棋和自然语言任务上的评估表明,其性能接近密集模型,且可解释性优于基于稀疏自编码器(SAE)的方法,困惑度低于GPT-2。
原文摘要 · Abstract (English)
Neurons in large language models often exhibit \emph{polysemanticity}, simultaneously encoding multiple unrelated concepts and obscuring interpretability. Instead of relying on post-hoc methods, we present \textbf{MoE-X}, a Mixture-of-Experts (MoE) language model designed to be \emph{intrinsically} interpretable. Our approach is motivated by the observation that, in language models, wider networks with sparse activations are more likely to capture interpretable factors. However, directly training such large sparse networks is computationally prohibitive. MoE architectures offer a scalable alternative by activating only a subset of experts for any given input, inherently aligning with interpretability objectives. In MoE-X, we establish this connection by rewriting the MoE layer as an equivalent sparse, large MLP. This approach enables efficient scaling of the hidden size while maintaining sparsity. To further enhance interpretability, we enforce sparse activation within each expert and redesign the routing mechanism to prioritize experts with the highest activation sparsity. These designs ensure that only the most salient features are routed and processed by the experts. We evaluate MoE-X on chess and natural language tasks, showing that it achieves performance comparable to dense models while significantly improving interpretability. MoE-X achieves a perplexity better than GPT-2, with interpretability surpassing even sparse autoencoder (SAE)-based approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。