REVIEW 3 major objections 5 minor 18 references
Protoreasoning in Tiny Transformers
T0 review · 3 major / 5 minor · reviewed 2026-08-06 · deepseek-v4-flash
Pith's one-line read This paper claims that tiny transformers can learn a primitive form of chain-of-thought reasoning—deterministic pruning traces over Dyck bracket trees—and that this training signal substantially narrows their out-of-distribution…
desk verdict A cheap, well-designed reasoning testbed with a strong but overconfident headline; the Low pattern and missing paired statistics are the real soft spots. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is the protoreasoning trace: a deterministic scratchpad of pruning steps over the tree or forest encoded by a Dyck-$k$ sentence, where each step is the previous state after deleting nodes that cannot belong to the answer, so the trace ends exactly when only the answer remains. Step dropout randomly skips 15% of reasoning steps during training, and a positional-index augmentation called SkipAlign remaps position indices so that the longer trace sequences do not confound the comparison with vanilla training. The trace carries the argument because it converts each task from a one-shot input-output mapping into a sequence of intermediate states, and the dot-filler ablation isolates the content of those states as the source of the out-of-distribution gain.
What would settle it
Take any one of the six hold-out patterns and recompute the trace-versus-vanilla comparison with per-seed validity reported separately and with a fresh set of 128 validation prompts drawn for each seed; if the typical seed-level difference is near zero, or if the dot-filler replacement matches full-trace validity once token counts are equalized, the central claim fails.
Extended reading notes
Core claim
The central discovery is that tiny transformers trained on an input followed by a deterministic pruning trace generalize markedly better out of distribution than models trained on input-output pairs alone. In the deepest-path task the trace removes the shallowest remaining leaf at each step until only the path to the deepest leaf remains; in the maximum-order preleaf task it peels trees from the root and eliminates the smallest two-level sibling brood until one tree remains. Evaluated on held-out strata of depth or preleaf size, the trace format almost always beats the vanilla format across two tasks and six hold-out patterns, with the largest gains in interpolation; extrapolation remains hard for both formats. Replacing trace content with filler dots destroys the advantage, supporting the interpretation that trace content, not token count, drives the gain.
Load-bearing premise
The conclusion that trace content helps assumes that averaging output validity over 10 random seeds on fixed 128-prompt validation sets is a stable estimator, despite many seed-level standard deviations above 10-20 points in the reported tables; if the aggregates are dominated by a few lucky seeds or particular validation prompt draws, the trace advantage is not reliable.
Editorial extensions
If this is right
- Step-by-step reasoning is not exclusive to frontier models: ~1M-parameter transformers already generalize better out of distribution when trained on deterministic reasoning traces than on input-output pairs alone.
- The trace advantage appears across two tasks and six hold-out patterns and is largest for interpolation; extrapolation to unseen extremes remains hard for both trace and vanilla formats.
- Extra tokens alone do not explain the gain: replacing trace content with filler dots dramatically deteriorates validity, and step dropout plus SkipAlign improve generalization by removing length and scaffold confounders.
- Under mixed-task training, trace tolerates the harder setting far better than vanilla, and the absence of cross-task transfer indicates the generalization limit is about how each task's solution is wired rather than about input representability.
- The tiny-model setup makes dense experimentation feasible, so reasoning can be studied with exact verification, controlled difficulty, and thousands of training runs on modest compute.
Reading between the lines
- Beyond the paper: if protoreasoning traces hold up, they offer a cheap testbed for mechanistic interpretability of reasoning, since the trace provides exact intermediate states that internal representations can be probed against.
- Beyond the paper: the mixed-task result suggests compositionality will not emerge from trace training alone; a natural next test is whether traces that share sub-steps across tasks produce modular transfer.
- Beyond the paper: the dot-filler ablation implies that 'thinking tokens' without task-relevant content should not be expected to improve out-of-distribution generalization in small models, a prediction that could be tested in larger models by perturbing chain-of-thought content.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper introduces "protoreasoning traces" for tiny (~1M-parameter) transformers: deterministic pruning scratchpads that solve two tasks defined over Dyck languages (deepest path and maximum preleaf) by repeatedly removing tree nodes that cannot be part of the answer. Across six hold-out patterns and a mixed-task setting, the paper compares vanilla input-output training with trace training, reporting that traces "almost always" improve out-of-distribution output validity (Tables 2-3). A dot-erasure ablation (Table 4) is used to argue that the improvement comes from trace content rather than extra tokens. The authors also examine the roles of SkipAlign positional augmentation and step dropout, and report that extrapolation remains difficult for both formats.
Significance. If the reported effects are robust, the paper provides a valuable minimal testbed for studying step-by-step reasoning at model scales far below natural-language competence. Strengths of the design include the use of two exactly verifiable tasks with controllable difficulty axes, ten seeds, disjoint train/validation/test strata, deterministic trace construction independent of the results, and a dense experimental campaign (over 1800 final training runs). The finding that trace content, not just token count, drives generalization would be an interesting contribution to the literature on scratchpads and Chain-of-Thought. However, the statistical support for the headline claims is currently incomplete, and the key content ablation is missing in the main experimental setting, so the significance of the claims is not yet fully established.
major comments (3)
- [Tables 2-3; Sec. 2.3, A.4] The headline claim that traces "almost always" beat vanilla is not supported by any inference that accounts for seed variance. For example, in Table 2 the DP Low row shows trace at 21.6±7.4 versus vanilla at 39.6±2.0, and PL Low shows 41.0±5.4 versus 43.9±3.0; in Table 3, DP High Mixed shows trace at 43.0±4.9 versus vanilla at 46.6±5.2. Several of the positive comparisons, such as DP Even (98.8±0.9 vs 98.5±2.5), are smaller than the seed standard deviations. The paper reports no paired seed-level differences, no confidence intervals for the differences, and no multiple-comparison-aware testing across the 24 task×pattern comparisons. Since the abstract and conclusion generalize from these aggregates, the "almost always" claim needs paired effect-size reporting (e.g., mean paired difference with confidence interval over seeds, or per-seed scatter plots) and ideally variance estimates over prompt draws.
- [Sec. 3.4, Table 4] The dot-erasure ablation, which is the sole evidence for the claim that trace content rather than extra tokens drives the gain, is reported only under RoPE and not under SkipAlign, the positional augmentation used in all main comparisons. This matters because SkipAlign has large effects on trace validity; for instance, Table 4 shows DP Random at 59.8±24.4 under RoPE versus 85.2±8.5 under SkipAlign for the trace condition. Without the erased condition under SkipAlign, the content-not-tokens conclusion is not deconfounded from the positional-encoding regime. Please add the SkipAlign erased condition or explicitly justify why it is unnecessary.
- [Sec. A.4, Sec. 2.3] The fixed 128-prompt validation set is used for both hyperparameter selection and best-checkpoint selection, but the paper does not state how many test prompts per stratum are used for the reported OOD numbers or whether test sentences are freshly drawn per seed. If the test evaluation also uses small fixed prompt sets, the reported means and standard deviations may reflect prompt-specific draws rather than model behavior. Please report the test sampling procedure and, if the test set is fixed and small, re-evaluate with fresh prompts or supply bootstrap intervals over prompt draws.
minor comments (5)
- [Title/Author block and Abstract] Several words are run together or split in the author block and abstract (e.g., "Eduardo V alle", "F ergal Reid", "wecallprotoreasoning"); these should be cleaned up in the final version.
- [Sec. A.4] The contribution statement says "over 1700 training runs" while Appendix A.4 reports 1860 final runs plus 1488 hyperparameter-search runs (3348 total); please clarify which number is meant.
- [Table 4] The table header formatting is confusing: the grouping of columns under "Trace", "Erased", and "Vanilla" is not clear from the text. Please reformat so each condition and augmentation is unambiguous.
- [Sec. 2.2] Please specify whether the dot-by-dot replacement preserves the exact token count and positional indices of the trace, including delimiters and newlines, since the "extra tokens" confounder is central to the ablation.
- [Fig. 7] The top x-axis in Fig. 7 is labeled "number of unique bracket nestings (log)" without specifying the base; the text uses log10, so the figure should match.
Circularity Check
No significant circularity: the paper's central claims are empirical comparisons over held-out strata with externally generated pruning traces, not derivations from fitted inputs.
full rationale
This paper makes no formal derivation; its central claims are empirical comparisons of output validity across formats, tasks, hold-out patterns, and ablations. The protoreasoning traces are generated by an explicit, deterministic pruning procedure (Section 2.2) defined independently of model performance: for DP the algorithm repeatedly eliminates the shallowest leaf until the tree degenerates into a path, and for PL it peels trees by the root and eliminates the smallest two-level tree, with ties broken leftmost. None of the trace-construction constants, the step-dropout probability, the SkipAlign augmentation, or the hold-out splits is fitted to the reported test validity numbers. Hyperparameters are selected using validation strata that are explicitly disjoint from both training and test strata (Section A.4), and the test results are reported on unseen test strata; this is a selection procedure, not circular reasoning. The abstract's claim that the trace's content rather than its extra tokens drives the gain is supported by the dot-erasure ablation in Table 4, which is an empirical control and not a re-labelled fit. The weaknesses flagged in the reader's take, such as large seed standard deviations, fixed 128-prompt validation sets, and the absence of paired significance tests, are legitimate statistical-evidence concerns about robustness, but they do not make any reported quantity equal to its input by construction. There are no load-bearing self-citations, no imported uniqueness theorems, and no ansatz smuggled in via citation. The paper is therefore self-contained as an empirical study, and no circular step can be exhibited.
Assumptions & free parameters
free parameters (6)
- learning rate and weight decay =
selected per treatment by 8-trial Sobol search (LR 1e-4 to 1e-2, WD 1e-7 to 1e-1)
- step dropout probability =
0.15
- SkipAlign Dirichlet alpha =
0.4
- model size =
width 128, depth 4, 16 heads (~1M params)
- training schedule =
batch 256, 2K pretrain + 4K finetune steps, 600 warmup/decay
- tie-breaking rules =
rightmost for answers, leftmost for pruning
assumptions (6)
- standard math Dyck sentences correspond bijectively to forests, with inner brackets as descendants
- domain assumption Every input has exactly one correct output and one deterministic gold trace
- domain assumption Training with traces transfers to no-trace prompts at test time
- domain assumption Hyperparameters selected on OOD validation strata generalize to disjoint OOD test strata
- domain assumption SkipAlign position-index remapping preserves task semantics
- ad hoc to paper Step dropout yields an order-preserving subsequence from which the model can recover the answer
Cite this review
Pith. "Pith review of Protoreasoning in Tiny Transformers." pith.science (2026). https://pith.science/paper/LCAPWQQD
@misc{pith2026260804980,
author = {Pith},
title = {Pith review of: Protoreasoning in Tiny Transformers},
year = {2026},
howpublished = {\url{https://pith.science/paper/LCAPWQQD}},
note = {Machine review of arXiv:2608.04980}
}
read the original abstract
We show that tiny transformers can profitably employ a simple form of Chain of Thought, which we call protoreasoning, allowing us to study step-by-step reasoning on ~1M-parameter models and opening up opportunities for much more detailed experimentation and analysis than is feasible for larger models. Current Large Language Models exhibit impressive step-by-step reasoning, but we have yet to understand its generality, i.e., when and how LLMs learn genuinely general algorithms rather than "bags of heuristics." Such questions are hard to settle on compute-intensive frontier models trained on opaque data. To work at model scales far below the threshold for natural-language competence, we define reasoning-friendly tasks on Dyck languages (sentences of correctly nested brackets). We find that protoreasoning traces substantially close the out-of-distribution generalization gap, and ablations confirm that the trace's content, not merely its extra tokens, drives the gain.
Figures
Figures from the paper (6 more)
Reference graph
Works this paper leans on
-
[5]
The illusion of the illusion of thinking: A comment on Shojaee et al
Alex Lawsen. The illusion of the illusion of thinking: A comment on Shojaee et al. (2025). 2506.09250, 6
arXiv 2025
-
[9]
An expressivity analysis of hierarchical modelling in deep transformers via bounded-depth grammars
URLhttps://openreview.net/forum?id=9XFSbDPmdW. VinothNandakumar, QiangQu, PramodThebe, SakshiKhachariyapatel, andTongliangLiu. Anexpressivity analysis of hierarchical modelling in deep transformers via bounded-depth grammars. arXiv:2606.17522,
-
[10]
Jacob Pfau, William Merrill, and Samuel R
URLhttps://proceedings.iclr.cc/paper_files/paper/2025/hash/ 8c5f30296296d2ae402ebbd09aaa9c12-Abstract-Conference.html. Jacob Pfau, William Merrill, and Samuel R. Bowman. Let’s think dot by dot: Hidden computation in transformer language models. InProceedings of the 2024 Conference on Language Modeling,
work page 2025
-
[11]
URLhttps://dl.acm.org/doi/10.1145/3774896
doi: 10.1145/3774896. URLhttps://dl.acm.org/doi/10.1145/3774896. Rylan Schaeffer, Brando Miranda, and Sanmi Koyejo. Are emergent abilities of large lan- guage models a mirage? InAdvances in Neural Information Processing Systems, volume 36, pp. 55565–55581,
-
[12]
URLhttps://proceedings.neurips.cc/paper_files/paper/2023/hash/ adc98a266f45005c403b8311ca7e8bd7-Abstract-Conference.html. Timo Schick, Jane Dwivedi-Yu, Roberto Dessì, Roberta Raileanu, Maria Lomeli, Luke Zettlemoyer, Nicola Cancedda, and Thomas Scialom. Toolformer: Language models can teach themselves to use tools. In Advances in Neural Information Proces...
work page 2023
-
[13]
Parshin Shojaee, Iman Mirzadeh, Keivan Alizadeh, Maxwell Horton, Samy Bengio, and Mehrdad Farajtabar
URL https://openreview.net/forum?id=sPIXjSkDFG. Parshin Shojaee, Iman Mirzadeh, Keivan Alizadeh, Maxwell Horton, Samy Bengio, and Mehrdad Farajtabar. The illusion of thinking: Understanding the strengths and limitations of reasoning models via the lens of problem complexity. arXiv:2506.06941, 6
-
[14]
ISSN 2307-387X. doi: 10.1162/tacl_a_00663. URLhttps://doi.org/10.1162/tacl_a_00663. Jiankai Sun, Chuanyang Zheng, Enze Xie, Zhengying Liu, Ruihang Chu, Jianing Qiu, Jiaqi Xu, Mingyu Ding, Hongyang Li, Mengzhe Geng, Yue Wu, Wenhai Wang, Junsong Chen, Zhangyue Yin, Xiaozhe Ren, Jie Fu, Junxian He, Wu Yuan, Qi Liu, Xihui Liu, Yu Li, Hao Dong, Yu Cheng, Ming ...
-
[15]
URLhttps: //doi.org/10.1145/3729218
doi: 10.1145/3729218. URLhttps: //doi.org/10.1145/3729218. Miles Turpin, Julian Michael, Ethan Perez, and Samuel R. Bowman. Language models don’t always say what they think: Unfaithful explanations in chain-of-thought prompting. InAdvances in Neural Information Processing Systems (NeurIPS),
Show all 18 references
-
[16]
Iñaki Dellibarda Varela, Pablo Romero-Sorozabal, Eduardo Rocon, and Manuel Cebrian
URLhttps://proceedings.neurips.cc/paper_files/ paper/2024/file/2f6a6317bada76b26a4f61bb70a7db59-Paper-Conference.pdf. Iñaki Dellibarda Varela, Pablo Romero-Sorozabal, Eduardo Rocon, and Manuel Cebrian. Rethinking the illusion of thinking. arXiv:2507.01231, 7
2024 arXiv
-
[18]
ShunyuYao, DianYu, JeffreyZhao, IzhakShafran, ThomasL.Griffiths, YuanCao, andKarthikNarasimhan
URLhttps: //arxiv.org/abs/2405.03939. ShunyuYao, DianYu, JeffreyZhao, IzhakShafran, ThomasL.Griffiths, YuanCao, andKarthikNarasimhan. Tree of thoughts: Deliberate problem solving with large language models. InAdvances in Neural Infor- mation Processing Systems (NeurIPS), 2023a...
-
[19]
URL https://doi.org/10.1109/TPAMI.2025.3637037
doi: 10.1109/TPAMI.2025.3637037. URL https://doi.org/10.1109/TPAMI.2025.3637037. Dawei Zhu, Nan Yang, Liang Wang, Yifan Song, Wenhao Wu, Furu Wei, and Sujian Li. PoSE: Efficient con- text window extension of LLMs via positional skip-wise training. InThe Twelfth International C...
2025
-
[20]
A Appendix A.1 Holdout patterns Fig
URLhttps://openreview.net/forum?id=3Z1gxuAQrA. A Appendix A.1 Holdout patterns Fig. 7 shows precisely how we split each task’s strata into training, validation, and test sets for single-task training (Sec. 3.2). The top x-axis shows thelog10 count of unique bracket nestings in...
2024
-
[2020]
doi: 10.1162/tacl_a_ 00306
ISSN 2307-387X. doi: 10.1162/tacl_a_ 00306. URLhttps://doi.org/10.1162/tacl_a_00306. Jie Huang and Kevin Chen-Chuan Chang. Towards reasoning in large language models: A survey. In Anna Rogers, Jordan Boyd-Graber, and Naoaki Okazaki (eds.),Findings of the Association for Comput...
-
[2021]
doi: 10.18653/v1/2021.naacl-main.405
Association for Com- putational Linguistics. doi: 10.18653/v1/2021.naacl-main.405. URLhttps://aclanthology.org/2021. naacl-main.405/. William Merrill and Ashish Sabharwal. A logic for expressing log-precision transformers. InProceedings of the 37th International Conference on ...
2021 doi
-
[2023]
the illusion of thinking
Association for Computational Linguis- tics. doi: 10.18653/v1/2023.findings-acl.67. URLhttps://aclanthology.org/2023.findings-acl.67/. 12 Sheraz Khan, Subha Madhavan, and Kannan Natarajan. A comment on "the illusion of thinking": Reframing the reasoning cliff as an agentic gap...
2023 arXiv
-
[2024]
URLhttps://aclanthology.org/2024
doi: 10.18653/v1/2024.acl-long.65. URLhttps://aclanthology.org/2024. acl-long.65/. Bishwamittra Ghosh, Soumi Das, Till Speicher, Qinyuan Wu, Mohammad Aflah Khan, Deepak Garg, Kr- ishna P. Gummadi, and Evimaria Terzi. Fine-tuning vs. in-context learning in large language models...
2024 doi
-
[2025]
Measuring faithfulness in chain-of- thought reasoning
Tamera Lanham, Anna Chen, Ansh Radhakrishnan, Benoit Steiner, Carson Denison, Danny Hernandez, Dustin Li, Esin Durmus, Evan Hubinger, Jackson Kernion, et al. Measuring faithfulness in chain-of- thought reasoning. arXiv:2307.13702,
-
[2026]
Chu-Cheng Lin, Aaron Jaech, Xin Li, Matthew R
URL https://arxiv.org/abs/2604.20811. Chu-Cheng Lin, Aaron Jaech, Xin Li, Matthew R. Gormley, and Jason Eisner. Limitations of autoregres- sive models and their alternatives. In Kristina Toutanova, Anna Rumshisky, Luke Zettlemoyer, Dilek Hakkani-Tur, Iz Beltagy, Steven Bethard...
2021 arXiv
Reviewed August 6, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.