Transformers Stop Thinking Too Early, and a Tiny LoRA Fixes It
Zehao Jin, Ruixuan Deng, Junran Wang
Georgia Institute of Technology
Published September 29, 2026 · on the feed October 2, 2026 · ▲ 73 · arXiv 2609.36585
The abstract
Pretrained transformers use little of their depth to follow references in context. Thirteen base models reliably follow only 1.4-3.6 lines, and extra pretrained loops add little. A task-trained rank-8 LoRA at one early layer extends this computation with all model weights frozen. Qwen3-8B improves from 15.5% to 99% exact accuracy on 24-line chains; a longer-trained LoRA reaches 50 lines. Ouro-1.4B reaches 60 lines after four loops and at least 160 after eight. The LoRA starts a relay: program lines pass on their chain identity through a short range of middle layers. Frozen heads read progressively further up the chain, and removing parent-line attention stops the relay. A frozen-model measurement locates the last useful intervention layer within tolerance in three of four held-out models. Task-specific LoRAs also improve MuSiQue. Default answers therefore understate the computation accessible through a tiny edit. Code and an interactive demo are available at https://lunamos.github.io/stop-thinking-too-early/
The Brief
1:48A short podcast summary of the paper. Written and voiced by AI.
Transcript
Transformers Stop Thinking Too Early, and a Tiny LoRA Fixes It. By Zehao Jin and two others, from the Georgia Institute of Technology. Published on the twenty-ninth of September. Its code repository is worked out: about eighty source files, with a readme and a licence.
Pretrained transformers use little of their depth to follow references in context. Think of a chain of program lines, each pointing back to an earlier one, and a model asked to follow it. Across thirteen base models, the reliable reach is only about one point four to three point six lines, and adding extra pretrained loops adds little. The claim is that this is a default, not a ceiling: a tiny edit unlocks far more, without changing any of the model's own weights.
The edit is a rank-eight LoRA, trained on the task and placed at one early layer, with every model weight frozen. On twenty-four-line chains, Qwen3-8B goes from fifteen point five percent to ninety-nine percent exact accuracy, and a longer-trained LoRA reaches fifty lines. Ouro-1.4B reaches sixty lines after four loops, and at least one hundred and sixty after eight.
The authors also trace how. The LoRA starts a relay: program lines pass on their chain identity through a short range of middle layers, and frozen heads read progressively further up the chain. Remove the attention to the parent line and the relay stops. A measurement on the frozen model locates the last useful intervention layer within tolerance in three of four held-out models.
These are chain-following tasks with task-trained adapters. Beyond them, the abstract reports only that task-specific LoRAs also improve MuSiQue, and the layer-locating measurement worked in three held-out models out of four, not all of them.
What a model does by default and what it can be made to do with a rank-eight nudge are far apart. Default answers, on this reading, understate the computation that a very small edit can reach.
Paper Quest · 7 chapters · about 40 min
Play this paper
Learn it by doing: make the calls the authors made, guess the result before you see it, and defend it in a viva graded on your own words. Chapter 1 is right here.
Chapter 1 🧩 · The problem
Some models seem to give up after a couple of steps. Is that really all they have?
You give a language model a short program: each line names an earlier variable, and at the bottom of the chain sits a noun. It must print that noun.
The road ahead
- 1 · The problem ~7 min Pretrained language models follow only a few lines of a reference chain, and it is unclear whether that is a hard limit or a habit.
- 2 · What came before ~3 min
- 3 · The key idea ~8 min A tiny trained edit at one early layer, with every model weight frozen, lets the frozen layers carry the chain much further.
- 4 · Choose your own research ~3 min
- 5 · Build the method ~8 min Freeze the pretrained model, add a rank-8 LoRA at one early layer, then train only that LoRA on chain-following programs.
- 6 · The experiment ~6 min You’ll guess: With a rank-8 LoRA at one early layer and every model weight frozen, what exact accuracy did Qwen3-8B reach on 24-line reference chains?
- 7 · The viva ~5 min Three questions from the examiners, answered in your own words. Win it for the paper’s trophy.
Free, with Google or an email address. Your stars and progress are kept.