Same Beliefs, Different Policies: Learning Reusable Representations for Early Stopping in LLM Reasoning
Organizations: ETH Zürich · MBZUAI · University of Zürich · The University of Melbourne
Abstract
Reasoning LLMs often continue reasoning after reaching a correct answer, wasting computation and sometimes replacing it with an incorrect one. Deciding when to stop is a metareasoning problem: the model must weigh what further reasoning could achieve against its cost, given what it can already answer. Direct utility training learns a stopping representation at a chosen token price; supporting new prices can require additional controller training. We introduce BELT (BELief-based Termination), which separates outcome prediction from stopping control. It learns beliefs about current and future answers from a frozen LLM's hidden states, then fits a linear stopping policy on the frozen features. The prediction targets are independent of token price, so one probe supports different accuracy-cost trade-offs through CPU policy refitting or threshold adjustment. Averaged over five out-of-distribution (OOD) math benchmarks, BELT reduces generated tokens by 41% on Qwen3-1.7B and 51% on OLMo-3-7B-Think without loss of accuracy. The same math-trained BELT transfers to coding and agentic tasks without training on code or agentic data. In all three domains, its accuracy-cost trade-offs are competitive with or better than those of end-to-end controllers. On OLMo3, its probe has 21.2x fewer parameters than one checkpoint of the strongest baseline, OS-Pruner, and needs 55x fewer training GPU-hours than OS-Pruner's five price-specific checkpoints.
Figures & tables
| Method | Probe Params | Acc. tolerance: 0 pp | Acc. tolerance: 1 pp | Acc. tolerance: 2 pp | |||
|---|---|---|---|---|---|---|---|
| ID | OOD | ID | OOD | ID | OOD | ||
| CoDE-Stop ‡ | 0 | ||||||
| DEER ‡ | 0 | ||||||
| TC | 514 | ||||||
| ROM | 5M | ||||||
| OSP | 101M | ||||||
| Method | Probe Params | Saving: 70% | Saving: 60% | Saving: 50% | Saving: 40% | ||||
|---|---|---|---|---|---|---|---|---|---|
| ID | OOD | ID | OOD | ID | OOD | ID | OOD | ||
| Natural †‡ | – | ||||||||
| Fixed cap ‡ | 0 | ||||||||
| CoDE-Stop ‡ | 0 | ||||||||
| DEER ‡ | 0 | ||||||||
| TC | 514 | ||||||||
| Method | Probe Params | Token saving (%) at tolerance | Accuracy (%) at saving | |||||
| 0.00 | 0.01 | 0.02 | 70% | 60% | 50% | 40% | ||
| Natural †‡ | – | – | – | – | ||||
| Fixed cap ‡ | 0 | – | – | – | ||||
| CoDE-Stop ‡ | 0 | |||||||
| DEER ‡ | 0 | |||||||
| TC | 514 | |||||||
| Method | Probe Params | ALFWorld (success) | ScienceWorld (score) | ||||||||
| Saving (%) at tolerance | Success at saving | Saving (%) | Score at saving | ||||||||
| 0 pp | 2 pp | 5 pp | 20% | 40% | at 0 pt | 80% | 70% | 60% | 50% | ||
| Natural | – | – | – | – | – | ||||||
| OSP-1 | 405M | ||||||||||
| BELT-B-T | 19M | ||||||||||
| BELT-H-T | 19M | ||||||||||
| Belief set | (a) Prediction targets | (b) Policy inputs | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| 2 | 5 | 10 | 20 | 50 | 2 | 5 | 10 | 20 | 50 | |
| Now , End | 0.7451 | 0.7307 | 0.7067 | 0.6586 | 0.5820 | 0.7447 | 0.7302 | 0.7061 | 0.6595 | 0.5831 |
| Now , End , Later | 0.7448 | 0.7304 | 0.7065 | 0.6585 | 0.5853 | 0.7449 | 0.7304 | 0.7063 | 0.6592 | 0.5818 |
| Now , End , Same | 0.7459 | 0.7313 | 0.7068 | 0.6648 | 0.5882 | 0.7456 | 0.7311 | 0.7070 | 0.6652 | 0.5838 |
| All four | 0.7454 | 0.7310 | 0.7072 | 0.6660 | 0.5851 | 0.7454 | 0.7310 | 0.7072 | 0.6660 | 0.5851 |
Appendix figures & tables78 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Distribution | Questions | Composition |
|---|---|---|---|
| DAPO | ID | 740 | Held-out DAPO source questions |
| Numina | ID | 473 | Held-out NuminaMath-hard source questions |
| DSR | ID | 481 | Held-out DeepScaleR source questions |
| M500 | OOD | 295 | Retained MATH-500 questions |
| Olymp. | OOD | 366 | English text-only mathematics, single answer |
| GK | OOD | 191 | Gaokao 2023 mathematics, English |
| Backbone | Bank | Dataset | Rows | Unpars. % | Recovered | Still unpars. % | Nat. acc. % |
|---|---|---|---|---|---|---|---|
| Qwen3-1.7B | math_union_fit | ALL | 42,408 | 18.0 | 2,918 / 2,097 | 9.1 | 48.62 53.57 |
| dapo | 18,048 | 10.7 | 1,169 / 797 | 1.8 | 52.00 56.42 | ||
| deepscaler | 12,288 | 11.8 | 837 / 661 | 3.5 | 55.87 61.25 | ||
| numina_hard | 12,072 | 35.0 | 912 / 639 | 25.3 | 36.19 41.48 | ||
| Qwen3-1.7B | math_union_select | ALL | 10,648 | 15.9 | 581 / 418 | 8.7 | 50.93 54.86 |
| dapo | 4,656 | 8.7 | 249 / 158 | 1.3 | 54.10 57.50 |
| Population | Questions | Acc. micro (%) | Acc. macro (%) | Tokens (micro) |
|---|---|---|---|---|
| Qwen3-1.7B ID | 1,694 | 49.52 53.92 | 48.85 53.38 | 6,564.0 6,564.5 |
| Qwen3-1.7B OOD | 1,025 | 65.73 72.73 | 60.58 66.66 | 5,021.8 5,022.1 |
| OLMo-3-7B-Think OOD | 1,025 | 81.83 82.10 | 78.35 78.63 | 6,868.0 6,868.0 |
| Id | Statement | Submitted | Random init | ReProbe init | State (random init) |
|---|---|---|---|---|---|
| H1 | abstract: Qwen3 OOD token reduction "66 %" (BELT-H, 0 pp, macro) | 66.18 | 39.11 | 40.65 | changed |
| H1b | abstract: OLMo3 OOD token reduction "53 %" (BELT-H, 0 pp, macro) | 53.34 | 51.59 | 51.59 | changed |
| H2 | abstract / conclusion: belief-trained raw and utility-trained features (dev-selected OOD micro utility) | BELT-H - layer16 (pp, 5 prices): 4.89, 4.66, 4.78, 5.76, 6.94; BELT-H - OSP-1 features (pp, 5 prices): 1.64, 1.62, 1.57, 1.92, 1.4 | BELT-H - layer16 (pp, 5 prices): 1.14, 1.22, 1.4, 2.17, 3.61; BELT-H - OSP-1 features (pp, 5 prices): 0.68, 0.67, 0.69, 0.94, 0.32 | BELT-H - layer16 (pp, 5 prices): 0.98, 1.06, 1.21, 2.11, 4.43; BELT-H - OSP-1 features (pp, 5 prices): 0.52, 0.51, 0.5, 0.88, 1.14 | holds |
| H3 | abstract: trade-offs "competitive with or better than" end-to-end controllers (34 math readings) | vs OSP-full W/T/L: 11, 20, 2; vs OSP-1 W/T/L: 19, 9, 6 | vs OSP-full W/T/L: 11, 12, 11; vs OSP-1 W/T/L: 31, 0, 3 | vs OSP-full W/T/L: 17, 13, 4; vs OSP-1 W/T/L: 27, 6, 1 | rewritten |
| H4 | abstract: 6.3x fewer controller parameters | 6.30 | 6.31 | 6.31 | holds |
| H4b | abstract: ~25x less training compute (five prices) | 25 | 22 | 22 | rewritten |
| Setting | Qwen3 | OLMo3 |
|---|---|---|
| Optimizer | AdamW | |
| Encoder learning rate | ||
| Belief-head learning rate | ||
| AdamW ; | ; | |
| Weight decay (all parameter groups) | ||
| Gradient clipping (global norm) | ||
| Setting | Value |
|---|---|
| Optimizer / precision | L-BFGS / float64 |
| Learning rate / line search | 1.0 / strong Wolfe |
| History size | 20 |
| Maximum iterations | 500 |
| Gradient / change tolerance | / |
| Development evaluation | Every 25 iterations |
| Method | Probe Params | DAPO | Numina | DSR | Micro | Macro |
|---|---|---|---|---|---|---|
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M | |||||
| OSP | 101M | |||||
| OSP-1 | 101M |
| Method | Probe Params | M500 | Olymp. | GK | AIME | AMC |
|---|---|---|---|---|---|---|
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M | |||||
| OSP | 101M | |||||
| OSP-1 | 101M |
| Method | Probe Params | DAPO | Numina | DSR | Micro | Macro |
|---|---|---|---|---|---|---|
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M | |||||
| OSP | 101M | |||||
| OSP-1 | 101M |
| Method | Probe Params | M500 | Olymp. | GK | AIME | AMC |
|---|---|---|---|---|---|---|
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M | |||||
| OSP | 101M | |||||
| OSP-1 | 101M |
| Method | Probe Params | DAPO | Numina | DSR | Micro | Macro |
|---|---|---|---|---|---|---|
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M | |||||
| OSP | 101M | |||||
| OSP-1 | 101M |
| Method | Probe Params | M500 | Olymp. | GK | AIME | AMC |
|---|---|---|---|---|---|---|
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M | |||||
| OSP | 101M | |||||
| OSP-1 | 101M |
| Method | Probe Params | DAPO | Numina | DSR | Micro | Macro |
|---|---|---|---|---|---|---|
| Natural †‡ | – | |||||
| Fixed cap ‡ | 0 | |||||
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M |
| Method | Probe Params | M500 | Olymp. | GK | AIME | AMC |
|---|---|---|---|---|---|---|
| Natural †‡ | – | |||||
| Fixed cap ‡ | 0 | |||||
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M |
| Method | Probe Params | DAPO | Numina | DSR | Micro | Macro |
|---|---|---|---|---|---|---|
| Natural †‡ | – | |||||
| Fixed cap ‡ | 0 | |||||
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M |
| Method | Probe Params | M500 | Olymp. | GK | AIME | AMC |
|---|---|---|---|---|---|---|
| Natural †‡ | – | |||||
| Fixed cap ‡ | 0 | |||||
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M |
| Method | Probe Params | DAPO | Numina | DSR | Micro | Macro |
|---|---|---|---|---|---|---|
| Natural †‡ | – | |||||
| Fixed cap ‡ | 0 | |||||
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M |
| Method | Probe Params | M500 | Olymp. | GK | AIME | AMC |
|---|---|---|---|---|---|---|
| Natural †‡ | – | |||||
| Fixed cap ‡ | 0 | |||||
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M |
| Method | Probe Params | DAPO | Numina | DSR | Micro | Macro |
|---|---|---|---|---|---|---|
| Natural †‡ | – | |||||
| Fixed cap ‡ | 0 | |||||
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M |
| Method | Probe Params | M500 | Olymp. | GK | AIME | AMC |
|---|---|---|---|---|---|---|
| Natural †‡ | – | |||||
| Fixed cap ‡ | 0 | |||||
| CoDE-Stop ‡ | 0 | |||||
| DEER ‡ | 0 | |||||
| TC | 514 | |||||
| ROM | 5M |
| Method | Probe Params | 0 pp | 1 pp | 2 pp |
|---|---|---|---|---|
| CoDE-Stop ‡ | 0 | |||
| DEER ‡ | 0 | |||
| TC | 514 | |||
| ROM | 8M | |||
| OSP | 405M | |||
| OSP+T | 405M |
| Method | Probe Params | 0 pp | 1 pp | 2 pp |
|---|---|---|---|---|
| CoDE-Stop ‡ | 0 | |||
| DEER ‡ | 0 | |||
| TC | 514 | |||
| ROM | 8M | |||
| OSP | 405M | |||
| OSP+T | 405M |
| Method | Probe Params | 70% | 60% | 50% | 40% |
|---|---|---|---|---|---|
| Natural †‡ | – | ||||
| Fixed cap ‡ | 0 | ||||
| CoDE-Stop ‡ | 0 | ||||
| DEER ‡ | 0 | ||||
| TC | 514 | ||||
| ROM | 8M |
| Method | Probe Params | 70% | 60% | 50% | 40% |
|---|---|---|---|---|---|
| Natural †‡ | – | ||||
| Fixed cap ‡ | 0 | ||||
| CoDE-Stop ‡ | 0 | ||||
| DEER ‡ | 0 | ||||
| TC | 514 | ||||
| ROM | 8M |
| Method | Probe Params | 0 pp | 1 pp | 2 pp |
|---|---|---|---|---|
| Natural † | – | – | – | – |
| OSP | 101M | |||
| OSP-1 | 101M | |||
| OSP-1+T | 101M | |||
| BELT-B-R | 16M | |||
| BELT-H-R | 16M |
| Method | Probe Params | 70% | 60% | 50% | 40% |
|---|---|---|---|---|---|
| Natural † | – | ||||
| OSP | 101M | ||||
| OSP-1 | 101M | ||||
| OSP-1+T | 101M | ||||
| BELT-B-R | 16M | ||||
| BELT-H-R | 16M |
| Method | Probe Params | Token saving (%) at tolerance | Accuracy (%) at saving | |||||
| 0.00 | 0.01 | 0.02 | 70% | 60% | 50% | 40% | ||
| Natural † | – | – | – | – | ||||
| OSP | 405M | |||||||
| OSP-1 | 405M | |||||||
| BELT-B-T | 19M | |||||||
| BELT-H-T | 19M | |||||||
| Method | Probe Params | Token saving (%) at tolerance | Accuracy (%) at saving | |||||
| 0.00 | 0.01 | 0.02 | 70% | 60% | 50% | 40% | ||
| Natural † | – | – | – | – | ||||
| OSP | 405M | |||||||
| OSP-1 | 405M | |||||||
| BELT-B-T | 19M | |||||||
| BELT-H-T | 19M | |||||||
| Method | Probe Params | Token saving (%) at tolerance | Success (%) at saving | |||||
| 0 pp | 2 pp | 5 pp | 50% | 40% | 30% | 20% | ||
| Natural | – | – | – | – | ||||
| OSP-1 | 405M | |||||||
| BELT-B-T | 19M | |||||||
| BELT-H-T | 19M | |||||||
| BELT-B-R | 19M | |||||||
| Method | Run | Success / realised saving [projected saving] (%) | |||
|---|---|---|---|---|---|
| OSP-1 | s0 | 29.9 / 21.5 [20.0] | 29.1 / 36.8 [35.0] | 26.1 / 51.9 [50.0] | 22.4 / 65.6 [65.0] |
| BELT-B-T | s0 | 35.1 / 21.5 [20.0] | 33.6 / 37.1 [35.0] | 23.1 / 52.3 [50.0] | 20.9 / 68.3 [65.0] |
| BELT-B-T | s1 | 32.1 / 21.3 [20.0] | 29.9 / 35.7 [35.0] | 26.1 / 50.1 [50.0] | 19.4 / 63.3 [65.0] |
| BELT-B-T | s2 | 25.4 / 18.4 [20.0] | 27.6 / 34.7 [35.0] | 26.9 / 51.8 [50.0] | 24.6 / 68.2 [65.0] |
| BELT-H-T | s0 | 24.6 / 13.9 [20.0] | 24.6 / 29.4 [35.0] | 29.9 / 45.8 [50.0] | 22.4 / 63.5 [65.0] |
| BELT-H-T | s1 | 29.9 / 19.7 [20.0] | 29.1 / 33.5 [35.0] | 26.9 / 50.2 [50.0] | 17.9 / 64.3 [65.0] |
| Comparison | Saving difference at tolerance | Success difference at saving | |||||
|---|---|---|---|---|---|---|---|
| 0 pp | 2 pp | 5 pp | 50% | 40% | 30% | 20% | |
| BELT-B-T vs. OSP-1 | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] |
| BELT-H-T vs. OSP-1 | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] |
| BELT-H-R vs. OSP-1 | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] |
| Method | Probe Params | Token saving (%) at tolerance | Score at saving | |||||
| 0 pt | 2 pt | 5 pt | 80% | 70% | 60% | 50% | ||
| Natural | – | – | – | – | ||||
| OSP-1 | 405M | |||||||
| BELT-B-T | 19M | |||||||
| BELT-H-T | 19M | |||||||
| BELT-B-R | 19M | |||||||
| Method | Run | Score / realised saving [projected saving] (%) | ||||||
|---|---|---|---|---|---|---|---|---|
| OSP-1 | s0 | 20.7 / 10.9 [20.0] | 23.3 / 26.3 [35.0] | 27.3 / 34.9 [50.0] | 28.0 / 50.2 [65.0] | 25.7 / 69.9 [80.0] | 25.0 / 83.0 [90.0] | 24.8 / 84.9 [91.5] |
| BELT-B-T | s0 | 26.5 / 13.9 [20.0] | 28.0 / 28.9 [35.0] | 26.2 / 38.1 [50.0] | 26.0 / 49.1 [65.0] | 26.5 / 65.4 [80.0] | 23.9 / 83.6 [90.0] | 24.8 / 84.9 [91.5] |
| BELT-B-T | s1 | 23.8 / 12.8 [20.0] | 22.1 / 21.4 [35.0] | 18.5 / 34.9 [50.0] | 23.4 / 54.3 [65.0] | 24.4 / 67.6 [80.0] | 24.4 / 81.8 [90.0] | 24.8 / 84.9 [91.5] |
| BELT-B-T | s2 | 24.6 / 10.3 [20.0] | 25.2 / 23.9 [35.0] | 28.3 / 38.2 [50.0] | 29.2 / 48.1 [65.0] | 26.9 / 69.6 [80.0] | 24.8 / 83.0 [90.0] | 25.7 / 85.0 [91.5] |
| BELT-H-T | s0 | 18.1 / 15.6 [20.0] | 27.4 / 27.4 [35.0] | 26.5 / 40.6 [50.0] | 28.2 / 55.5 [65.0] | 22.6 / 74.5 [80.0] | 21.4 / 83.2 [90.0] | 24.8 / 84.9 [91.5] |
| BELT-H-T | s1 | 22.5 / 12.9 [20.0] | 25.6 / 23.8 [35.0] | 24.2 / 39.9 [50.0] | 26.9 / 53.8 [65.0] | 27.4 / 70.0 [80.0] | 26.5 / 83.1 [90.0] | 24.8 / 85.0 [91.5] |
| Comparison | Saving difference at tolerance | Score difference at saving | |||||
|---|---|---|---|---|---|---|---|
| 0 pt | 2 pt | 5 pt | 80% | 70% | 60% | 50% | |
| BELT-H-T vs. OSP-1 | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] |
| BELT-B-T vs. OSP-1 | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] |
| BELT-H-R vs. OSP-1 | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] |
| BELT-B-R vs. OSP-1 | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] | [ , ] |
| Method | Utility by ( ) | ||||
|---|---|---|---|---|---|
| 2 | 5 | 10 | 20 | 50 | |
| Math ID | |||||
| OSP | |||||
| OSP-1 | |||||
| BELT-B-R | |||||
| BELT-H-R | |||||
| Method | 0 pp | 1 pp | 2 pp |
|---|---|---|---|
| ID | |||
| OSP | |||
| OSP+T | |||
| OSP-1+T | |||
| BELT-B-T | |||
| BELT-H-T | |||
| Method | 70% | 60% | 50% | 40% |
|---|---|---|---|---|
| ID | ||||
| OSP | ||||
| OSP+T | ||||
| OSP-1+T | ||||
| BELT-B-T | ||||
| BELT-H-T | ||||
| Method | 0 pp | 1 pp | 2 pp |
|---|---|---|---|
| Pre | |||
| OSP | |||
| OSP+T | |||
| OSP-1+T | |||
| BELT-B-T | |||
| BELT-H-T | |||
| Method | 70% | 60% | 50% | 40% |
|---|---|---|---|---|
| Pre | ||||
| OSP | NF | |||
| OSP+T | ||||
| OSP-1+T | ||||
| BELT-B-T | ||||
| BELT-H-T | ||||
| Method | |||||
|---|---|---|---|---|---|
| Qwen3 ID (Macro) | |||||
| OSP | |||||
| OSP+T | |||||
| BELT-B-R | |||||
| BELT-B-R+T | |||||
| BELT-H-R | |||||
| Method | 70% | 60% | 50% | 40% |
|---|---|---|---|---|
| OSP | ||||
| OSP-1 | ||||
| BELT-B-T | ||||
| BELT-H-T | ||||
| BELT-B-R | ||||
| BELT-H-R |
| Method | 70% | 60% | 50% | 40% |
|---|---|---|---|---|
| OSP | ||||
| OSP-1 | ||||
| BELT-B-T | ||||
| BELT-H-T | ||||
| BELT-B-R | ||||
| BELT-H-R |
| Comparison | Policy fitting / development selection | Test grading | SD across |
|---|---|---|---|
| Qwen3 BELT utility controls | Numeric-equality labels | Strict math | 3 independent probe training runs |
| Fixed feature controls | Numeric-equality labels | Strict math | 5 policy-fitting starts |
| BELT-L | Numeric-equality labels | Strict math | 3 probe training seeds |
| Matched OSP-1/BELT features | Shared fitter; strict math development utility | Strict math; code unit tests | 3 independent training runs per method |
| OLMo3 BELT | Loose-label utility (refit) | Strict math | 3 independent probe training runs |
| Control | Utility by ( ) | ||||
|---|---|---|---|---|---|
| 2 | 5 | 10 | 20 | 50 | |
| (a) Features for a linear stopping policy | |||||
| Pretrained ReProbe representation | |||||
| Reasoner state (layer 16) | |||||
| Reasoner state (PCA, 768-d) | |||||
| Mean-pooled reasoner states | |||||
| Control | Utility by ( ) | ||||
|---|---|---|---|---|---|
| 2 | 5 | 10 | 20 | 50 | |
| BELT probe training targets | |||||
| Now , End | |||||
| Now , End , Later | |||||
| Now , End , Same | |||||
| All four | |||||
| Token saving (%) at tolerance | Accuracy (%) at saving | ||||||
| Belief set | 0 pp | 1 pp | 2 pp | 70% | 60% | 50% | 40% |
| (a) Retrain prediction targets | |||||||
| Now, End | |||||||
| Now, End, Later | NF | ||||||
| Now, End, Same | |||||||
| All four | |||||||
| Now, End, Same targets | All four targets | |||
|---|---|---|---|---|
| ( ) | Accuracy (%) | Total tokens | Accuracy (%) | Total tokens |
| 2 | ||||
| 5 | ||||
| 10 | ||||
| 20 | ||||
| 50 | ||||
| Token saving (%) at tolerance | Accuracy (%) at saving | ||||||
| Scope | 0 pp | 1 pp | 2 pp | 70% | 60% | 50% | 40% |
| Qwen3 ID | |||||||
| Qwen3 OOD | NF | ||||||
| Code Pre | |||||||
| Code Post | |||||||
| OLMo3 OOD | – | – | – | – | – | – | – |
| Features / fitting | 2 | 5 | 10 | 20 | 50 |
|---|---|---|---|---|---|
| OSP-1 features | |||||
| OSP-1 features (four starts) | |||||
| BELT-H-R | |||||
| BELT-H-R (four starts) | |||||
| BELT-B-R | |||||
| BELT-B-R (four starts) |
| Features / fitting | 2 | 5 | 10 | 20 | 50 |
|---|---|---|---|---|---|
| OSP-1 features | |||||
| OSP-1 features (four starts) | |||||
| BELT-H-R | |||||
| BELT-H-R (four starts) | |||||
| BELT-B-R | |||||
| BELT-B-R (four starts) |
| Features / fitting | 2 | 5 | 10 | 20 | 50 |
|---|---|---|---|---|---|
| OSP-1 features | |||||
| OSP-1 features (four starts) | |||||
| BELT-H-R | |||||
| BELT-H-R (four starts) | |||||
| BELT-B-R | |||||
| BELT-B-R (four starts) |
| Features / fitting | 2 | 5 | 10 | 20 | 50 |
|---|---|---|---|---|---|
| OSP-1 features | |||||
| OSP-1 features (four starts) | |||||
| BELT-H-R | |||||
| BELT-H-R (four starts) | |||||
| BELT-B-R | |||||
| BELT-B-R (four starts) |
| Features / fitting / seed | Utility by ( ) | ||||
|---|---|---|---|---|---|
| 2 | 5 | 10 | 20 | 50 | |
| OSP-1 features, main, s0 | 0.6027 | 0.5897 | 0.5679 | 0.5202 | 0.4172 |
| OSP-1 features, main, s1 | 0.6278 | 0.6090 | 0.5788 | 0.5207 | 0.4099 |
| OSP-1 features, main, s2 | 0.6204 | 0.6053 | 0.5800 | 0.5309 | 0.4237 |
| OSP-1 features, four starts, s0 | 0.6027 | 0.5897 | 0.5679 | 0.5202 | 0.4177 |
| OSP-1 features, four starts, s1 | 0.6278 | 0.6090 | 0.5788 | 0.5207 | 0.4097 |
| Features / fitting / seed | Utility by ( ) | ||||
|---|---|---|---|---|---|
| 2 | 5 | 10 | 20 | 50 | |
| OSP-1 features, main, s0 | 0.2794 | 0.2640 | 0.2383 | 0.2015 | 0.1223 |
| OSP-1 features, main, s1 | 0.2867 | 0.2615 | 0.2221 | 0.1811 | 0.1120 |
| OSP-1 features, main, s2 | 0.2841 | 0.2651 | 0.2333 | 0.1867 | 0.1047 |
| OSP-1 features, four starts, s0 | 0.2794 | 0.2640 | 0.2383 | 0.2015 | 0.1228 |
| OSP-1 features, four starts, s1 | 0.2867 | 0.2615 | 0.2221 | 0.1811 | 0.1116 |
| Scope | Interface | Refit | Threshold | Threshold Refit | Positive seeds |
|---|---|---|---|---|---|
| Math ID | H | 3/3 | |||
| Math OOD | H | 2/3 | |||
| Code Pre | B | 3/3 | |||
| Code Post | B | 0/3 |
| Setting | B/H | Contrast | Save 0 | Save 1 | Save 2 | Acc 0.3 | Acc 0.4 | Acc 0.5 | Acc 0.6 |
|---|---|---|---|---|---|---|---|---|---|
| Qwen3 ID | B | R+T T | |||||||
| B | R T | ||||||||
| H | R+T T | ||||||||
| H | R T | ||||||||
| Qwen3 OOD | B | R+T T | |||||||
| B | R T |
| Architecture / prediction targets | Utility by ( ) | ||||
|---|---|---|---|---|---|
| 2 | 5 | 10 | 20 | 50 | |
| BELT probe / Now , End | |||||
| Top-two branch / Now , End | |||||
| BELT probe / Now , End , Later | |||||
| Top-two branch / Now , End , Later | |||||
| Setting | Policy | Save 1 | 2e-6 | 5e-6 | 1e-5 | 2e-5 | 5e-5 |
|---|---|---|---|---|---|---|---|
| Qwen3 OOD | Now + length | ||||||
| Now , Same + length | |||||||
| All four + length (B-R) | |||||||
| Dense forced cap | |||||||
| BELT-H-R | |||||||
| OSP |
| Setting | Cap family | Save 0 | Save 1 | Save 2 | 1e-5 | 2e-5 | 5e-5 |
|---|---|---|---|---|---|---|---|
| Qwen3 ID | Published | ||||||
| Dense | |||||||
| Qwen3 OOD | Published | ||||||
| Dense | |||||||
| OLMo3 | Published | ||||||
| Dense |
| 2.00 | 18.91 | 108.58 | |
|---|---|---|---|
| Length | |||
| Threshold | |||
| Scale |
| Qwen ID | Qwen OOD | LCB Pre | LCB Post | |||||
|---|---|---|---|---|---|---|---|---|
| Reading | H-R | B-R | H-R | B-R | H-R | B-R | H-R | B-R |
| Save 0 | ||||||||
| Save 1 | ||||||||
| Save 2 | ||||||||
| Acc 0.3 | ||||||||
| Acc 0.4 | ||||||||
| Backbone | BELT | OSP |
|---|---|---|
| Qwen3 | 15,957,765 | 100,676,098 |
| OLMo3 | 19,103,494 | 404,791,298 |
| Method | Qwen3 ID/OOD (2,719) | OLMo3 OOD (1,025) | LiveCodeBench (549 + 416) | ||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Probe Params | med. | max | SD | Probe Params | med. | max | SD | Probe Params | med. | max | SD | ||||
| Natural | – | 1 | +0.000 | 0.00 | 0.01 | – | – | – | – | – | – | – | – | – | – |
| Fixed cap | 0 | 5 | +0.000 | 1.13 | 1.07 | 0 | 8 | +0.002 | 1.20 | 0.66 | – | – | – | – | – |
| CoDE-Stop | 0 | 225 | +0.001 | 1.52 | 1.07 | 0 | 75 | +0.001 | 0.79 | 0.65 | – | – | – | – | – |
| DEER | 0 | 54 | +0.001 | 1.31 | 1.05 | 0 | 18 | +0.000 | 0.99 | 0.61 | – | – | – | – | – |
| TC | 514 | 300 | +0.002 | 0.86 | 1.00 | 514 | 150 | +0.002 | 0.93 | 1.14 | – | – | – | – | – |
| Setting | Comparator | BELT | Comparator | Verdict | Pooled interval; 2 SE | |
|---|---|---|---|---|---|---|
| Qwen3 ID | full OSP | 50.64 | 49.72 | T | – | |
| Qwen3 ID | OSP-1 | 50.64 | 46.59 | W | ; no | |
| Qwen3 OOD | full OSP | 45.98 | 45.19 | T | – | |
| Qwen3 OOD | OSP-1 | 45.98 | 36.89 | W | ; yes | |
| LiveCodeBench Pre | full OSP | 38.25 | 38.25 | T | – | |
| LiveCodeBench Pre | OSP-1 | 38.25 | 15.25 | W | ; yes |
| Variant | Comparator | Qwen3 ID | Qwen3 OOD | LCB Pre | LCB Post | OLMo3 | OLMo3 LCB before 8/2024 | OLMo3 LCB from 8/2024 | Total |
|---|---|---|---|---|---|---|---|---|---|
| H-R | full OSP | 1/1/5 1/2 | 0/6/1 0/0 | 1/5/0 0/0 (NF 1) | 0/3/4 0/2 | 6/1/0 6/0 | 5/2/0 1/0 | 0/6/1 0/0 | 13/24/11 8/4 |
| OSP-1 | 4/3/0 4/0 | 5/2/0 5/0 | 7/0/0 7/0 | 3/4/0 1/0 | 7/0/0 6/0 | 7/0/0 5/0 | 5/2/0 3/0 | 38/11/0 31/0 | |
| OSP+T | 0/3/4 0/4 | 0/1/6 0/3 | 0/5/2 0/1 | 0/0/7 0/7 | 5/2/0 4/0 | – | – | 5/11/19 4/15 | |
| H-T | full OSP | 1/5/1 0/1 | 1/4/2 1/2 | 0/6/0 0/0 (NF 1) | 0/6/1 0/0 | 2/4/1 1/0 | 2/5/0 0/0 | 0/4/3 0/1 | 6/34/8 2/4 |
| OSP-1 | 6/0/1 6/1 | 5/0/2 5/1 | 7/0/0 7/0 | 2/5/0 0/0 | 5/1/1 3/0 | 7/0/0 2/0 | 3/4/0 2/0 | 35/10/4 25/2 | |
| OSP+T | 0/6/1 0/1 | 0/3/4 0/3 | 0/5/2 0/1 | 0/0/7 0/1 | 1/5/1 0/0 | – | – | 1/19/15 0/6 |
| Setting | Variant | Comparator | Save 0 | Save 1 | Save 2 | Acc 0.3 | Acc 0.4 | Acc 0.5 | Acc 0.6 |
|---|---|---|---|---|---|---|---|---|---|
| Qwen3 ID | H-R | full OSP | L e | L e | L r | W r | T | L e | L r |
| H-R | OSP-1 | T | T | T | W r | W r | W r | W r | |
| H-R | OSP+T | L r | T | L r | T | L r | T | L r | |
| H-T | full OSP | W e | T | T | L r | T | T | T | |
| H-T | OSP-1 | W r | W r | W r | L r | W r | W r | W r | |
| H-T | OSP+T | T | T | T | L r | T | T | T |
| Variant | Comparator | Qwen3 ID | Qwen3 OOD | LCB Pre | LCB Post | OLMo3 | OLMo3 LCB before 8/2024 | OLMo3 LCB from 8/2024 | Total |
|---|---|---|---|---|---|---|---|---|---|
| H-R | full OSP | 2/5/0 2/0 | 3/4/0 3/0 | 6/0/0 5/0 (NF 1) | 0/7/0 0/0 | 6/1/0 6/0 | 5/2/0 1/0 | 0/6/1 0/0 | 22/25/1 17/0 |
| OSP-1 | 7/0/0 7/0 | 7/0/0 6/0 | 7/0/0 7/0 | 7/0/0 6/0 | 7/0/0 6/0 | 7/0/0 5/0 | 5/2/0 3/0 | 47/2/0 40/0 | |
| OSP+T | 1/5/1 0/0 | 1/4/2 1/1 | 3/4/0 2/0 | 0/0/7 0/1 | 5/2/0 4/0 | – | – | 10/15/10 7/2 | |
| H-T | full OSP | 2/5/0 1/0 | 0/6/1 0/1 | 5/1/0 4/0 (NF 1) | 0/7/0 0/0 | 2/4/1 1/0 | 2/5/0 0/0 | 0/4/3 0/1 | 11/32/5 6/2 |
| OSP-1 | 5/2/0 5/0 | 4/2/1 4/1 | 7/0/0 7/0 | 2/5/0 2/0 | 5/1/1 3/0 | 7/0/0 2/0 | 3/4/0 2/0 | 33/14/2 25/1 | |
| OSP+T | 1/6/0 0/0 | 0/5/2 0/1 | 2/4/1 1/0 | 0/6/1 0/0 | 1/5/1 0/0 | – | – | 4/26/5 1/1 |
| Seed | Recipe | Near-never-stop checkpoints | Saving (%) at 1 pp | Acc. (%) at 0.5 cost | Acc. (%) at 0.3 cost | Utility |
|---|---|---|---|---|---|---|
| 0 | Direct | 2/4 | 6.47 | 62.98 | 52.36 | 0.627 |
| 0 | Belief warm-up direct | 0/4 | 5.89 | 61.83 | 52.46 | 0.627 |
| 1 | Direct | 0/2 | 4.75 | 62.12 | 53.72 | 0.627 |
| 1 | Belief warm-up direct | 0/2 | 2.94 | 54.05 | 41.32 | 0.627 |
| 2 | Direct | 0/1 | 7.35 | 65.48 | 56.45 | 0.627 |
| 2 | Belief warm-up direct | 1/1 | 1.21 | 34.39 | 25.25 | 0.627 |
| Run | Source-val | Native OOD | OOD [95% CI] |
|---|---|---|---|
| Seed 1 | [ ] | ||
| Seed 2 | [ ] |