Next Thoughts Are Distributions: Generative Autoregressive Reasoning in the Latent Space
Not on the map. Why?
Abstract (arXiv)
Reasoning problems often admit multiple valid ways to proceed. Continuous reasoning promises to move computation beyond language tokens into a more compact latent space, but representing several plausible ways to think next remains difficult. We introduce Autoregressive Thought Flow (ATF), which models the next continuous thought as a multimodal distribution. A causal autoregressive model performs the reasoning computation, while a lightweight diffusion head generates a plausible next thought from the resulting condition. The sampled thought is fed back into the model, allowing continuous reasoning to unfold for a variable number of steps while preserving the pretrained backbone. Across mathematical reasoning tasks, ATF improves accuracy with compact latent traces and benefits from reinforcement learning and additional test-time thinking. Multi-sample evaluation shows broader solution coverage, indicating that its multimodal predictions capture useful diversity among reasoning paths. Our results suggest that continuous reasoning is more effective when multiple possible next thoughts remain available rather than being collapsed into a single prediction.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 36 of its 41 references that are in the map. First posted on arXiv 27 Sep 2026.
Builds on
36 of 41 references.
References · 36
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 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
- 2025Qwen3 Technical Reportcited by 8,820
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 797
- 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 661
24 more
- 2024Autoregressive Image Generation without Vector Quantizationcited by 651
- 2024Transfusion: Predict the Next Token and Diffuse Images with One Multi-Modal Modelcited by 483
- 2025CODI: Compressing Chain-of-Thought into Continuous Space via Self-Distillationcited by 225
- 2024Compressed Chain of Thought: Efficient Reasoning Through Dense Representationscited by 201
- 2023Implicit Chain of Thought Reasoning via Knowledge Distillationcited by 176
- 2025AIMO-2 Winning Solution: Building State-of-the-Art Mathematical Reasoning Models with OpenMathReasoning datasetcited by 174
- 2026Beyond Benchmarks: MathArena as an Evaluation Platform for Mathematics with LLMscited by 137
- 2025Soft Thinking: Unlocking the Reasoning Potential of LLMs in Continuous Concept Spacecited by 117
- 2023SONAR: Sentence-Level Multimodal and Language-Agnostic Representationscited by 108
- 2025Token Assorted: Mixing Latent and Text Tokens for Improved Language Model Reasoningcited by 91
- 2024Large Concept Models: Language Modeling in a Sentence Representation Spacecited by 89
- 2025Think Silently, Think Fast: Dynamic Latent Compression of LLM Reasoning Chainscited by 85
- 2025NextStep-1: Toward Autoregressive Image Generation with Continuous Tokens at Scalecited by 57
- 2024Multimodal Latent Language Modeling with Next-Token Diffusioncited by 48
- 2025LLMs are Single-threaded Reasoners: Demystifying the Working Mechanism of Soft Thinkingcited by 38
- 2024Causal Diffusion Transformers for Generative Modelingcited by 31
- 2025LaDiR: Latent Diffusion Enhances LLMs for Text Reasoningcited by 29
- 2025Latent Thought Models with Variational Bayes Inference-Time Computationcited by 25
- 2025Mull-Tokens: Modality-Agnostic Latent Thinkingcited by 19
- 2025Parallel Test-Time Scaling for Latent Reasoning Modelscited by 14
- 2025Continuous Autoregressive Language Modelscited by 10
- 2025Deep Thinking by Markov Chain of Continuous Thoughtscited by 9
- 2026LEPO: Latent Reasoning Policy Optimization for Large Language Modelscited by 4
- 2026Latent Thought Flow: Efficient Latent Reasoning in Large Language Modelscited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Reasoning with Autoregressive-Diffusion Collaborative Thoughts
- 2026Latent Reasoning with Normalizing Flows
- 2026Rationale Distribution Distillation via Flow Matching for Distributional Continuous Latent Reasoning
- 2026Emergent Search and Backtracking in Latent Reasoning Models
- 2026Reasoning with Latent Tokens in Diffusion Language Models
- 2025Learning When to Think: Shaping Adaptive Reasoning in R1-Style Models via Multi-Stage RL
- 2025Reinforcing the Diffusion Chain of Lateral Thought with Diffusion Language Models
- 2025Diffuse Thinking: Exploring Diffusion Language Models as Efficient Thought Proposers for Reasoning