arXiv:2602.20904cs.LG2026-02

用可解释的适配器分析推理模型微调前后的差异

Transcoder Adapters for Reasoning-Model Diffing

  • 设计转码适配器,捕捉微调前后MLP计算的变化
  • 适配器恢复了50%-90%的推理性能提升,且能准确预测下一个词
  • 发现仅约8%特征与推理行为相关,其中少数特征决定犹豫词生成

尽管推理模型日益普及,其内部机制在推理训练后的变化仍不明确。本文提出转码适配器,用于学习微调前后MLP计算差异的可解释近似。我们将其应用于Qwen2.5-Math-7B与其推理蒸馏版DeepSeek-R1-Distill-Qwen-7B的对比。适配器忠实反映目标模型的内部计算和下一个词预测。在推理基准测试中,适配器匹配推理模型的响应长度,通常恢复50%-90%的推理微调带来的准确率提升。适配器特征稀疏激活且可解释。分析发现,仅有约8%的特征与推理行为直接相关。深入研究一种行为——犹豫令牌(如“wait”)的生成,通过归因图发现仅约2.4%的适配器特征(共5.6k个)承担两种功能:这些特征对生成犹豫令牌是必要且充分的;移除后响应长度下降,但准确率通常不受影响。结果为理解推理训练提供洞见,也表明转码适配器可能适用于更广泛的微调研究。

原文摘要 · Abstract (English)

While reasoning models are increasingly ubiquitous, the effects of reasoning training on a model's internal mechanisms remain poorly understood. In this work, we introduce transcoder adapters, a technique for learning an interpretable approximation of the difference in MLP computation before and after fine-tuning. We apply transcoder adapters to characterize the differences between Qwen2.5-Math-7B and its reasoning-distilled variant, DeepSeek-R1-Distill-Qwen-7B. Learned adapters are faithful to the target model's internal computation and next-token predictions. When evaluated on reasoning benchmarks, adapters match the reasoning model's response lengths and typically recover 50-90% of the accuracy gains from reasoning fine-tuning. Adapter features are sparsely activating and interpretable. When examining adapter features, we find that only ~8% have activating examples directly related to reasoning behaviors. We deeply study one such behavior -- the production of hesitation tokens (e.g., "wait"). Using attribution graphs, we trace hesitation to only ~2.4% of adapter features (5.6k total) performing one of two functions. These features are necessary and sufficient for producing hesitation tokens; removing them reduces response length, often without affecting accuracy. Overall, our results provide insight into reasoning training and suggest transcoder adapters may be useful for studying fine-tuning more broadly.

模型可解释性推理增强微调分析

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