ThinkSwitch: Context Distillation with LoRA and Weight Interpolation for Specific-Purpose Reasoning Tasks
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
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 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.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 20 of its 25 references that are in the map. First posted on arXiv 31 May 2026.
Builds on
20 of 25 references.
Closest ancestors · 2
Same direction, newest first
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameters6 Aug
- 2022STaR: Bootstrapping Reasoning With Reasoning28 Mar
Other references · 18
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2025Qwen3 Technical Reportcited by 8,195
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
- 2022Large Language Models are Zero-Shot Reasonerscited by 4,323
- 2023QLoRA: Efficient Finetuning of Quantized LLMscited by 3,355
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
- 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2021A General Language Assistant as a Laboratory for Alignmentcited by 1,023
- 2023Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizescited by 710
6 more
- 2024From Explicit CoT to Implicit CoT: Learning to Internalize CoT Step by Stepcited by 225
- 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 218
- 2023Implicit Chain of Thought Reasoning via Knowledge Distillationcited by 165
- 2025R2E-Gym: Procedural Environments and Hybrid Verifiers for Scaling Open-Weights SWE Agentscited by 140
- 2018Supervising strong learners by amplifying weak expertscited by 139
- 2022Learning by Distilling Contextcited by 128
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Thinking with Nothinking Calibration: A New In-Context Learning Paradigm in Reasoning Large Language Models
- 2025ThinkSwitcher: When to Think Hard, When to Think Fast
- 2025Thought calibration: Efficient and confident test-time scaling
- 2025Thinking Before You Speak: A Proactive Test-time Scaling Approach
- 2026ThinkRetrieve: Retrieval-Augmented Reasoning Traces for Test-Time Scaling
- 2026Batched Contextual Reinforcement: A Task-Scaling Law for Efficient Reasoning
- 2026MI-Distillation: Selecting from Model-Interpolated Instruct-Reasoning Data Spectrum for Chain-of-Thought Distillation
- 2026Reasoning Compression with Mixed-Policy Distillation