用稀疏自编码器生成可解释的文本嵌入,低成本发现数据中的语义差异和隐藏关联。
Interpretable Embeddings with Sparse Autoencoders: A Data Analysis Toolkit
- 通过稀疏自编码器构建映射到可解释概念的嵌入表示。
- 在2-8倍更低成本下,比大模型更可靠地识别数据偏差与差异。
- 支持按兴趣维度筛选概念,适合研究模型行为演化和触发词
分析大规模文本语料是机器学习的核心挑战,对识别模型不良行为或训练数据偏见至关重要。现有方法多依赖昂贵的大模型标注(如对比数据集差异)或密集嵌入模型(如聚类),难以控制关注属性。本文提出使用稀疏自编码器(SAEs)构建SAE嵌入:其维度对应可解释概念。在四项数据分析任务中,SAE嵌入比大模型更经济可靠,比密集嵌入更具可控性。利用SAE庞大的假设空间,可揭示(1)数据集间的语义差异,(2)文档中意外的概念关联。例如,比较模型响应发现,Grok-4澄清模糊问题的频率高于其他九个前沿模型。相比大模型,SAE嵌入以2-8倍更低成本揭示更大差异,并更可靠识别偏见。此外,SAE嵌入具有可控性:通过过滤概念,可(3)沿兴趣轴聚类文档,(4)在基于属性的检索上优于密集嵌入。通过两个案例研究:分析OpenAI模型随时间的行为变化,以及发现Tulu-3(Lambert et al., 2024)从训练数据中学到的‘触发’短语。结果表明SAE是未结构化数据分析的通用工具,凸显通过数据解读模型的重要意义。
原文摘要 · Abstract (English)
Analyzing large-scale text corpora is a core challenge in machine learning, crucial for tasks like identifying undesirable model behaviors or biases in training data. Current methods often rely on costly LLM-based techniques (e.g. annotating dataset differences) or dense embedding models (e.g. for clustering), which lack control over the properties of interest. We propose using sparse autoencoders (SAEs) to create SAE embeddings: representations whose dimensions map to interpretable concepts. Through four data analysis tasks, we show that SAE embeddings are more cost-effective and reliable than LLMs and more controllable than dense embeddings. Using the large hypothesis space of SAEs, we can uncover insights such as (1) semantic differences between datasets and (2) unexpected concept correlations in documents. For instance, by comparing model responses, we find that Grok-4 clarifies ambiguities more often than nine other frontier models. Relative to LLMs, SAE embeddings uncover bigger differences at 2-8x lower cost and identify biases more reliably. Additionally, SAE embeddings are controllable: by filtering concepts, we can (3) cluster documents along axes of interest and (4) outperform dense embeddings on property-based retrieval. Using SAE embeddings, we study model behavior with two case studies: investigating how OpenAI model behavior has changed over time and finding "trigger" phrases learned by Tulu-3 (Lambert et al., 2024) from its training data. These results position SAEs as a versatile tool for unstructured data analysis and highlight the neglected importance of interpreting models through their data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。