mlsanityData to 7 Oct 2026

Programs-of-Layers in LLMs through the Lens of Cortical Areas

arXiv 2609.31360 ↗First posted 25 Sep 2026cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

Inference in LLMs is conventionally a fixed-depth, fixed-order forward pass through every layer, regardless of how difficult the input is. The human brain does not work this way: using the thalamus as a central hub, it routes information flexibly to all regions of the cortex according to demand. Li et al. (2026) recently showed, with a system they call program-of-layers (PoLar), that transformers can be given an analogous flexibility if their layers are treated as a library of functions rather than a fixed sequence. Performance improves over the standard forward pass when each input is dynamically routed through an adaptive sequence of skipped or repeated contiguous layer blocks. We reconstructed PoLar's diagnostic MCTS in more detail than the original paper and applied it across 5 models. We reproduced several of PoLar's findings: skipping outperformed the standard pass, repeating outperformed skipping, and combining both outperformed either alone. Shorter programs sufficed for easier questions, while harder questions required more layer repeats. However, we failed to replicate the main claim regarding their learned router for single-shot inference: its top-ranked prediction consistently collapsed back to the standard pass, even though its top-k predicted programs, taken together, did show a real accuracy gain. Beyond reproduction, we find that a small number of generic programs are enough to solve most of the questions. We also provide a much deeper analysis of these programs' structure and robustness: for example, we found that programs that correct errors are highly brittle: undoing even a single edit inside a program typically breaks the correction. Connecting this to the brain's routing mechanisms, PoLar mirrors principles of thalamo-cortical coordination between cortical-area-like transformer layers. We publicly release the code at https://datexis.github.io/RE-PoLar/

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 44 of its 68 references that are in the map. First posted on arXiv 25 Sep 2026.

Builds on

44 of 68 references.

References · 44

  1. 2017Attention Is All You Needcited by 59,932
  2. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  3. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  4. 2021Pre-train, Prompt, and Predict: A Systematic Survey of Prompting Methods in Natural Language Processingcited by 2,222
  5. 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,758
  6. 2023Quantifying Language Models' Sensitivity to Spurious Features in Prompt Design or: How I learned to start worrying about prompt formattingcited by 778
  7. 2018Universal Transformerscited by 613
  8. 2019Reducing Transformer Depth on Demand with Structured Dropoutcited by 434
  9. 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 369
  10. 2020DeeBERT: Dynamic Early Exiting for Accelerating BERT Inferencecited by 322
  11. 2021A Diverse Corpus for Evaluating and Developing English Math Word Problem Solverscited by 314
  12. 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 302
32 more
  1. 2024HippoRAG: Neurobiologically Inspired Long-Term Memory for Large Language Modelscited by 273
  2. 2020BERT Loses Patience: Fast and Robust Inference with Early Exitcited by 267
  3. 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 242
  4. 2024Let's Think Dot by Dot: Hidden Computation in Transformer Language Modelscited by 186
  5. 2020The Hardware Lotterycited by 160
  6. 2024LaCo: Large Language Model Pruning via Layer Collapsecited by 124
  7. 2014Deep Neural Networks Reveal a Gradient in the Complexity of Neural Representations across the Brain's Ventral Visual Pathwaycited by 93
  8. 2024DART-Math: Difficulty-Aware Rejection Tuning for Mathematical Problem-Solvingcited by 92
  9. 2024Looped Transformers for Length Generalizationcited by 91
  10. 2021Coordination Among Neural Modules Through a Shared Global Workspacecited by 83
  11. 2025Inner Thinking Transformer: Leveraging Dynamic Depth Scaling to Foster Adaptive Internal Thinkingcited by 56
  12. 2024MoEUT: Mixture-of-Experts Universal Transformerscited by 55
  13. 2024Transformer Layers as Painterscited by 50
  14. 2023Rosetta Neurons: Mining the Common Units in a Model Zoocited by 46
  15. 2024Memory Layers at Scalecited by 39
  16. 2025Adaptive Layer-skipping in Pre-trained LLMscited by 24
  17. 2025Continuous Thought Machinescited by 21
  18. 2025Block-Recurrent Dynamics in Vision Transformerscited by 16
  19. 2025Skip a Layer or Loop it? Test-Time Depth Adaptation of Pretrained LLMscited by 14
  20. 2023Calc-X and Calcformers: Empowering Arithmetical Chain-of-Thought through Interaction with Symbolic Systemscited by 13
  21. 2025Dr.LLM: Dynamic Layer Routing in LLMscited by 10
  22. 2025Mixture of Cognitive Reasoners: Modular Reasoning with Brain-Like Specializationcited by 9
  23. 2024Router-Tuning: A Simple and Effective Approach for Enabling Dynamic-Depth in Transformerscited by 8
  24. 2025Meaningless Tokens, Meaningful Gains: How Activation Shifts Enhance LLM Reasoningcited by 8
  25. 2025What Layers When: Learning to Skip Compute in LLMs with Residual Gatescited by 6
  26. 2026Skip a Layer or Loop It? Learning Program-of-Layers in LLMscited by 6
  27. 2026Is One Layer Enough? Understanding Inference Dynamics in Tabular Foundation Modelscited by 4
  28. 2025Representational Alignment Across Model Layers and Brain Regions with Multi-Level Optimal Transportcited by 3
  29. 2026Scalable Machines with Intrinsic Higher Mental-State Dynamicscited by 2
  30. 2026From Noise to Diversity: Random Embedding Injection in LLM Reasoningcited by 2
  31. 2025Contrastive Consolidation of Top-Down Modulations Achieves Sparsely Supervised Continual Learningcited by 1
  32. 2026The Neuroscience of Transformerscited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.