Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approach
Geiping, Jonas, McLeish, Sean, Jain, Neel, Kirchenbauer, John, Singh, Siddharth, Bartoldson, Brian R., Kailkhura, Bhavya, Bhatele, Abhinav, Goldstein, Tom
–arXiv.org Artificial Intelligence
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.
arXiv.org Artificial Intelligence
Feb-17-2025
- Country:
- North America > United States (1.00)
- Europe (1.00)
- Genre:
- Research Report (0.50)
- Instructional Material (0.48)
- Industry:
- Education (1.00)
- Energy (0.93)
- Government > Regional Government
- Technology: