CoT2-Meta: Budgeted Metacognitive Control for Test-Time Reasoning
Organizations: Nanyang Technological University · Carnegie Mellon University · Zhejiang University · Sun Yat-Sen University · National University of Singapore · Fudan University · City University of Hong Kong (Dongguan) · Zhejiang Lab
Abstract
Recent test-time reasoning methods improve performance by generating more candidate chains or searching over larger reasoning trees, but they typically lack explicit control over when to expand, what to prune, how to repair, and when to abstain. We introduce CoT2-Meta, a training-free metacognitive reasoning framework that combines object-level chain-of-thought generation with meta-level control over partial reasoning trajectories. The framework integrates four components: strategy-conditioned thought generation, tree-structured search, an online process oracle for step-level reasoning evaluation, and a meta-controller that allocates computation through expansion, pruning, repair, stopping, and fallback decisions. Under matched inference budgets, CoT2-Meta consistently outperforms strong single-path, sampling-based, and search-based baselines, including ReST-MCTS. On the default backbone, it achieves 92.8 EM on MATH, 90.4 accuracy on GPQA, 98.65 EM on GSM8K, 75.8 accuracy on BBEH, 85.6 accuracy on MMMU-Pro, and 48.8 accuracy on HLE, with gains over the strongest non-CoT2-Meta baseline of +3.6, +5.2, +1.15, +2.0, +4.3, and +4.3 points, respectively. Beyond these core results, the framework remains effective across a broader 15-benchmark suite spanning knowledge and QA, multi-hop reasoning, coding, and out-of-distribution evaluation. Additional analyses show better compute scaling, improved calibration, stronger selective prediction, targeted repair behavior, and consistent gains across backbone families. These results suggest that explicit metacognitive control is a practical design principle for reliable and compute-efficient test-time reasoning systems.
Figures & tables
| Category | Dataset | Greedy CoT | Best-of-16 | Vanilla ToT | ReST-MCTS* | Ours (CoT 2 -Meta) | Gain |
| Mathematics | MATH ( Hendrycks et al., 2021 ) | 78.5 | 84.2 | 87.4 | 89.2 | 92.8 | +3.6 |
| GSM8K ( Cobbe et al., 2021 ) | 94.2 | 96.1 | 96.8 | 97.5 | 98.6 | +1.1 | |
| Knowledge & QA | GPQA ( Rein et al., 2024 ) | 74.2 | 79.6 | 83.5 | 85.2 | 90.4 | +5.2 |
| MMLU-Pro ( Wang et al., 2024 ) | 78.5 | 82.2 | 84.5 | 86.1 | 88.4 | +2.3 | |
| TruthfulQA ( Lin et al., 2022 ) | 81.5 | 85.4 | 88.2 | 90.2 | 92.8 | +2.6 | |
| Natural Questions ( Kwiatkowski et al., 2019 ) | 36.5 | 39.8 | 41.5 | 43.2 | 45.4 | +2.2 |
| Backbone | Strategy | MATH | GPQA | BBEH | MMMU-Pro | (Avg) |
| Claude-4.5 | Greedy CoT ( Wei et al., 2022 ) | 78.5 | 74.2 | 65.4 | 68.4 | – |
| Best-of-16 ( Wang et al., 2022 ) | 84.2 | 79.6 | 68.9 | 73.1 | +4.8 | |
| Vanilla ToT ( Yao et al., 2023 ) | 87.4 | 83.5 | 71.2 | 77.8 | +8.3 | |
| Ours (CoT 2 -Meta) | 92.8 | 90.4 | 75.8 | 85.6 | +14.5 | |
| DeepSeek-V3.2 | Greedy CoT ( Wei et al., 2022 ) | 70.8 | 82.4 | 74.4 | 64.6 | – |
| Best-of-16 ( Wang et al., 2022 ) | 75.3 | 85.1 | 75.9 | 68.2 | +3.0 |
| Variant Configuration | MATH | GPQA | BBEH | Avg Calls | Avg Tokens |
| Full CoT 2 -Meta (Ours) | 92.8 | 90.4 | 75.8 | 15.2 | 5,280 |
| w/o Thought Tree (Single-chain) | 78.5 | 74.2 | 65.4 | 1.0 | 1,929 |
| w/o Process Evaluator | 86.5 | 82.3 | 69.3 | 16.0 | 6,120 |
| w/o Metacognitive Pruning | 88.2 | 84.1 | 71.3 | 16.0 | 9,850 |
| Process-only Controller ( ) | 89.5 | 86.2 | 72.5 | 16.0 | 5,840 |
| Outcome-only Controller ( ) | 87.8 | 84.5 | 70.1 | 16.0 | 6,430 |
Appendix figures & tables28 assets
Supplementary material from the paper’s appendix.
Appendix
| Signal | Meaning | Parsing Rule |
| Semantic | Semantic consistency with the problem and the current reasoning goal | Numeric score in |
| Logical | Whether the current step is internally coherent and supports subsequent reasoning | Numeric score in |
| Fix | Whether the step explicitly detects, revises, rejects, or corrects a previous incorrect intermediate claim | Numeric score in |
| Method | Gen. Calls | Eval. Calls | Verify Calls | Repair Calls | Selection Overhead | Fallback Calls | Total Budgeted |
| Greedy CoT | 1 | 0 | 0 | 0 | 0 | 0 | 1 |
| Best-of-16 | 16 | 0 | 0 | 0 | 0 | 0 | 16 |
| Vanilla ToT | branch exp. | 0 | 0 | 0 | 0 | 0 | budget-matched |
| ReST-MCTS | branch exp. | value/reward q. | 0 | 0 | controller q. | 0 | budget-matched |
| CoT 2 -Meta | branch exp. | process-oracle q. | verify-style conf. q. | selective repair q. | controller q. | optional fallback q. | budget-matched |
| Active Backbone | Role | Backend / Access Mode | Temp. | Max Tokens | Usage |
| Claude-4.5 | Object-level generator | Official API | 0.4 | 2048 | Direct / decompose / verify generation |
| Claude-4.5 | Online process oracle | Official API | 0.0 | 512 | Structured step-level process evaluation |
| Claude-4.5 | Fallback policy | Official API | 0.4 | 2048 | Best-of-16 voting fallback |
| DeepSeek-V3.2 | Object-level generator, oracle, and fallback | Official API | task-dependent | task-dependent | Cross-backbone evaluation |
| Qwen2.5-VL-7B-Instruct | Object-level generator, oracle, and fallback | vLLM local deployment | task-dependent | task-dependent | Cross-backbone evaluation |
| Category | Dataset | Metric | Baseline | Ours (CoT 2 -Meta) | Gain | -value |
| Reasoning | MATH | EM | 87.40 0.9 | 92.80 0.4 | +5.40 | |
| GPQA | Acc. | 83.50 1.2 | 90.40 0.6 | +6.90 | ||
| BBEH | Acc. | 71.20 0.8 | 75.80 0.5 | +4.60 | 0.002 | |
| Multimodal | MMMU-Pro | Acc. | 77.80 1.1 | 85.60 0.7 | +7.80 | |
| HLE | Acc. | 42.10 1.5 | 48.80 0.9 | +6.70 | 0.002 |
| Dataset | Method | at (vs. ToT) | |||
| GPQA | Best-of-16 | 79.60 | 82.10 | 84.20 | – |
| Vanilla ToT | 76.80 | 80.10 | 83.50 | – | |
| ReST-MCTS | 78.40 | 81.50 | 85.20 | +1.60 | |
| CoT 2 -Meta | 82.30 | 87.40 | 90.40 | +5.50 | |
| BBEH | Best-of-16 | 65.80 | 67.50 | 68.90 | – |
| Vanilla ToT | 68.20 | 71.40 | 72.80 | – |
| Dataset | ||||||
| SBMB | Ours | SBMB | Ours | SBMB | Ours | |
| GSM8K | 95.10 | 97.20 | 96.20 | 98.10 | 96.80 | 98.60 |
| MATH | 81.20 | 86.50 | 84.50 | 90.20 | 87.40 | 92.80 |
| GPQA | 76.80 | 82.30 | 80.10 | 87.40 | 83.50 | 90.40 |
| MMLU-Pro | 82.10 | 84.50 | 84.20 | 86.90 | 86.10 | 88.40 |
| TruthfulQA | 86.40 | 89.20 | 88.50 | 91.40 | 90.20 | 92.80 |
| Dataset (Radar Axis) | Context Trunc. (S4) | Noisy Format (S5) | ||
| S.B.M.B. | Ours | S.B.M.B. | Ours | |
| GSM8K (EM) | 90.20 | 94.50 | 91.20 | 97.80 |
| MATH (EM) | 78.40 | 88.10 | 79.50 | 91.50 |
| GPQA (Acc.) | 72.40 | 85.20 | 75.60 | 89.20 |
| MMLU-Pro (Acc.) | 79.50 | 84.30 | 80.20 | 87.50 |
| TQA (Binary BvI Acc.) | 84.10 | 89.10 | 82.10 | 92.10 |
| Textified Vision Axis (S6) | S.B.M.B. | Ours |
| MMMU-Pro (Overall) | 52.40 | 79.20 |
| HLE-Vision (Overall) | 28.30 | 44.60 |
| Science (Bio, Chem, Phys) | 32.50 | 65.20 |
| Health & Medicine | 38.40 | 68.30 |
| Tech & Engineering | 22.10 | 55.10 |
| Art & Design | 45.10 | 71.40 |
| Total LLM Calls | Greedy CoT | Best-of- | Vanilla ToT | Q* Search | ReST- MCTS | rStar- Math | CoT 2 - Meta |
| 1 | 50.2 | 50.2 | 50.2 | 50.2 | 50.2 | 50.2 | 50.2 |
| 2 | – | 53.4 | 52.8 | 53.5 | 53.1 | 53.8 | 54.5 |
| 4 | – | 56.8 | 58.5 | 59.2 | 58.8 | 60.1 | 62.4 |
| 8 | – | 59.5 | 64.2 | 66.1 | 67.4 | 68.2 | 70.5 |
| 16 | – | 62.1 | 68.7 | 71.5 | 74.2 | 74.8 | 77.1 |
| 32 | – | 64.3 | 71.4 | 75.2 | 79.5 | 80.3 | 82.4 |
| Method | Avg In Tok. | Avg Out Tok. | Avg Total Tok. | Acc. (%) | Acc. / 1K Tok. | Tok. / +1% Acc |
| Greedy CoT ( ) | 1,244 | 685 | 1,929 | 50.2 | 26.02 | – |
| Self-Consistency (Maj@16) | 1,984 | 7,488 | 9,472 | 62.1 | 6.56 | 634 |
| Vanilla ToT | 4,350 | 2,120 | 6,470 | 68.7 | 10.62 | 245 |
| ReST-MCTS | 7,840 | 6,150 | 13,990 | 74.2 | 5.30 | 502 |
| CoT 2 -Meta (Ours) | 3,820 | 1,460 | 5,280 | 77.1 | 14.60 | 124 |
| Difficulty | Count ( ) | Greedy ( ) | Vanilla ToT | ReST-MCTS | CoT 2 -Meta |
| Easy (L1–L2) | 150 | 81.3% | 93.3% | 94.0% | 95.3% |
| Medium (L3–L4) | 200 | 42.5% | 65.0% | 74.5% | 79.5% |
| Hard (L5) | 150 | 29.3% | 48.6% | 54.0% | 55.3% |
| Overall | 500 | 50.2% | 68.7% | 74.2% | 77.1% |
| Failure Type | Defining Characteristics (Automated Rule) | Count | % of Fails | Avg Calls |
| F1: Search Not Converged | Budget exhausted ( ); top- leaves show high entropy; no path surpasses the confidence threshold. | 21 | 18.3% | 16.0 |
| F2: Evaluator Misjudgment | Oracle assigns a higher process reward to an incorrect leaf than to a generated correct leaf. | 16 | 13.9% | 16.0 |
| F3: Over-pruning | Shadow run (looser pruning bounds) solves the case; the correct trajectory was pruned too early. | 11 | 9.6% | 10.4 |
| F4: Base-Model Limit | All baselines fail; no valid reasoning trace is generated; the case appears to require knowledge absent in the LLM. | 67 | 58.2% | 13.5 |
| Total Failures | – | 115 | 100% | – |
| Method | Topology subset ( ) | Phys/Chem subset ( ) | Combinatorics subset ( ) |
| Greedy CoT ( ) | 35.4 0.0 | 41.5 0.0 | 45.4 0.0 |
| Self-Consistency (Best-of-16) | 37.1 0.5 | 43.2 0.4 | 47.8 0.6 |
| Vanilla ToT ( ) | 39.7 1.0 | 47.5 0.9 | 50.7 0.6 |
| ReST-MCTS ( ) | 42.3 0.8 | 50.1 0.7 | 53.5 0.5 |
| CoT 2 -Meta (Ours, ) | 44.4 0.7 | 52.8 0.3 | 55.9 0.4 |
| Method | Accuracy ( ) | ECE ( ) | Brier Score ( ) | AURC ( ) |
| Best-of-16 | 62.1% | 0.158 [0.142, 0.175] | 0.235 [0.215, 0.255] | 0.278 [0.258, 0.298] |
| Vanilla ToT | 68.7% | 0.124 [0.109, 0.140] | 0.205 [0.188, 0.222] | 0.224 [0.205, 0.243] |
| ReST-MCTS ∗ | 74.2% | 0.092 [0.078, 0.107] | 0.154 [0.140, 0.169] | 0.162 [0.147, 0.178] |
| Pure CoT 2 -Meta (Standalone, No Fallback) | 77.1% | 0.035 [0.026, 0.045] | 0.126 [0.112, 0.141] | 0.108 [0.095, 0.122] |
| Verify Prompt Variant | ECE ( ) | Brier Score ( ) | AURC ( ) |
| Variant A (Current default) | 0.035 [0.026, 0.045] | 0.126 [0.112, 0.141] | 0.108 [0.095, 0.122] |
| Variant B (Wording change) | 0.037 [0.028, 0.047] | 0.129 [0.114, 0.144] | 0.110 [0.096, 0.125] |
| Variant C (Constrained numeric) | 0.032 [0.023, 0.042] | 0.124 [0.110, 0.138] | 0.106 [0.093, 0.120] |
| Variant | Accuracy | Avg Tokens | Rescue Rate ( ) | Hurt Rate ( ) | Net Trajectory Gain |
| No Repair | 73.4% | 4,650 | – | – | – |
| Repair-All (No selective trigger) | 74.8% | 6,920 | 18.2% | 31.4% | +107 |
| Full Selective Repair (Ours) | 77.1% | 5,280 | 42.5% | 11.6% | +275 |
| System setup | Accuracy | Avg. tokens | Acc. vs. matched |
| No Repair (Base) | 73.4% | 4,650 | – |
| No Repair + Extra Expansion (Matched) | 74.6% | 5,290 | – |
| Full Selective Repair (Ours) | 77.1% | 5,280 | +2.5% |
| System setup | Fallback | Avg. calls | Accuracy | vs. best standalone |
| ReST-MCTS only | 100% | 16.0 | 74.2% | -2.9% |
| CoT 2 -Meta only | 0% | 16.0 | 77.1% | 0.0% |
| Hybrid cascade ( ) | 20% | 19.2 | 79.1% | +2.0% |
| Hybrid cascade ( ) | 40% | 22.4 | 79.4% | +2.3% |
| Field | Content |
| Role | You are a reasoning-process evaluator. |
| Inputs | • a problem • a multi-step reasoning trajectory |
| Instruction | Evaluate each reasoning step independently. |
| Returned scores | • Semantic : semantic alignment with the problem and current reasoning goal • Logical : logical coherence and inferential validity • Fix : whether the step explicitly detects, revises, rejects, or corrects a previous incorrect intermediate claim |
| Output format | Step 1: Semantic=0.xx, Logical=0.xx, Fix=0.xx |
| Step 2: Semantic=0.xx, Logical=0.xx, Fix=0.xx |
| Field | Content |
| Purpose | Re-issue a format-constrained request when the original oracle output cannot be parsed. |
| Instruction | Return the scores again using exactly the following format. |
| Required format | Step 1: Semantic=0.xx, Logical=0.xx, Fix=0.xx Step 2: Semantic=0.xx, Logical=0.xx, Fix=0.xx ... |
| Constraint | Output only the step-wise scores. Do not include any other text. |
| Signal | Meaning | Parsing rule |
| Semantic | Semantic consistency with the problem and the current reasoning goal | Numeric score in |
| Logical | Whether the current step is internally coherent and supports subsequent reasoning | Numeric score in |
| Fix | Whether the step explicitly detects, revises, rejects, or corrects a previous incorrect intermediate claim | Numeric score in |
| Metric | Value | Description |
| First-pass success | 98.2% | Step-aligned numeric fields were extracted successfully on the first attempt. |
| Re-prompt recovery | 1.7% | Formatting errors were corrected by a single constrained re-prompt. |
| Final failure | 0.1% | The oracle still failed after retry; these cases were assigned conservative default scores. |
| Algorithm 2: CoT 2 -Meta inference loop |
| Input: problem , budget , backbone |
| Initialize: root node , frontier |
| while budget remains and do |
| Select frontier node with highest controller score |
| Generate candidate child thoughts under Direct / Decompose / Verify modes |
| For each child trajectory : |