Soft Teacher Forcing
On the map: LLM reasoning → Test-time compute and efficient thinking also fits: Prompting and chain-of-thought reasoning
Abstract (arXiv)
Soft tokens have so far been used to replace discrete chain-of-thought in latent reasoning methods. These approaches typically require multi-stage training and, in some cases, auxiliary modules, yet they still underperform supervised fine-tuning (SFT) on <input, chain-of-thought, output> triples. In this paper, we ask whether soft tokens can be used to improve SFT itself, instead of as an avenue for latent reasoning. We introduce Soft Teacher Forcing (Soft-TF), a simple training algorithm that combines teacher-forced embeddings with soft tokens through random convex combinations. The soft tokens are computed in a single parallel forward pass over the ground-truth sequence. Soft-TF adds no parameters, no external model and no curriculum, and leaves inference unchanged. Experiments show that Soft-TF improves over SFT on GSM8K, MATH, MathQA and AQuA, by +2.1 to +4.1 points on GSM8K across seven Gemma, Llama and Mistral models.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 34 of its 41 references that are in the map. First posted on arXiv 12 Oct 2026.
Builds on
34 of 41 references.
Closest ancestors · 1
Same direction, newest first
Other references · 33
- 2017Attention Is All You Needcited by 60,361
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,475
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
- 2023Let's Verify Step by Stepcited by 3,789
- 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
21 more
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
- 2025Gemma 3 Technical Reportcited by 1,824
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,210
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 905
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 797
- 2019MathQA: Towards Interpretable Math Word Problem Solving with Operation-Based Formalismscited by 662
- 2017Program Induction by Rationale Generation : Learning to Solve and Explain Algebraic Word Problemscited by 650
- 2021Domain Generalization with MixStylecited by 566
- 2023MAmmoTH: Building Math Generalist Models through Hybrid Instruction Tuningcited by 487
- 2017Regularizing Neural Networks by Penalizing Confident Output Distributionscited by 464
- 2023Orca: Progressive Learning from Complex Explanation Traces of GPT-4cited by 379
- 2024From Explicit CoT to Implicit CoT: Learning to Internalize CoT Step by Stepcited by 235
- 2025CODI: Compressing Chain-of-Thought into Continuous Space via Self-Distillationcited by 225
- 2025Soft Thinking: Unlocking the Reasoning Potential of LLMs in Continuous Concept Spacecited by 117
- 2025SoftCoT: Soft Chain-of-Thought for Efficient Reasoning with LLMscited by 116
- 2023NEFTune: Noisy Embeddings Improve Instruction Finetuningcited by 100
- 2025SIM-CoT: Supervised Implicit Chain-of-Thoughtcited by 70
- 2019Scheduled Sampling for Transformerscited by 48
- 2025LLMs are Single-threaded Reasoners: Demystifying the Working Mechanism of Soft Thinkingcited by 38
- 2025Soft Tokens, Hard Truthscited by 35
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Improving Latent Reasoning in LLMs via Soft Concept Mixing
- 2026SoFT: Soft Targets for Generalizable LLM Fine-Tuning
- 2026Which Tokens Should SFT Actually Learn? A Token-Trimming Perspective on Mathematical Reasoning
- 2025SoftCoT++: Test-Time Scaling with Soft Chain-of-Thought Reasoning
- 2026From Noise to Diversity: Random Embedding Injection in LLM Reasoning
- 2026SeLaR: Selective Latent Reasoning in Large Language Models
- 2026One Student, Many Teachers: Multi-Task On-Policy Distillation via Soft-Prompt Privileged Context
- 2026RuPLaR : Efficient Latent Compression of LLM Reasoning Chains with Rule-Based Priors From Multi-Step to One-Step