Efficient Post-Training Refinement of Latent Reasoning in Large Language Models
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Reasoning is a key component of language understanding in Large Language Models. While Chain-of-Thought prompting enhances performance via explicit intermediate steps, it suffers from sufficient token overhead and a fixed reasoning trajectory, preventing step-wise refinement. Recent advances in latent reasoning address these limitations by refining internal reasoning processes directly in the model's latent space, without producing explicit outputs. However, a key challenge remains: how to effectively update reasoning embeddings during post-training to guide the model toward more accurate solutions. To overcome this challenge, we propose a lightweight post-training framework that refines latent reasoning trajectories using two novel strategies: 1) Contrastive reasoning feedback, which compares reasoning embeddings against strong and weak baselines to infer effective update directions via embedding enhancement; 2) Residual embedding refinement, which stabilizes updates by progressively integrating current and historical gradients, enabling fast yet controlled convergence. Extensive experiments and case studies are conducted on five reasoning benchmarks to demonstrate the effectiveness of the proposed framework. Notably, a 5% accuracy gain on MathQA without additional training.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 27 of its 65 references that are in the map. First posted on arXiv 10 Jun 2025.
Builds on
27 of 65 references.
Closest ancestors · 3
Same direction, newest first
- 2025Reasoning to Learn from Latent Thoughts24 Mar
- 2025Reasoning with Latent Thoughts: On the Power of Looped Transformers24 Feb
- 2024Language Models are Hidden Reasoners: Unlocking Latent Reasoning Capabilities via Self-Rewarding6 Nov
Other references · 24
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
- 2022Emergent Abilities of Large Language Modelscited by 2,483
- 2022Least-to-Most Prompting Enables Complex Reasoning in Large Language Modelscited by 1,397
- 2022Program of Thoughts Prompting: Disentangling Computation from Reasoning for Numerical Reasoning Taskscited by 1,057
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
- 2022Automatic Chain of Thought Prompting in Large Language Modelscited by 684
- 2019MathQA: Towards Interpretable Math Word Problem Solving with Operation-Based Formalismscited by 653
- 2017Program Induction by Rationale Generation : Learning to Solve and Explain Algebraic Word Problemscited by 642
- 2022Complexity-Based Prompting for Multi-Step Reasoningcited by 387
- 2022Selection-Inference: Exploiting Large Language Models for Interpretable Logical Reasoningcited by 331
- 2025MixLLM: Dynamic Routing in Mixed Large Language Modelscited by 56
12 more
- 2025Think Before Recommend: Unleashing the Latent Reasoning Power for Sequential Recommendationcited by 50
- 2024LLM-Enhanced User-Item Interactions: Leveraging Edge Information for Optimized Recommendationscited by 24
- 2024Knockoff-Guided Feature Selection via A Single Pre-trained Reinforced Agentcited by 14
- 2025A Survey on Data-Centric AI: Tabular Learning from Reinforcement Learning and Generative AI Perspectivecited by 14
- 2025Sculpting Features from Noise: Reward-Guided Hierarchical Diffusion for Task-Optimal Feature Transformationcited by 13
- 2024Topology-aware Reinforcement Feature Space Reconstruction for Graph Datacited by 11
- 2025Unsupervised Feature Transformation via In-context Generation, Generator-critic LLM Agents, and Duet-play Teamingcited by 11
- 2025Agentic Feature Augmentation: Unifying Selection and Generation with Teaming, Planning, and Memoriescited by 9
- 2025Bridging the Domain Gap in Equation Distillation with Reinforcement Feedbackcited by 4
- 2025Brownian Bridge Augmented Surrogate Simulation and Injection Planning for Geological CO2 Storagecited by 4
- 2025Towards Data-Centric AI: A Comprehensive Survey of Traditional, Reinforcement, and Generative Approaches for Tabular Data Transformationcited by 2
- 2024Building a Chinese Medical Dialogue System: Integrating Large-scale Corpora and Novel Modelscited by 1
Built on by
Citing papers per year: 2025: 3; 2026: 4.
7 papers.
- 2025SwiReasoning: Switch-Thinking in Latent and Explicit for Pareto-Superior Reasoning LLMscited by 20
- 2025Teaching Time Series to See and Speak: Forecasting with Aligned Visual and Textual Perspectivescited by 7
- 2025StorageXTuner: An LLM Agent-Driven Automatic Tuning Framework for Heterogeneous Storage Systemscited by 1
- 2026Why Limit the Residual Stream to Layers and Not Tokens? Persistent Memory for Continuous Latent Reasoningcited by 1
- 2026Multi-Agent Procedural Graph Extraction with Structural and Logical Refinementcited by 0
- 2026Invariant Reasoning Directions in Latent Trajectories of Language Modelscited by 0
- 2026Right Direction, Wrong Step: Geometric Analysis of Finite-Step Failure in Looped Transformerscited by 0
- 2026Right Direction, Wrong Step: Geometric Analysis of Finite-Step Failure in Looped Transformers15 Sep 2026
- 2026Invariant Reasoning Directions in Latent Trajectories of Language Models28 Jun 2026
- 2026Why Limit the Residual Stream to Layers and Not Tokens? Persistent Memory for Continuous Latent Reasoning5 Jun 2026
- 2026Multi-Agent Procedural Graph Extraction with Structural and Logical Refinement27 Jan 2026
- 2025StorageXTuner: An LLM Agent-Driven Automatic Tuning Framework for Heterogeneous Storage Systems28 Oct 2025
- 2025SwiReasoning: Switch-Thinking in Latent and Explicit for Pareto-Superior Reasoning LLMs6 Oct 2025
- 2025Teaching Time Series to See and Speak: Forecasting with Aligned Visual and Textual Perspectives30 Jun 2025
Similar, not cited
Close in wording, no citation either way
- 2025A Survey on Latent Reasoning
- 2026Thinking in Latents: Adaptive Anchor Refinement for Implicit Reasoning in LLMs
- 2026SPOT: Span-level Pause-of-Thought for Efficient and Interpretable Latent Reasoning in Large Language Models
- 2026SeLaR: Selective Latent Reasoning in Large Language Models
- 2025Latent Reasoning via Sentence Embedding Prediction
- 2026Latent Thought Flow: Efficient Latent Reasoning in Large Language Models
- 2026Latent Recurrent Thoughts: Recurrent Refinement of Proposed Latents for Reasoning with Frozen LLMs
- 2025Seek in the Dark: Reasoning via Test-Time Instance-Level Policy Gradient in Latent Space