arXiv:2606.01080cs.LGcs.AI2026-06

用轻量方法让模型自动生成推理答案并优化自身,提升解题能力。

ThinkSwitch: Context Distillation with LoRA and Weight Interpolation for Specific-Purpose Reasoning Tasks

论文配图:ThinkSwitch: Context Distillation with LoRA and Weight Interpolation for Specific-Purpose Reasoning Tasks
图 1 · 摘自论文原文
  • 通过自生成答案进行知识蒸馏,用QLoRA更新指令模型。
  • 在AIME和PubMedQA上分别提升10分和7分,效果显著。
  • 适合想低成本提升模型推理能力的研究者和开发者。

大型语言模型常通过推理过程来提升复杂任务表现,但会增加延迟、消耗更多令牌且部署复杂。本文提出ThinkSwitch,一种低计算开销的协同训练方法,用于同时优化指令模型与思考模型。基于兼容的Qwen3-4B指令与思考模型,每轮迭代由思考模型生成答案,移除推理过程,将仅含答案的数据对用QLoRA蒸馏至指令模型,并通过球面权重插值重建思考模型。全程仅需任务提示,标签由模型自动生成。在30题AIME 2026测试中,指令模型得分从10/30升至20/30,思考模型从14/30升至22/30;在30题PubMedQA子集上,指令模型从13/30升至18/30,思考模型从18/30升至25/30。实验使用每个领域15个训练提示,单张RTX 3070云卡花费2.86美元。结果虽为小规模,但表明针对性蒸馏可将显式推理的部分优势融入模型权重,同时保留独立思考模式。

原文摘要 · Abstract (English)

Large language models often improve on difficult tasks by spending inference-time compute on a reasoning trace before producing the final answer. That extra computation can be useful, but it also raises latency, token cost, and deployment complexity. We introduce \textbf{ThinkSwitch}, a low-compute procedure for co-training paired instruct and thinking checkpoints. Starting from compatible Qwen3-4B instruct and thinking models, each iteration asks the thinking checkpoint to generate answers, removes the reasoning trace, distills the answer-only pairs into the instruct checkpoint with QLoRA, and reconstructs a thinking checkpoint with spherical weight interpolation. The only human-supplied inputs are task prompts; the labels are generated by the model itself. On a 30-question AIME 2026 evaluation, ThinkSwitch improves the instruct checkpoint from 10/30 to 20/30 and the thinking checkpoint from 14/30 to 22/30. On a 30-question PubMedQA subset, it improves the instruct checkpoint from 13/30 to 18/30 and the thinking checkpoint from 18/30 to 25/30. The complete experiment uses 15 training prompts per domain and costs \$2.86 on a single cloud RTX 3070. The results are small-scale, but they indicate that targeted distillation loops can move part of the benefit of explicit reasoning into weights while preserving a separate thinking mode.

模型优化知识蒸馏推理增强QLoRA

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