Scaling Up Test-time Compute With Latent Reasoning: A Recurrent Depth Approach | Awesome LLM Papers Contribute to Awesome LLM Papers

Scaling Up Test-time Compute With Latent Reasoning: A Recurrent Depth Approach

Jonas Geiping, Sean McLeish, Neel Jain, John Kirchenbauer, Siddharth Singh, Brian R. Bartoldson, Bhavya Kailkhura, Abhinav Bhatele, Tom Goldstein . No Venue 2025

[Paper] [Paper]   Search on Google Scholar   Search on Semantic Scholar
Model Architecture Prompting Training Techniques

We study a novel language model architecture that is capable of scaling test-time computation by implicitly reasoning in latent space. Our model works by iterating a recurrent block, thereby unrolling to arbitrary depth at test-time. This stands in contrast to mainstream reasoning models that scale up compute by producing more tokens. Unlike approaches based on chain-of-thought, our approach does not require any specialized training data, can work with small context windows, and can capture types of reasoning that are not easily represented in words. We scale a proof-of-concept model to 3.5 billion parameters and 800 billion tokens. We show that the resulting model can improve its performance on reasoning benchmarks, sometimes dramatically, up to a computation load equivalent to 50 billion parameters.

https://huggingface.co/discussions/paper/67a97e29495b23306cd5eae5

Similar Work