用可追踪的原型构建模型,让推理过程更透明且高效。
Prototype Language Models

- 用稀疏原型混合生成输出,替代密集网络,提升可解释性。
- 在130M到1.6B参数模型上,性能接近或优于基线,平均差距≤2.5%。
- 支持快速训练数据溯源(快500倍),适合需要可解释性的应用。
理解语言模型输出的来源对审计、修正和分析至关重要,但现代大模型中这一过程成本高、近似且事后进行。标准模型通过密集网络生成文本,使训练数据的影响分散在参数中,缺乏显式可追踪结构。我们提出原型语言模型架构PRISM,每个预测由学习到的原型的稀疏非负混合构成,通过聚类目标将每个原型锚定在训练数据的连贯邻域。在130M至1.6B参数规模、最多500亿个标记的多个模型上,原型模型在下游任务中的平均准确率与匹配的密集基线持平或仅低2.5个百分点。我们发现稀疏原型结构能局域化损失曲面的曲率,使海森矩阵更易处理,从而实现训练数据归属,速度比事后方法快约500倍,内存消耗相当。校准线性原型控制器可提升下游准确率约3个百分点,并可追溯修正来源;针对性抑制原型可移除特定行为,无需微调且生成质量无明显下降。
原文摘要 · Abstract (English)
Knowing which training examples drive outputs is fundamental to auditing, correcting, and understanding language models, yet for modern LLMs this remains expensive, approximate, and largely post-hoc. Standard language models generate tokens through a dense network pathway, causing training data's influence to be distributed across parameters rather than organized along explicit, traceable components. We introduce a prototype language model architecture, Prototypes for Interpretable Sequence Modeling (PRISM), that forms each prediction via a sparse, non-negative mixture of learned prototypes, trained with clustering objectives that anchor each prototype to coherent neighborhoods of training examples. Across architectures from 130M to 1.6B parameters trained on up to 50B tokens, prototype language models either surpass or remain within 2.5 percentage points on average downstream accuracy of matched dense baselines. We show that sparse prototype structure localizes curvature in the loss landscape, yielding a more tractable Hessian and enabling training data attribution that is ~500x faster than post hoc baselines when consuming equivalent memory. Calibrating linear prototype controllers can improve downstream accuracy by roughly 3 points while tracing those corrections back to training neighborhoods, and targeted prototype suppression can remove model behaviors without finetuning or measurable loss in generation quality.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。