arXiv:2506.09967cs.CLcs.AI2025-06被引 3

用稀疏自编码器低成本激活大模型的推理能力

Resa: Transparent Reasoning Models via SAEs

  • 通过训练稀疏自编码器提取源模型推理特征,引导目标模型学习
  • 推理性能保留97%以上,训练成本降2000倍、时间缩短450倍
  • 能力可迁移复用,无需重训即可在不同模型上提升表现

如何以低成本激发语言模型的强推理能力?我们提出Resa,一个15亿参数的推理模型家族,采用新型高效的稀疏自编码器微调(SAE-Tuning)方法。该方法首先训练稀疏自编码器(SAE)从源模型中捕获推理能力,再利用训练好的SAE指导标准监督微调过程,仅使用验证过的问答数据,无需任何推理轨迹。值得注意的是,当应用于某些基础模型并在后续强化学习后训练时,SAE-Tuning在保持超过97%强化学习模型推理性能的同时,将训练成本降低超过2000倍至约1美元,训练时间缩短超过450倍至约20分钟。此外,应用于轻度强化学习训练的模型(如在2块GPU上运行1小时),可实现AIME24上43.33% Pass@1和AMC23上90% Pass@1的推理性能,额外成本仅约1美元。令人惊讶的是,通过SAE提取的推理能力具有潜在的通用性和模块性:从某一数据集提取的能力仍能提升更大重叠语料库上的表现;从Qwen或Qwen-Math提取的能力可在测试时无缝附加到R1-Distill模型上,无需重训即获得相当提升。大量消融实验验证了这些发现,所有代码与资源均已开源。

原文摘要 · Abstract (English)

How cost-effectively can we elicit strong reasoning in language models by leveraging their underlying representations? We answer this question with Resa, a family of 1.5B reasoning models trained via a novel and efficient sparse autoencoder tuning (SAE-Tuning) procedure. This method first trains an SAE to capture reasoning abilities from a source model, and then uses the trained SAE to guide a standard supervised fine-tuning process to elicit such abilities in a target model, all using verified question-answer data without any reasoning traces. Notably, when applied to certain base models before further RL post-training, SAE-Tuning retains >97% of its RL-trained counterpart's reasoning performance while reducing training costs by >2000x to roughly \$1 and training time by >450x to around 20 minutes. Furthermore, when applied to lightly RL-trained models (e.g., within 1 hour on 2 GPUs), it enables reasoning performance such as 43.33% Pass@1 on AIME24 and 90% Pass@1 on AMC23 for only around \$1 additional cost. Surprisingly, the reasoning abilities extracted via SAEs are potentially both generalizable and modular. Generality means abilities extracted from one dataset still elevate performance on a larger and overlapping corpus. Modularity means abilities extracted from Qwen or Qwen-Math can be attached to the R1-Distill model at test time, without any retraining, and yield comparable gains. Extensive ablations validate these findings and all artifacts are fully open-sourced.

推理增强稀疏编码低成本微调模型可迁移

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