Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approach
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Commonsense Reasoning | WinoGrande | Accuracy59.4 | 1581 | |
| Commonsense Reasoning | HellaSwag | HellaSwag Accuracy65.2 | 897 | |
| Language Modeling | WikiText | PPL41.31 | 740 | |
| Multitask Language Understanding | MMLU | Accuracy31.4 | 568 | |
| Mathematical Reasoning | AIME 2024 | Accuracy50.0016 | 525 | |
| Mathematical Reasoning | SVAMP | Accuracy54.8 | 403 | |
| Language Modeling | Wiki | Perplexity (PPL)12.14 | 298 | |
| Reading Comprehension | BoolQ | Accuracy (BoolQ)69.8 | 258 | |
| Code Generation | HumanEval | Accuracy34.3617 | 224 | |
| Language Modeling | The Pile | Perplexity6.29 | 132 |