acceptodds
Under review as a conference paper at ICLR 2027

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

Abstract

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 three task domains (mathematical reasoning, code generation, and agentic decision-making), we observe a remarkably stable pattern: RL gains are highly concentrated in a small subset of transformer layers, and in many cases in a single layer. 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. Our findings have two important implications. First, they reveal a previously unrecognized structural property of RL post-training: most RL gains are concentrated in a small subset of transformer layers rather than being uniformly distributed throughout the network. Second, they suggest new opportunities for improving RL training. Guided by the above observation, we develop simple layer-aware training strategies that consistently outperform standard full-parameter RL training, while ensembles of layer-specialized models provide additional gains through complementary behaviors. Together, our results provide new insights into how RL modifies large language models and suggest a new perspective for understanding and improving RL post-training.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.