Why does cheap AI fine-tuning flop if you only tweak the attention layers and skip the rest of the network?
Why does LoRA fail when applied only to attention layers?
This explores why LoRA, the cheap fine-tuning method that trains small add-on matrices instead of the whole model, falls short when those add-ons are attached only to the attention layers rather than spread across every layer of the network.
This explores why LoRA, which trains small add-on matrices instead of retraining the whole model, underperforms when it is attached only to attention layers. The clearest evidence in the corpus is about coverage. Thinking Machines Lab found that LoRA can match full fine-tuning in sample efficiency and final performance, but only when it covers all the weight matrices, including the feed-forward (MLP) blocks that sit between attention layers, and only while the dataset stays below LoRA's capacity Can LoRA match full fine-tuning performance in practice?. Read the other way, that finding tells you the attention-only version is the one that falls short. The problem isn't LoRA as a technique. It's where you put it.
The corpus gives an indirect answer to why placement matters so much: attention mostly decides *where the model looks*, not *what it knows*. Several notes treat attention as a routing system. Up to 40% of attention heads can be swapped for short executable programs without hurting question-answering Can executable programs replace transformer attention heads?. A handful of enormous, input-independent activations act as hidden bias terms that steer attention toward particular tokens Do hidden massive activations act as attention bias terms?. Much of attention's failure behavior comes down to noise, meaning focus spread onto irrelevant context. Cancelling that noise improves hallucination, in-context learning and activation outliers all at once Can subtracting attention maps remove noise from context? Can fixing attention noise solve multiple LLM failures at once?. If attention is mainly the switchboard, then fine-tuning only attention lets you re-route information but not change much of what gets computed or stored. The corpus doesn't test this mechanism directly, so treat it as a strong inference, not a proven result.
The flip side is surprising: how much LoRA needs depends on what you're teaching. For reinforcement learning, LoRA at rank 1, the smallest possible add-on, matches full fine-tuning. The reason is that an RL episode carries only a few bits of feedback, about 1000× less information per token than supervised training Why does LoRA rank 1 work for reinforcement learning?. In a related finding, a 1.5B model trained with LoRA alone matched larger fully trained RL models on reasoning, apparently because RL mostly teaches how to lay out the output, not new facts Can small models reason well by just learning output format?. So the attention-only shortfall should matter most when you're trying to add knowledge or rich behavior through supervised data, and least when you're only nudging style or format.
One caution against concluding that attention doesn't matter: changing *how* attention works can be transformative. Switching a decoder-only model from one-directional to two-directional attention turns it into a top text encoder. The bottleneck there was the attention mask, not model size Why do decoder-only models underperform as text encoders?. The takeaway: attention is powerful when you change its structure, but on its own it's a thin place to fit new learning. The corpus doesn't include a head-to-head study of attention-only versus MLP-only LoRA, so the case above is assembled from adjacent evidence, not one decisive experiment.
Sources 8 notes
Thinking Machines Lab found LoRA achieves equivalent sample efficiency and final performance to full fine-tuning on instruction and reasoning datasets, provided it covers all weight matrices and dataset size stays below LoRA's capacity threshold.
Program synthesis recovers executable code matching 99% of attention head behavior. Substituting the best-fit 30–40% of heads with their synthesized programs preserves QA ability, offering formal, testable interpretability instead of natural-language summaries.
A very small number of input-agnostic activations with values up to 100,000× larger than others act as indispensable implicit bias terms and concentrate attention probability onto specific tokens. This phenomenon appears across model sizes and Vision Transformers.
DIFF Transformer computes attention as the difference between two softmax maps, mechanically canceling the shared noise floor while preserving differential signal—analogous to noise-cancelling headphones. This produces sparse, focused attention patterns without sacrificing efficiency.
Cancelling attention noise through differential attention improves hallucination rates, makes in-context learning robust to example order, and reduces activation outliers—suggesting these separate failure modes share one root cause: over-attention to irrelevant context.
Show all 8 sources
Thinking Machines Lab found LoRA at rank 1 achieves parity with full fine-tuning in RL, attributed to RL's O(1) bits of signal per episode versus supervised learning's O(tokens) bits. This means RL absorbs roughly 1000 times less information per token than SFT.
A 1.5B parameter model with LoRA-only post-training matched larger full-parameter RL models on reasoning tasks, suggesting RL teaches output format organization rather than new factual knowledge. This efficiency indicates reasoning and knowledge storage are separable capabilities.
LLM2Vec's unsupervised 3-step process (bidirectional attention + masked prediction + contrastive learning) achieves SOTA on MTEB. The research shows causal masking, not model size, is the representation bottleneck in decoder-only encoders.
Papers this line draws on 8
The research behind the notes this line reads — ranked by how closely each paper relates.
- Tina: Tiny Reasoning Models via LoRA
- Reasoning Circuits in Language Models: A Mechanistic Interpretation of Syllogistic Inference
- LoRA Without Regret
- Differential Transformer
- Reinforcement Learning Finetunes Small Subnetworks in Large Language Models
- Persona Features Control Emergent Misalignment
- Multi-Token Attention
- The Sparse Frontier: Sparse Attention Trade-offs in Transformer LLMs