Not All Queries Need Deep Thought: CoFiCot for Adaptive Coarse-to-fine Stateful Refinement
Organizations: Xi’an Jiaotong University · Zhejiang University · The Chinese University of Hong Kong (Shenzhen) · Institute of Automation, CASIA
Abstract
Scaling test-time computation enhances LLM reasoning ability but faces a uniform computation paradox. Allocating identical resources leads to over-correction on simple tasks and insufficient refinement on complex ones. To address this, we propose CoFiCot, a coarse-to-fine adaptive framework that dynamically tailors inference strategies to problem difficulty. Specifically, we implement a multi-metric classifier that triages queries by synthesizing semantic entropy, consensus reliability, and predicted reasoning depth . This enables a differentiated refinement stage that applies efficient aggregation for simple queries while routing complex ones to a context-aware correction loop . We formalize correction as a stateful sequential propagation process , where each repair is strictly conditioned on the verified history of prior rectifications. By integrating Process Reward Models (PRMs) within this state-dependent trajectory, CoFiCot effectively bridges the gap between granular error localization and global logical coherence, preventing the context fragmentation typical of stateless refinement methods.
Figures & tables
| Method | MMLU | MATH | SVAMP | GSM8K | SAT | Avg. |
|---|---|---|---|---|---|---|
| Llama3-8B-Instruct | ||||||
| Zero-shot CoT [ Wei et al., 2022 ] | 50.4 | 24.2 | 72.4 | 80.1 | 58.2 | 57.1 |
| Self-Refine [ Madaan et al., 2023 ] | 49.6 | 24.6 | 72.0 | 79.0 | 57.7 | 56.6 |
| Best-of- ( = 120) [ Lightman et al., 2023 ] | 62.6 | 41.4 | 88.7 | 90.1 | 72.4 | 71.0 |
| -way SC ( = 120) [ Wang et al., 2022 ] | 63.0 | 40.6 | 89.8 | 90.3 | 70.5 | 70.8 |
| Self-Refine + -way SC [ Chen et al., 2024a ] | 62.1 | 40.4 | 88.6 | 90.1 | 68.2 | 69.9 |
| Method | ARC | Date |
|---|---|---|
| Zero-shot [ Wei et al., 2022 ] | 66.5 | 52.5 |
| 40-way SC [ Wang et al., 2022 ] | 85.5 | 72.5 |
| 120-way SC [ Chen et al., 2024a ] | 86.0 | 72.5 |
| CoFiCot(ours) | 88.2 | 80.8 |
| Method | MMLU | MATH | SVAMP | GSM8K | SAT | Avg. |
|---|---|---|---|---|---|---|
| Qwen2.5-Math-7B [ Yang et al., 2024 ] | 73.9 | 78.8 | 91.8 | 94.9 | 92.3 | 86.3 |
| -way SC ( = 40) [ Wang et al., 2022 ] | 81.3 | 87.0 | 95.5 | 97.2 | 97.3 | 91.7 |
| -way SC ( = 120) [ Chen et al., 2024a ] | 82.0 | 86.8 | 95.4 | 97.3 | 97.3 | 91.8 |
| Best-of- ( = 120) [ Lightman et al., 2023 ] | 82.6 | 86.0 | 93.4 | 96.9 | 95.2 | 90.8 |
| CoFiCot(ours) | 84.9 | 91.7 | 95.8 | 97.2 | 97.6 | 93.4 |
| PRM | ORM | Acc. |
|---|---|---|
| Math-Shepherd-7B | InternLM-7B | 48.3 |
| Qwen-Math 7B | InternLM-7B | 54.9 |
| Math-Shepherd-7B | Llama-3.1-8B | 51.2 |
| Methods | GSM8K | MATH |
|---|---|---|
| Accuracy | Accuracy | |
| CoFiCot | 91.8 | 47.9 |
| w/o Coarse Stage | 88.5 (3.3↓) | 45.1 (2.8↓) |
| w/o Fine Stage | 89.1 (2.7↓) | 41.2 (6.7↓) |
| Module | Datasets | |||||
|---|---|---|---|---|---|---|
| Cofi. | Relia. | Comp. | GSM8K | MATH | MMLU | SAT |
| - | - | 86.8 | 43.4 | 63.9 | 71.2 | |
| - | - | 87.2 | 44.6 | 64.1 | 70.9 | |
| - | 89.7 | 45.9 | 67.0 | 73.4 | ||
| - | 89.3 | 46.2 | 67.4 | 74.1 | ||
| 91.8 | 47.9 | 68.8 | 75.0 | |||
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Criterion | GSM8K | MATH |
|---|---|---|
| Pessimistic | 91.4 | 47.2 |
| Optimistic | 90.8 | 46.5 |
| Democratic | 90.3 | 45.8 |
| CoFiCot(ours) | 91.8 | 47.9 |
| Modules | MMLU | MATH |
|---|---|---|
| ORM-Only | 67.8 | 47.1 |
| PRM-Only | 67.3 | 46.5 |
| Both | 68.8 | 47.9 |
LLM Self-Refinement
Large language model self-refinement studies how models can improve outputs through repeated critique, verification, revision, and feedback.
See all 114 papers