REVIEW 2 major objections 2 minor 27 references
Small language models can reach larger-model reasoning performance by learning to select among their own top token candidates.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · grok-4.3
2026-07-01 08:27 UTC pith:AI3Q6DGU
load-bearing objection The 95% hit rate for LLM tokens in SLM top-8 is the real hook, but the post-fine-tuning stability of that property is unverified and the abstract gives almost no experimental detail. the 2 major comments →
Select to Think: Unlocking SLM Potential with Local Sufficiency
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
Local sufficiency means that at divergence points the LLM's preferred token lies inside the SLM's top-K candidates with high probability; S2T therefore trains the SLM to perform the selection step itself, removing any runtime dependence on the larger model while preserving most of the accuracy gain.
What carries the argument
Local sufficiency, the empirical property that the LLM's token resides in the SLM's top-K predictions at divergence points, which is used to create a simplified supervision signal of discrete candidate rankings for distillation into S2T-Local.
Load-bearing premise
The observed pattern that the large model's token appears reliably in the small model's top-K list continues to hold during both the distillation training and later autonomous inference without needing per-task adjustments.
What would settle it
Measure the fraction of divergence points where the 32B model's token falls outside the 1.5B model's top-8 predictions on a new set of reasoning problems; if that fraction exceeds roughly 10 percent, the performance claims should collapse.
If this is right
- A single inference trajectory from the distilled 1.5B model reaches the accuracy previously obtained only by running eight independent trajectories and voting.
- Deployment cost drops because the large model is needed only during the one-time distillation phase, not at every user query.
- The same selection signal can be applied to any task where token-level divergence between model sizes can be recorded.
- No additional prompt engineering or external verifier is required once the selection logic has been internalized.
Where Pith is reading between the lines
- The approach could be tested on non-math domains such as code generation or multi-hop question answering to check whether local sufficiency generalizes.
- If the hit rate remains high across model-size gaps larger than 1.5B-to-32B, the method might scale to even tinier student models.
- One could measure whether the distilled selection head also improves calibration or reduces hallucination rates on factual recall tasks.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper claims that 'local sufficiency' holds at reasoning divergence points, with a 32B LLM's preferred token residing in a 1.5B SLM's top-8 predictions at 95% hit rate. It proposes Select to Think (S2T) to reduce the LLM's role to discrete selection among SLM candidates, then introduces S2T-Local to distill this selection logic into the SLM via supervised fine-tuning, enabling autonomous re-ranking. The central empirical claim is that S2T-Local yields a 24.1% relative gain on Math Avg. over greedy decoding for the 1.5B model, matching the performance of 8-path self-consistency at single-trajectory cost.
Significance. If the performance claims prove reproducible under standard controls, the approach of distilling discrete selection rather than full generation could offer a practical route to closing the reasoning gap between SLMs and LLMs without persistent LLM access at inference, with potential impact on efficient deployment.
major comments (2)
- [Abstract] Abstract: the 95% hit rate is measured on the base 1.5B SLM before fine-tuning. S2T-Local performs supervised fine-tuning on discrete rankings, which necessarily alters the SLM's parameters and next-token distribution; no post-training measurement is reported to confirm that the LLM-preferred token remains inside the new top-8 at divergence points, which is required for the autonomous re-ranking claim to hold.
- [Empirical results] Empirical results (throughout): the abstract asserts concrete accuracy gains and a 24.1% relative improvement, yet the manuscript supplies no experimental protocol, dataset names or splits, baseline implementations (including the 8-path self-consistency comparator), number of evaluation runs, or statistical significance tests. These omissions render the central performance claims unevaluable.
minor comments (2)
- [Method] Provide a formal definition or algorithmic description of 'divergence points' and the exact procedure used to identify them.
- [Abstract] Clarify whether the reported hit rate uses the same K=8 and the same divergence detection logic that will be available to the fine-tuned SLM at inference.
Simulated Author's Rebuttal
Thank you for the referee's constructive feedback. We address each major comment below and commit to revisions that strengthen the manuscript's clarity and reproducibility.
read point-by-point responses
-
Referee: [Abstract] Abstract: the 95% hit rate is measured on the base 1.5B SLM before fine-tuning. S2T-Local performs supervised fine-tuning on discrete rankings, which necessarily alters the SLM's parameters and next-token distribution; no post-training measurement is reported to confirm that the LLM-preferred token remains inside the new top-8 at divergence points, which is required for the autonomous re-ranking claim to hold.
Authors: We agree that the 95% hit rate is reported exclusively for the base 1.5B model. S2T-Local applies SFT on discrete rankings and therefore modifies the next-token distribution; no post-SFT hit-rate verification at divergence points is currently provided. We will add this measurement in the revision, recomputing the hit rate on the fine-tuned model using the same divergence-point identification procedure, to directly support the autonomous re-ranking claim. revision: yes
-
Referee: [Empirical results] Empirical results (throughout): the abstract asserts concrete accuracy gains and a 24.1% relative improvement, yet the manuscript supplies no experimental protocol, dataset names or splits, baseline implementations (including the 8-path self-consistency comparator), number of evaluation runs, or statistical significance tests. These omissions render the central performance claims unevaluable.
Authors: We accept that the current manuscript lacks sufficient experimental detail for full reproducibility. We will expand the experimental section to include: explicit dataset names and splits, complete protocol for all baselines (including the precise 8-path self-consistency implementation), number of evaluation runs, and statistical significance tests. These additions will make the reported gains (including the 24.1% relative improvement) directly verifiable. revision: yes
Circularity Check
No circularity: empirical observation drives method, no definitional reduction
full rationale
The paper's core claims rest on an empirical observation (95% hit rate of LLM token in base SLM top-8 at divergence points) followed by a distillation procedure whose efficacy is measured by separate post-training benchmarks (24.1% relative gain). No equations, fitted parameters, or self-citations are invoked to make the reported gains true by construction. The local-sufficiency property is measured on the unmodified model and then used to motivate training; the training outcome is evaluated independently. This matches the default expectation of a non-circular empirical paper.
Axiom & Free-Parameter Ledger
axioms (1)
- domain assumption LLM preferred token resides in SLM top-K predictions at reasoning divergence points
read the original abstract
Small language models (SLMs) offer efficient deployment, yet they often lag behind their larger counterparts (LLMs) in reasoning. Existing remedies either invoke an LLM at points of reasoning divergence, incurring substantial latency and cost, or rely on standard distillation, which is limited by the SLM's capacity to accurately mimic the LLM's complex generative distribution. We address this dilemma by identifying local sufficiency: at divergence points, the LLM's preferred token often resides within the SLM's top-K next-token predictions, even when failing to emerge as the SLM top-1 choice. We therefore propose Select to Think (S2T), which reframes the LLM's role from open-ended generation to selection among the SLM's proposals, simplifying the supervision signal to discrete candidate rankings. Leveraging this, we introduce S2T-Local, which distills the selection logic into the SLM, empowering it to perform autonomous re-ranking without inference-time LLM dependency. Empirically, a 1.5B SLM's top-8 candidates contain the 32B LLM's choice with a 95% hit rate, and S2T-Local improves the 1.5B SLM's Math Avg. over greedy decoding by 24.1% relative gain, matching the efficacy of 8-path self-consistency with single-trajectory efficiency.
Figures
Reference graph
Works this paper leans on
-
[1]
Achiam, J., Adler, S., Agarwal, S., Ahmad, L., Akkaya, I., Aleman, F. L., Almeida, D., Altenschmidt, J., Altman, S., Anadkat, S., et al. Gpt-4 technical report.arXiv preprint arXiv:2303.08774,
work page internal anchor Pith review Pith/arXiv arXiv
-
[2]
Accelerating Large Language Model Decoding with Speculative Sampling
URL https://openreview. net/forum?id=zph7e5JaXc. Chen, C., Borgeaud, S., Irving, G., Lespiau, J.-B., Sifre, L., and Jumper, J. Accelerating large language model decoding with speculative sampling.arXiv preprint arXiv:2302.01318, 2023a. Chen, L., Zaharia, M., and Zou, J. Frugalgpt: How to use large language models while reducing cost and improving performa...
work page internal anchor Pith review Pith/arXiv arXiv
-
[3]
Unveiling the key factors for distilling chain-of-thought reasoning
Chen, X., Sun, Z., Wenjin, G., Zhang, M., Chen, Y ., Sun, Y ., Su, H., Pan, Y ., Klakow, D., Li, W., and Shen, X. Unveiling the key factors for distilling chain-of-thought reasoning. InFindings of the Association for Computa- tional Linguistics: ACL 2025, pp. 15094–15119,
work page 2025
-
[4]
DoLa: Decoding by Contrasting Layers Improves Factuality in Large Language Models
Chuang, Y .-S., Xie, Y ., Luo, H., Kim, Y ., Glass, J., and He, P. Dola: Decoding by contrasting layers improves factuality in large language models.arXiv preprint arXiv:2309.03883,
work page internal anchor Pith review Pith/arXiv arXiv
-
[5]
Training Verifiers to Solve Math Word Problems
Cobbe, K., Kosaraju, V ., Bavarian, M., Chen, M., Jun, H., Kaiser, L., Plappert, M., Tworek, J., Hilton, J., Nakano, R., Hesse, C., and Schulman, J. Training verifiers to solve math word problems.arXiv preprint arXiv:2110.14168,
work page internal anchor Pith review Pith/arXiv arXiv
-
[6]
Fu, T., Ge, Y ., You, Y ., Liu, E., Yuan, Z., Dai, G., Yan, S., Yang, H., and Wang, Y . R2r: Efficiently navigating divergent reasoning paths with small-large model token routing.arXiv preprint arXiv:2505.21600, 2025a. Fu, T., You, Y ., Chen, Z., Dai, G., Yang, H., and Wang, Y . Think-at-hard: Selective latent iterations to improve reasoning language mode...
-
[7]
Relayllm: Efficient reasoning via collaborative decod- ing.arXiv preprint arXiv:2601.05167,
Huang, C., Zheng, T., Huang, L., Li, J., Liu, H., and Huang, J. Relayllm: Efficient reasoning via collaborative decod- ing.arXiv preprint arXiv:2601.05167,
-
[8]
Scalable Best-of-N Selection for Large Language Models via Self-Certainty
Kang, Z., Zhao, X., and Song, D. Scalable best-of-n selec- tion for large language models via self-certainty.arXiv preprint arXiv:2502.18581,
-
[9]
arXiv preprint arXiv:2504.09923 , year=
Kim, Y ., Yi, E., Kim, M., Yun, S.-Y ., and Kim, T. Guiding reasoning in small language models with llm assistance. arXiv preprint arXiv:2504.09923,
-
[10]
Small models struggle to learn from strong reasoners
9 Select to Think: Unlocking SLM Potential with Local Sufficiency Li, Y ., Yue, X., Xu, Z., Jiang, F., Niu, L., Lin, B. Y ., Ra- masubramanian, B., and Poovendran, R. Small models struggle to learn from strong reasoners.arXiv preprint arXiv:2502.12143,
-
[11]
Critical Tokens Matter: Token-Level Contrastive Estimation Enhances LLM's Reasoning Capability
Lin, Z., Liang, T., Xu, J., Lin, Q., Wang, X., Luo, R., Shi, C., Li, S., Yang, Y ., and Tu, Z. Critical tokens matter: Token-level contrastive estimation enhances llm’s reason- ing capability.arXiv preprint arXiv:2411.19943,
-
[12]
Liu, A., Feng, B., Xue, B., Wang, B., Wu, B., Lu, C., Zhao, C., Deng, C., Zhang, C., Ruan, C., et al. Deepseek-v3 technical report.arXiv preprint arXiv:2412.19437,
work page internal anchor Pith review Pith/arXiv arXiv
-
[13]
Manvi, R., Hong, J., Seyde, T., Labonne, M., Lechner, M., and Levine, S. Zip-rc: Zero-overhead inference-time pre- diction of reward and cost for adaptive and interpretable generation.arXiv preprint arXiv:2512.01457,
- [14]
-
[15]
Pan, R., Dai, Y ., Zhang, Z., Oliaro, G., Jia, Z., and Ne- travali, R. Specreason: Fast and accurate inference- time compute via speculative reasoning.arXiv preprint arXiv:2504.07891,
-
[16]
Qwen. Qwen2.5 technical report.arXiv preprint arXiv:2412.15115,
work page internal anchor Pith review Pith/arXiv arXiv
-
[17]
Schick, T. and Sch ¨utze, H. It’s not just size that matters: Small language models are also few-shot learners. InPro- ceedings of the 2021 conference of the North American chapter of the association for computational linguistics: Human language technologies, pp. 2339–2352,
work page 2021
-
[18]
Trusting your evidence: Hallucinate less with context-aware decoding
Shi, W., Han, X., Lewis, M., Tsvetkov, Y ., Zettlemoyer, L., and Yih, W.-t. Trusting your evidence: Hallucinate less with context-aware decoding. InProceedings of the 2024 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 2: Short Papers), pp. 783–791,
work page 2024
-
[19]
Large Language Model Routing with Benchmark Datasets
Shnitzer, T., Ou, A., Silva, M., Soule, K., Sun, Y ., Solomon, J., Thompson, N., and Yurochkin, M. Large language model routing with benchmark datasets.arXiv preprint arXiv:2309.15789,
-
[20]
Multiplex thinking: Reasoning via token-wise branch-and-merge.arXiv preprint arXiv:2601.08808, 2026
Tang, Y ., Dong, L., Hao, Y ., Dong, Q., Wei, F., and Gu, J. Multiplex thinking: Reasoning via token-wise branch- and-merge.arXiv preprint arXiv:2601.08808,
-
[21]
Gemma 2: Improving Open Language Models at a Practical Size
Team, G., Riviere, M., Pathak, S., Sessa, P. G., Hardin, C., Bhupatiraju, S., Hussenot, L., Mesnard, T., Shahri- ari, B., Ram ´e, A., et al. Gemma 2: Improving open language models at a practical size.arXiv preprint arXiv:2408.00118,
work page internal anchor Pith review Pith/arXiv arXiv
-
[22]
Varshney, N. and Baral, C. Model cascading: Towards jointly improving efficiency and accuracy of nlp systems. arXiv preprint arXiv:2210.05528,
-
[23]
Yang, A., Zhang, B., Hui, B., Gao, B., Yu, B., Li, C., Liu, D., Tu, J., Zhou, J., Lin, J., et al. Qwen2. 5-math techni- cal report: Toward mathematical expert model via self- improvement.arXiv preprint arXiv:2409.12122,
work page internal anchor Pith review Pith/arXiv arXiv
-
[24]
Zhang, Y . and Math-AI, T. American invitational mathemat- ics examination (aime) 2025,
work page 2025
-
[25]
30:end while 31:Returnsequencey. B. Evaluation Details B.1. Benchmark Mathematical reasoning benchmarks. • GSM8K (Cobbe et al., 2021): A dataset of 8,500 linguistically diverse grade school math word problems requiring 2-8 step reasoning, serving as a standard benchmark for evaluating basic mathematical reasoning capabilities. 12 Select to Think: Unlockin...
work page 2021
-
[26]
amateur” (or counterfactual) signal from an “expert
reshape logits by subtracting an “amateur” (or counterfactual) signal from an “expert” one to suppress generic or hallucinated tokens. This paradigm has inspired various extensions, including context-aware variants to resolve evidence-parametric conflicts (Shi et al., 2024), and internal-contrast methods (e.g.,DoLa(Chuang et al., 2023)) that utilize intra...
work page 2024
-
[27]
During inference, we apply a sigmoid activation to the criticality logit and threshold at 0.7 (configurable) to make binary triggering decisions, eliminating the need to compute expensive cross-model KL divergence at every step. 15 Select to Think: Unlocking SLM Potential with Local Sufficiency Table A1.Accuracy comparison under varying trigger threshold ...
work page 2024
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.