REVIEW 4 major objections 6 minor 1 cited by
Skip a Layer or Loop it? Test-Time Depth Adaptation of Pretrained LLMs
T0 review · 4 major / 6 minor · reviewed 2026-08-06 · deepseek-v4-flash
Pith's one-line read The paper claims that the fixed-depth forward pass of a pretrained LLM can be replaced, per input and without finetuning, by a chain of layers that skips, repeats, or reorders existing layers, improving accuracy and cutting depth.
desk verdict The search space is new and the existence results are real, but the method peeks at the gold answer, so the reported accuracy gains are a fitting upper bound, not test-time adaptation. 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 chain-of-layers (CoLa) search space: for a model with layers $L_1,\dots,L_N$, a path $P=[L_i,L_j,\dots]$ is any sequence formed by skipping, repeating (looping), or reordering those frozen layers, with actions that skip or repeat blocks of $k$ layers $r$ times for $k,r\in\{1,2,3,4\}$. The search engine is Monte Carlo Tree Search with a UCB score $\mathrm{UCB}(P)=\frac{Q(P)}{v(P)}+c\sqrt{\frac{\ln V}{v(P)}}-\lambda\frac{\|P\|}{N}$, where $Q(P)$ is cumulative reward (1 for a correct output, optionally penalized by normalized path length), $v(P)$ is visit count, and the length penalty pushes toward compact paths. After a fixed simulation budget, the procedure returns Pareto-optimal paths trading accuracy against depth.
What would settle it
Run the same MCTS search with the reward taken from the model's own output confidence instead of gold labels; if the share of wrong-to-correct corrections collapses to near zero, the claimed error correction depends on oracle access to answers rather than on layer composition alone.
Extended reading notes
Core claim
The paper's central claim is that a fixed forward pass is rarely the best execution path for any given input. On the paper's terms, the layers of a pretrained LLM are latent modules that can be composed into a custom Chain-of-Layers (CoLa): some layers are skipped, some are repeated like an RNN loop, and the order can change, all without updating any weight. The authors develop an MCTS procedure that searches this space per sample and claims it consistently finds higher-quality and/or shallower paths, correcting more than 60% of originally incorrect predictions and shortening more than 75% of originally correct ones. They further report that the joint skip-plus-recurrence space beats either operation alone and that error-correcting paths are often shorter than paths that merely preserve a correct answer, which they interpret as evidence of 'overthinking' in static models.
Load-bearing premise
The search is scored against the gold answer for each test input, so the reported accuracy and depth gains presuppose that the correct label is available while searching.
Editorial extensions
If this is right
- For most test inputs the default forward path is not optimal: an MCTS search over layer compositions typically finds a different path that is at least as accurate and usually shorter.
- Allowing skipping and recurrence together is strictly more expressive than either alone, with the largest accuracy gains appearing on the hardest DART-Math levels.
- Correcting an originally wrong prediction usually takes fewer layers than preserving an already correct one, implying that many errors come from passing through misleading layers rather than from insufficient depth.
- Instruction-tuned models behave differently from base models: their layers engage more evenly, and they admit less aggressive depth compression.
- Across models, early layers are almost never skipped, middle layers are the most frequently skipped, and recurrence tends to concentrate in later layers.
Reading between the lines
- Because the MCTS reward is the gold answer, the >60% error-correction headline presupposes access to correct labels at search time; a deployment version would need a calibrated confidence proxy, and nothing in the paper guarantees the gains survive without oracle feedback.
- If per-input search can be made cheap, the same machinery suggests a test-time compute scheduler that allocates depth per query from an estimated difficulty, complementing early-exit and adaptive-computation lines of work.
- The very large jumps on ARC (for example, LLaMA-3-3B-Base from 27.8% to 95.8% on ARC-Easy) suggest that a few high-impact 'poison' layers in the default path may be responsible for a large share of errors, which could be tested directly by ablating individual layers.
- A concrete follow-up: train a small amortized predictor that proposes CoLa paths for new inputs without running MCTS, or distill the discovered paths into a single shallower static model.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes Chain-of-Layers (CoLa), a per-input layer composition space in which a frozen pretrained LLM's layers can be skipped, repeated, or reordered, and introduces an MCTS protocol to search this space. On ARC and DART-Math benchmarks, the authors report large accuracy gains over the original forward pass, as well as depth reductions for both originally correct and originally incorrect samples. The central problem is that the search reward is defined as correctness against the gold test answer, and the evaluation protocol explicitly feeds gold-answer comparisons back into the MCTS search. Consequently, the reported accuracy and depth results are oracle upper bounds: they show that there exist per-sample paths that produce the gold answer, not that a label-free test-time adaptation method can find them.
Significance. The compositional search space—jointly allowing skipping, recurrence, and reordering of pretrained layers—is novel and well motivated, and the existence question (does a fixed pretrained LLM contain better per-input paths?) is interesting. The MCTS protocol is clearly specified, with fixed hyperparameters and a reproducible sampling procedure. However, the paper's stated contribution of 'test-time depth adaptation' and its headline findings of error correction and depth reduction are not supported by the evaluation, because the search is rewarded with the gold answer for each test sample. If reframed as an oracle-guided existence proof, the work would be a useful empirical study; as written, the central practical claim is invalid.
major comments (4)
- [§3 Search Objective; §4 Evaluation Protocol] The search reward uses the gold test answer. Section 3 states 'A reward of 1 is assigned for a correct output,' and Section 4 states that 'model predictions are compared against gold answers' and that 'these outputs serve as feedback for MCTS to iteratively search optimized layer paths per input.' Algorithm 1's simulation step therefore evaluates each candidate path on the test input using that input's gold label. As a result, Table 1 reports the accuracy of paths selected to maximize the probability of matching the gold answer, i.e., an oracle upper bound, not the accuracy of a deployable test-time adaptation method. The findings that CoLa 'corrects' >60% of originally wrong samples and 'shortens' paths for >75% of originally correct samples follow directly from this reward definition.
- [§3, final paragraph; Algorithm 1] No label-free selection mechanism is provided. After the MCTS search, the paper returns Pareto-optimal paths balancing accuracy and length, but at inference time there is no way to know which of the returned paths is correct without the gold label. The paper does not propose a confidence score, a held-out validation criterion, or any surrogate for the reward. Thus, even if the paths exist, the procedure as described cannot be run at test time, and calling it 'test-time depth adaptation' is misleading.
- [§5.1, Finding 2 and Finding 3; Figures 2 and 4] The depth-reduction and error-correction claims are circular with respect to the search objective. For W→C samples, the path is selected precisely because it produces the gold answer; for C→C samples, the path is selected among those that already produce the gold answer and then minimized in depth. Therefore the observation that 'reducing depth helps correct errors' and the conclusion that the original model 'overthinks' are confounded by the selection criterion. The data show that an oracle can find short correct paths, not that the original model's depth is generally redundant for its actual predictions.
- [Table 1; §4 Datasets] There are no error bars, repeated seeds, or significance tests anywhere in the paper, and all results are based on 500 randomly sampled instances per dataset. Several numbers in Table 1 are identical across very different models (e.g., 95.80 on ARC-E and 98.20 on ARC-C appear for nearly all model variants), which suggests either a saturation effect of the oracle search or an artifact of the sampling protocol; neither possibility is discussed. This lack of uncertainty quantification undermines even the weaker existence-proof reading of the results.
minor comments (6)
- [Algorithm 1] The symbol N is used both for the number of layers in the model (Section 3) and as the loop variable in Algorithm 1, which is confusing; please use a different loop variable.
- [§3, UCB equation] The UCB formula uses v(P), V, c, and λ without defining their values in the main text; the length penalty ||P||/N should also clarify whether N is the original layer count. The value λ=5.0 appears only in Appendix B.
- [§4 Datasets] The description of 'randomly sample 500 instances from each dataset' is underspecified: please state the sampling seed, whether the samples are disjoint across model evaluations, and which official splits of ARC and DART-Math are used.
- [§5.2, Finding 4] The entropy values '3.46 v.s. 3.33' and maximum layer concentration '0.035 v.s. 0.040' are reported without stating how they are aggregated across datasets and whether the difference is statistically meaningful.
- [References] The reference list mixes arXiv identifiers without a consistent format; please unify the citation style.
- [Throughout] The claim 'without training any parameters' is technically true, but the MCTS hyperparameters (number of simulations, λ, c, random-child probability) are manually chosen and fixed; this should be stated explicitly in the main text to avoid implying that no hyperparameter choices are made.
Circularity Check
Reported accuracy and depth gains are the objective of an oracle search rewarded with gold answers, so the headline findings are upper bounds rather than a deployable test-time adaptation result.
-
fitted input called prediction
[Section 4, Evaluation Protocol]
"Each input is prompted as a natural language question, and model predictions are compared against gold answers. These outputs serve as feedback for MCTS to iteratively search optimized layer paths per input."
The MCTS search is rewarded for correctness against the gold answer on the actual test input, so the 'optimized' path is selected to maximize correctness for that sample. Reporting the accuracy of these selected paths (e.g., the abstract's '>60% of samples with originally incorrect predictions, we can identify CoLa achieving correct predictions') reports the optimized value of the search objective, not the performance of a label-free inference method. This is per-sample fitting to the test labels, not test-time adaptation: a deployable method would need to choose paths without access to the gold answer.
-
fitted input called prediction
[Section 3, Transition and Simulation (UCB objective)]
"A reward of 1 is assigned for a correct output, optionally minus a penalty proportional to the normalized path length to encourage compact solutions."
Both headline findings are encoded in the search objective: correctness is the reward term and compactness is the length penalty. The claimed 'shorter CoLa for >75% of samples with correct predictions' and 'correct predictions for >60% of samples with originally incorrect predictions' are therefore the result of optimizing these exact terms. They are properties of the objective being optimized, not independent discoveries about the pretrained model's layer space.
full rationale
The paper's central experimental claim is not self-contained as a test-time adaptation method: MCTS is run per test input with a reward defined as correctness against the gold label (Section 3: 'A reward of 1 is assigned for a correct output...'; Section 4: 'model predictions are compared against gold answers. These outputs serve as feedback for MCTS'). Consequently, the reported accuracy gains and depth reductions are the optimized values of an oracle objective. They are upper bounds on what a label-free method could achieve, so presenting them as evidence of test-time depth adaptation is a fitted-input-called-prediction circularity. The descriptive layer-usage statistics (Finding 4 and 5) are not circular—they analyze the paths found under this oracle—but they do not rescue the central predictive claim. No load-bearing self-citation or imported uniqueness theorem appears; the issue is the gold-answer reward, not citation practice. Score is 6 rather than 8 because the search is not guaranteed to attain the objective (the numbers are empirical upper bounds), and the layer-engagement analyses have independent descriptive content.
Assumptions & free parameters
free parameters (6)
- Per-sample CoLa path (skip/repeat decisions) =
selected by MCTS per test sample
- Number of MCTS simulations =
200
- Length penalty lambda =
5.0
- UCB exploration constant c =
not reported
- Skip/repeat block sizes k, r =
{1,2,3,4}
- Random unexplored child probability =
0.1
assumptions (2)
- domain assumption Transformer layers of a pretrained LLM can be skipped, repeated, and composed in new paths without finetuning and still yield valid predictions.
- domain assumption Ground-truth answers are available to score candidate paths at test time.
Cite this review
Pith. "Pith review of Skip a Layer or Loop it? Test-Time Depth Adaptation of Pretrained LLMs." pith.science (2026). https://pith.science/paper/7OYALTYK
@misc{pith2026250707996,
author = {Pith},
title = {Pith review of: Skip a Layer or Loop it? Test-Time Depth Adaptation of Pretrained LLMs},
year = {2026},
howpublished = {\url{https://pith.science/paper/7OYALTYK}},
note = {Machine review of arXiv:2507.07996}
}
read the original abstract
Can a pretrained neural network adapt its architecture to different inputs without any finetuning? Do we need all layers for simple tasks, and are they adequate for challenging tasks? We found that the layers of a pretrained large language model (LLM) can be manipulated as separate modules to build a better and even shallower model customized for each test sample. In particular, each layer from the pretrained model can be skipped/pruned or repeated multiple times as recurrent neural networks (RNN), and stacked with others in arbitrary orders, yielding a chain-of-layers (CoLa) per sample. This compositional space greatly expands the scope of existing works on looped/recurrent pretrained modules, layer pruning, or early-exit networks. We develop a Monte Carlo Tree Search (MCTS) protocol to explore and identify the optimal CoLa for each sample from math and commonsense reasoning benchmarks. Compared to a static model of a fixed depth, CoLa allows shortcut paths (fast thinking), recurrence of the same layer(s) (slow thinking), and combining both, offering more flexible, dynamic architectures for different inputs. We conduct an extensive analysis of the MCTS-optimized CoLa, which leads to two key findings: (1) For >75% of samples with correct predictions by the original LLM, we can find shorter CoLa, suggesting a large space for improving inference efficiency; (2) For >60% of samples with originally incorrect predictions, we can identify CoLa achieving correct predictions, suggesting a large space of performance enhancement. Our results highlight the shortcomings of using a fixed architecture of pre-trained LLMs for inference on different samples and pave the way to unlock the generalization power of test-time depth adaptation.
Figures
Figures from the paper (4 more)
Forward citations
Cited by 1 Pith paper
-
MACRO: Markov Chain Routing of Transformer Layers
A Markov-chain policy over transformer layers, decoded with top-k Viterbi, learns frozen-model routes that improve test accuracy by 5.0% on average.
Reference graph
Works this paper leans on
-
[1]
Neural module networks
Jacob Andreas, Marcus Rohrbach, Trevor Darrell, and Dan Klein. Neural module networks. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 39–48, 2016
2016
-
[2]
Early exit optimizations for additive machine learned ranking systems
B Barla Cambazoglu, Hugo Zaragoza, Olivier Chapelle, Jiang Chen, Ciya Liao, Zhaohui Zheng, and Jon Degenhardt. Early exit optimizations for additive machine learned ranking systems. In Proceedings of the third ACM international conference on Web search and data mining, pages 411–420, 2010
work page 2010
-
[3]
Inner thinking transformer: Leveraging dynamic depth scaling to foster adaptive internal thinking
Yilong Chen, Junyuan Shang, Zhenyu Zhang, Yanxi Xie, Jiawei Sheng, Tingwen Liu, Shuohuan Wang, Yu Sun, Hua Wu, and Haifeng Wang. Inner thinking transformer: Leveraging dynamic depth scaling to foster adaptive internal thinking. arXiv preprint arXiv:2502.13842, 2025
arXiv 2025
-
[4]
Think you have solved question answering? try arc, the ai2 reasoning challenge
Peter Clark, Isaac Cowhey, Oren Etzioni, Tushar Khot, Ashish Sabharwal, Carissa Schoenick, and Oyvind Tafjord. Think you have solved question answering? try arc, the ai2 reasoning challenge. arXiv preprint arXiv:1803.05457, 2018
arXiv 2018
-
[5]
Mostafa Dehghani, Stephan Gouws, Oriol Vinyals, Jakob Uszkoreit, and Łukasz Kaiser. Uni- versal transformers. arXiv preprint arXiv:1807.03819, 2018
arXiv 2018
-
[6]
Faith and fate: Limits of transformers on compositionality
Nouha Dziri, Ximing Lu, Melanie Sclar, Xiang Lorraine Li, Liwei Jiang, Bill Yuchen Lin, Sean Welleck, Peter West, Chandra Bhagavatula, Ronan Le Bras, et al. Faith and fate: Limits of transformers on compositionality. Advances in Neural Information Processing Systems, 36: 70293–70332, 2023
work page 2023
-
[7]
DACT-BERT: Differentiable Adaptive Computation Time for an Efficient BERT Inference
Cristóbal Eyzaguirre, Felipe del Rio, Vladimir Araujo, and Alvaro Soto. Dact-bert: Differen- tiable adaptive computation time for an efficient bert inference.arXiv preprint arXiv:2109.11745, 2021
work page Pith review arXiv 2021
-
[8]
Reducing transformer depth on demand with structured dropout
Angela Fan, Edouard Grave, and Armand Joulin. Reducing transformer depth on demand with structured dropout. arXiv preprint arXiv:1909.11556, 2019
arXiv 1909
Show all 26 references
-
[9]
Looped transformers for length generalization
Ying Fan, Yilun Du, Kannan Ramchandran, and Kangwook Lee. Looped transformers for length generalization. arXiv preprint arXiv:2409.15647, 2024
2024 arXiv
-
[10]
Scaling up test-time compute with latent reasoning: A recurrent depth approach
Jonas Geiping, Sean McLeish, Neel Jain, John Kirchenbauer, Siddharth Singh, Brian R Bartold- son, Bhavya Kailkhura, Abhinav Bhatele, and Tom Goldstein. Scaling up test-time compute with latent reasoning: A recurrent depth approach. arXiv preprint arXiv:2502.05171, 2025
2025 arXiv
-
[11]
Compressing bert: Studying the effects of weight pruning on transfer learning
Mitchell A Gordon, Kevin Duh, and Nicholas Andrews. Compressing bert: Studying the effects of weight pruning on transfer learning. arXiv preprint arXiv:2002.08307, 2020
2002 arXiv
-
[12]
Mixture of nested experts: Adaptive processing of visual tokens
Gagan Jain, Nidhi Hegde, Aditya Kusupati, Arsha Nagrani, Shyamal Buch, Prateek Jain, Anurag Arnab, and Sujoy Paul. Mixture of nested experts: Adaptive processing of visual tokens. arXiv preprint arXiv:2407.19985, 2024
2024 arXiv
-
[13]
Fastbert: a self-distilling bert with adaptive inference time
Weijie Liu, Peng Zhou, Zhe Zhao, Zhiruo Wang, Haotang Deng, and Qi Ju. Fastbert: a self-distilling bert with adaptive inference time. arXiv preprint arXiv:2004.02178, 2020
2004 arXiv
-
[14]
Faster depth-adaptive transformers
Yijin Liu, Fandong Meng, Jie Zhou, Yufeng Chen, and Jinan Xu. Faster depth-adaptive transformers. In Proceedings of the AAAI Conference on Artificial Intelligence , volume 35, pages 13424–13432, 2021
2021
-
[15]
Ebert: Efficient bert inference with dynamic structured pruning
Zejian Liu, Fanrong Li, Gang Li, and Jian Cheng. Ebert: Efficient bert inference with dynamic structured pruning. In Findings of the Association for Computational Linguistics: ACL-IJCNLP 2021, pages 4814–4823, 2021
2021
-
[16]
Anytime dense prediction with confidence adaptivity
Zhuang Liu, Zhiqiu Xu, Hung-Ju Wang, Trevor Darrell, and Evan Shelhamer. Anytime dense prediction with confidence adaptivity. arXiv preprint arXiv:2104.00749, 2021
2021 arXiv
-
[17]
On limitations of the transformer architecture
Binghui Peng, Srini Narayanan, and Christos Papadimitriou. On limitations of the transformer architecture. In First Conference on Language Modeling, 2024. 10
2024
-
[18]
Scaling language models: Methods, analysis & insights from training gopher
Jack W Rae, Sebastian Borgeaud, Trevor Cai, Katie Millican, Jordan Hoffmann, Francis Song, John Aslanides, Sarah Henderson, Roman Ring, Susannah Young, et al. Scaling language models: Methods, analysis & insights from training gopher. arXiv preprint arXiv:2112.11446, 2021
2021 arXiv
-
[19]
You need multiple exiting: Dynamic early exiting for accelerating unified vision language model
Shengkun Tang, Yaqing Wang, Zhenglun Kong, Tianchi Zhang, Yao Li, Caiwen Ding, Yanzhi Wang, Yi Liang, and Dongkuan Xu. You need multiple exiting: Dynamic early exiting for accelerating unified vision language model. In Proceedings of the IEEE/CVF Conference on Computer Vision ...
2023
-
[20]
Branchynet: Fast inference via early exiting from deep neural networks
Surat Teerapittayanon, Bradley McDanel, and Hsiang-Tsung Kung. Branchynet: Fast inference via early exiting from deep neural networks. In 2016 23rd international conference on pattern recognition (ICPR), pages 2464–2469. IEEE, 2016
2016
-
[21]
Dart-math: Difficulty- aware rejection tuning for mathematical problem-solving
Yuxuan Tong, Xiwen Zhang, Rui Wang, Ruidong Wu, and Junxian He. Dart-math: Difficulty- aware rejection tuning for mathematical problem-solving. 2024. URL https://arxiv.org/ abs/2407.13690
2024 arXiv
-
[22]
Routing experts: Learning to route dynamic experts in multi-modal large language models
Qiong Wu, Zhaoxi Ke, Yiyi Zhou, Xiaoshuai Sun, and Rongrong Ji. Routing experts: Learning to route dynamic experts in multi-modal large language models. arXiv preprint arXiv:2407.14093, 2024
2024 arXiv
-
[23]
Deebert: Dynamic early exiting for accelerating bert inference
Ji Xin, Raphael Tang, Jaejun Lee, Yaoliang Yu, and Jimmy Lin. Deebert: Dynamic early exiting for accelerating bert inference. arXiv preprint arXiv:2004.12993, 2020
2004 arXiv
-
[24]
Lgvit: Dynamic early exiting for accelerating vision transformer
Guanyu Xu, Jiawei Hao, Li Shen, Han Hu, Yong Luo, Hui Lin, and Jialie Shen. Lgvit: Dynamic early exiting for accelerating vision transformer. In Proceedings of the 31st ACM International Conference on Multimedia, pages 9103–9114, 2023
2023
-
[25]
Looped transformers are better at learning learning algorithms
Liu Yang, Kangwook Lee, Robert Nowak, and Dimitris Papailiopoulos. Looped transformers are better at learning learning algorithms. arXiv preprint arXiv:2311.12424, 2023
2023 arXiv
-
[26]
Bert loses patience: Fast and robust inference with early exit
Wangchunshu Zhou, Canwen Xu, Tao Ge, Julian McAuley, Ke Xu, and Furu Wei. Bert loses patience: Fast and robust inference with early exit. Advances in Neural Information Processing Systems, 33:18330–18341, 2020. 11 A Fine-Grained Depth Compression Analysis 20 22 24 26 28 303B D...
2020
Reviewed August 6, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.