Papers
arxiv:2607.01232

Is One Layer Enough? Training A Single Transformer Layer Can Match Full-Parameter RL Training

Published on Jul 2
· Submitted by
Mingyi Hong
on Jul 8
Authors:
,
,
,
,
,
,

Abstract

Reinforcement learning adaptation in transformer models shows highly concentrated improvements in specific middle layers rather than uniform parameter updates across all layers.

Reinforcement learning (RL) has become a central component of post-training large language models (LLMs), yet little is understood about how RL adaptation is distributed across transformer layers. Existing approaches typically update all model parameters uniformly, implicitly assuming that every layer contributes similarly to the gains obtained during RL post-training. In this work, we challenge this assumption through a systematic layer-wise study of RL training. Surprisingly, we find that training a single transformer layer can recover most of the gains achieved by full-parameter RL training, and in some cases even surpass it. To quantify this phenomenon, we introduce the quantity layer contribution, which measures the fraction of full RL improvement recovered by training a layer in isolation. Across seven models spanning two model families (Qwen3, Qwen2.5), three RL algorithms (GRPO, GiGPO, Dr. GRPO), and multiple task domains including mathematical reasoning, code generation, and agentic decision-making, we observe a remarkably stable pattern: RL gains are highly concentrated in a small subset of, and in many cases even a single, transformer layers. More strikingly, the same structural pattern consistently emerges: high-contribution layers concentrate in the middle of the transformer stack, while layers near the input and output ends contribute substantially less. The resulting layer rankings remain strongly correlated across datasets, tasks, model families, and RL algorithms.

Community

Reinforcement learning (RL) has become a central component of post-training large language models (LLMs), yet little is understood about how RL adaptation is distributed across transformer layers. Existing approaches typically update all model parameters uniformly, implicitly assuming that every layer contributes similarly to the gains obtained during RL post-training. In this work, we challenge this assumption through a systematic layer-wise study of RL training. Surprisingly, we find that training a single transformer layer can recover most of the gains achieved by full-parameter RL training, and in some cases even surpass it. To quantify this phenomenon, we introduce the quantity layer contribution, which measures the fraction of full RL improvement recovered by training a layer in isolation. Across seven models spanning two model families (Qwen3, Qwen2.5), three RL algorithms (GRPO, GiGPO, Dr. GRPO), and multiple task domains including mathematical reasoning, code generation, and agentic decision-making, we observe a remarkably stable pattern: RL gains are highly concentrated in a small subset of, and in many cases even a single, transformer layers. More strikingly, the same structural pattern consistently emerges: high-contribution layers concentrate in the middle of the transformer stack, while layers near the input and output ends contribute substantially less. The resulting layer rankings remain strongly correlated across datasets, tasks, model families, and RL algorithms.

This is a genuinely surprising result. Training a single transformer layer to match full-parameter RL gains suggests that most of the parameter update surface during RL post-training is either redundant or actively competing. I'd love to see the layer-wise gradient norms during training — if the signal is concentrated in one or two layers, that tells us something fundamental about how RL reshapes the model versus how pretraining or SFT does.

The practical implication is obvious: if this holds across model sizes and RL algorithms, you can cut post-training compute by an order of magnitude. But I'm skeptical about the generality. RL for math reasoning (where reward is dense and per-token) might distribute differently than RL for instruction following (where reward is sparse and end-of-sequence). The paper tests GRPO on math — I'd want to see the same experiment with PPO on a chat benchmark before betting on the claim.

Also worth noting: "matching" full-parameter performance doesn't mean the single layer learns the same thing. The internal representations could diverge significantly even if downstream metrics converge. That matters for downstream fine-tuning or chaining — if you later SFT on top of a model that was RL-trained on only one layer, do you retain the RL gains? The paper doesn't address this, but it's the next question anyone deploying this should ask.

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2607.01232
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2607.01232 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2607.01232 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2607.01232 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.