Are Latent Reasoning Models Easily Interpretable?
Organizations: Department of Computer Science, University of Maryland, College Park, MD, USA
Abstract
Latent reasoning models (LRMs) have attracted significant research interest due to their low inference cost (relative to explicit reasoning models) and theoretical ability to explore multiple reasoning paths in parallel. However, these benefits come at the cost of reduced interpretability: LRMs are difficult to monitor because they do not reason in natural language. This paper presents an investigation into LRM interpretability by examining two state-of-the-art LRMs. First, we find that latent reasoning tokens are often unnecessary for LRMs' predictions; on logical reasoning datasets, LRMs can almost always produce the same final answers without using latent reasoning at all. This underutilization of reasoning tokens may partially explain why LRMs do not consistently outperform explicit reasoning methods and raises doubts about the stated role of these tokens in prior work. Second, we demonstrate that when latent reasoning tokens are necessary for performance, we can decode gold reasoning traces up to 65-93% of the time for correctly predicted instances. This suggests LRMs often implement the expected solution rather than an uninterpretable reasoning process. Finally, we present a method to decode a verified natural language reasoning trace from latent tokens without knowing a gold reasoning trace a priori, demonstrating that it is possible to find a verified trace for a majority of correct predictions but only a minority of incorrect predictions. Our findings highlight that current LRMs largely encode interpretable processes, and interpretability itself can be a signal of prediction correctness.
Figures & tables
| GSM8k-Aug | PrOntoQA | ProsQA | |||||
| Method | Base Model | Acc. (%) | # Tok. | Acc. (%) | # Tok. | Acc. (%) | # Tok. |
| No-CoT | GPT-2 Small | 16.8 (16.5 † ) | 3.2 | 87.9 (93.8 † ) | 3.0 | 76.0 (76.7 † ) | 9.5 |
| CoT | GPT-2 Small | 41.6 (42.9 † ) | 31.0 | 99.3 (98.8 † ) | 92.7 | 74.2 (77.5 † ) | 51.6 |
| Coconut | GPT-2 Small | 33.1 (34.1 † ) | 9.2 | 99.0 (99.8 † ) | 9.0 | 98.0 (97.0 † ) | 15.5 |
| CODI | GPT-2 Small | 42.2 ∗ (43.7 ‡ ) | 12.3 | 95.1 | 12.0 | 81.6 | 18.2 |
| No-CoT | Llama-3.2-1B | 30.1 (30.9 ‡ ) | 4.2 | 99.8 | 3.0 | 87.8 | 8.6 |
| LRM | Base Model | Reasoning Mode | GSM8k-Aug | PrOntoQA | ProsQA |
| Coconut | GPT-2 Small | No-CoT | 0.3 | 0.0 | 0.0 |
| CoT | 19.2 | 0.0 | -6.0 | ||
| Llama-3.2-1B-Instruct | No-CoT | -0.9 | 0.0 | 0.0 | |
| CoT | 29.3 | -0.3 | -1.4 | ||
| CODI | GPT-2 Small | No-CoT | -5.9 | -0.3 | -0.2 |
| CoT | 3.5 | -0.4 | -7.4 |
Appendix figures & tables26 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Training | Validation | Original Test | Valid Gold Reasoning Trace Set | VP-Friendly Gold Reasoning Trace Set |
| GSM8k-Aug | 385620 | 500 | 1319 | 1194 | 460 |
| PrOntoQA | 9000 | 200 | 800 | - | - |
| ProsQA | 17886 | 300 | 500 | - | - |
| GPT-2 Small | Llama-3.2-1B-Instruct | |||||
| Hyperparameter | GSM8k-Aug | ProsQA | PrOntoQA | GSM8k-Aug | ProsQA | PrOntoQA |
| Latent Tokens Per Stage | 2 | 1 | 1 | 1 | 1 | 1 |
| Stage 0 Epochs | 3 | 5 | 5 | 3 | 3 | 3 |
| Epochs Per Stage | 3 | 5 | 5 | 1 | 1 | 2 |
| Max Latent Stage | 3 | 6 | 6 | 6 | 6 | 6 |
| Total Epochs | 50 | 50 | 50 | 10 | 10 | 20 |
| GPT-2 Small | Llama-3.2-1B-Instruct | |||||
| Hyperparameter | GSM8k-Aug | ProsQA | PrOntoQA | GSM8k-Aug | ProsQA | PrOntoQA |
| Latent Loss Weight ( ) | 1 | 1 | 1 | 1 | 1 | 1 |
| CoT Loss Weight ( ) | 1 | 1 | 1 | 1 | 1 | 1 |
| Distillation Loss Weight ( ) | 1 | 1 | 1 | 20 | 20 | 20 |
| Num Latent Tokens | 6 | 6 | 6 | 6 | 6 | 6 |
| Total Epochs | 40 | 40 | 40 | 10 | 10 | 10 |
| GPT-2 Small | Llama-3.2-1B-Instruct | |||||
| Hyperparameter | GSM8k-Aug | ProsQA | PrOntoQA | GSM8k-Aug | ProsQA | PrOntoQA |
| Total Epochs | 25 | 25 | 25 | 10 | 10 | 10 |
| Batch Size | 128 | 128 | 128 | 128 | 128 | 128 |
| Learning Rate | ||||||
| Weight Decay | 0.01 | 0.01 | 0.01 | 0.1 | 0.1 | 0.1 |
| BF16 Precision | ✗ | ✗ | ✗ | ✓ | ✓ | ✓ |
| GPT-2 Small | Llama-3.2-1B-Instruct | |||||
| Hyperparameter | GSM8k-Aug | ProsQA | PrOntoQA | GSM8k-Aug | ProsQA | PrOntoQA |
| Total Epochs | 25 | 25 | 25 | 10 | 10 | 20 |
| Batch Size | 128 | 128 | 128 | 128 | 128 | 128 |
| Learning Rate | ||||||
| Weight Decay | 0.01 | 0.01 | 0.01 | 0.1 | 0.1 | 0.1 |
| BF16 Precision | ✗ | ✗ | ✗ | ✓ | ✓ | ✓ |
| GPT-2 Small | Llama-3.2-1B-Instruct | |||||
| Hyperparameter | GSM8k-Aug | ProsQA | PrOntoQA | GSM8k-Aug | ProsQA | PrOntoQA |
| Latent Loss Weight ( ) | 1 | 1 | 1 | 1 | 1 | 1 |
| CoT Loss Weight ( ) | 1 | 1 | 1 | 1 | 1 | 1 |
| No-CoT Loss Weight ( ) | 1 | 1 | 1 | 1 | 1 | 1 |
| Latent Tokens Per Stage | 2 | 1 | 1 | 1 | 1 | 1 |
| Stage 0 Epochs | 3 | 5 | 5 | 3 | 3 | 3 |
| GPT-2 Small | Llama-3.2-1B-Instruct | |||||
| Hyperparameter | GSM8k-Aug | ProsQA | PrOntoQA | GSM8k-Aug | ProsQA | PrOntoQA |
| Latent Loss Weight ( ) | 1 | 1 | 1 | 1 | 1 | 1 |
| CoT Loss Weight ( ) | 1 | 1 | 1 | 1 | 1 | 1 |
| No-CoT Loss Weight ( ) | 1 | 1 | 1 | 1 | 1 | 1 |
| Distillation Loss Weight ( ) | 1 | 1 | 1 | 20 | 20 | 20 |
| Num Latent Tokens | 6 | 6 | 6 | 6 | 6 | 6 |
| Base Model | Mode | Control Tokens |
| GPT-2 Small | No-CoT | [prompt] + [eocot] answer |
| CoT | [prompt] + [bocot] cot + eocot answer | |
| Latent | [prompt] + [bocot] latent [eocot] answer | |
| Llama-3.2-1B | No-CoT | [prompt] + [eot] + [eocot] answer |
| CoT | [prompt] + [eot] + [bocot] cot + eocot answer | |
| Latent | [prompt] + [eot] + [bocot] latent [eocot] answer |
| LRM | Base Model | Reasoning Mode | GSM8k-Aug | PrOntoQA | ProsQA |
| Coconut | GPT-2 Small | No-CoT | 22.4 | 100.0 | 94.2 |
| CoT | 41.3 | 100.0 | 88.2 | ||
| Coconut | 22.1 | 100.0 | 94.2 | ||
| Llama-3.2-1B-Instruct | No-CoT | 29.2 | 94.9 | 99.8 | |
| CoT | 59.4 | 94.6 | 98.4 | ||
| Coconut | 30.1 | 94.9 | 99.8 |
| First Match (%) | Stable Match (%) | ||||||
| Base Model | Dataset | Explicit | Coconut | CODI | Explicit | Coconut | CODI |
| GPT-2 Small | GSM8k-Aug | 98.1 9.8 | 54.4 36.2 | 44.2 33.6 | 99.3 5.4 | 69.0 32.1 | 53.6 32.5 |
| ProsQA | 92.0 21.7 | 0.0 0.0 | 2.5 12.6 | 97.7 10.9 | 0.0 0.0 | 4.0 16.6 | |
| PrOntoQA | 45.5 32.8 | 0.1 2.9 | 0.3 4.4 | 47.1 33.0 | 0.1 2.9 | 0.5 5.8 | |
| Llama-3.2-1B | GSM8k-Aug | 89.3 19.1 | 19.5 30.0 | 30.9 33.9 | 90.6 17.4 | 22.5 32.5 | 37.7 35.9 |
| ProsQA | 71.1 22.6 | 0.0 0.7 | 0.1 2.2 | 74.7 22.5 | 0.0 0.7 | 0.7 7.4 | |
| Method | Base Model | GSM8k-Aug | PrOntoQA | ProsQA |
| Acc. (%) | Acc. (%) | Acc. (%) | ||
| Coconut | GPT-2 Small | |||
| CODI | GPT-2 Small | |||
| Coconut | Llama-3.2-1B | |||
| CODI | Llama-3.2-1B |
| Method | Base Model | Incorrect Instances | Correct Answer in Top-10 | Percent |
| Coconut | GPT-2 Small | 793 | 369 | 46.5% |
| CODI | GPT-2 Small | 675 | 337 | 49.9% |
| Coconut | Llama-3.2-1B-Instruct | 759 | 425 | 56.0% |
| CODI | Llama-3.2-1B-Instruct | 511 | 275 | 53.8% |
Implicit Reasoning in Language Models
Implicit reasoning in language models studies how models solve problems through internal computation rather than, or alongside, written chains of thought.
See all 95 papers