One Layer Is Enough: Single‑Layer RL Beats Full‑Parameter Training Across Models, Tasks, and Algorithms

A systematic study of reinforcement‑learning fine‑tuning for large language models reveals that RL gains are highly concentrated in a few middle Transformer layers, and training just one such layer can match or even exceed full‑parameter RL performance across multiple models, tasks, and algorithms.

Machine Learning Algorithms & Natural Language Processing
Machine Learning Algorithms & Natural Language Processing
Machine Learning Algorithms & Natural Language Processing
One Layer Is Enough: Single‑Layer RL Beats Full‑Parameter Training Across Models, Tasks, and Algorithms

Background and Motivation

Reinforcement learning with verified rewards (RLVR) has become a core component of post‑training large language models, improving mathematical reasoning, code generation, and decision‑making. Existing RL post‑training methods update all Transformer layers, implicitly assuming each layer contributes equally to RL gains. This work questions that assumption.

Layer Contribution Metric

The authors define a simple metric, Layer Contribution C(k), for a model with L layers: freeze all parameters except layer k, train only that layer with RL, and compare the performance gain to the full‑parameter RL baseline. C(k)=1.0 means single‑layer training matches full‑parameter RL; C(k)>1.0 means it surpasses it.

Layer contribution formula
Layer contribution formula

Experimental Setup

The study evaluates 7 models spanning two families (Qwen3 1.7B/4B/8B, Qwen2.5 1.5B/3B/DeepSeek‑Distilled‑7B) with three RL algorithms (GRPO, Dr. GRPO, GiGPO) across three task domains (math reasoning, code generation, agent decision‑making). For Qwen3‑1.7B‑Base, 12 benchmarks covering math, code, reasoning, and language are used. All experiments share identical hyper‑parameters (learning rate, batch size, KL coefficient, clip range, epochs) and use a learning‑rate‑ablation to confirm fairness.

Finding 1: Middle Layers Dominate RL Gains

Across all models, layers in the 40‑60% depth range consistently achieve the highest contribution scores, while early and late layers contribute far less. This pattern holds regardless of model size, RL algorithm, dataset, or task, indicating an intrinsic structural property of pretrained LLMs.

Layer contribution across models
Layer contribution across models

Finding 2: Single‑Layer Training Can Surpass Full‑Parameter RL

For every model, the best single layer attains C(k) ≥ 1.0, meaning it matches or exceeds the full‑parameter RL baseline. For example, on Qwen3‑1.7B‑Base, training only Layer 10 yields a score of 51.8% versus 50.8% for full‑parameter RL (C = 1.14), while the weakest Layer 24 reaches only 46.1% (C = 0.28), a >4× gap.

Single‑layer performance
Single‑layer performance

Practical Strategies Based on Layer Contribution

Strategy 1: Layer‑Adaptive Learning Rate – Increase the learning rate for high‑contribution layers (1e‑5) while keeping others at the default (5e‑6). This yields consistent gains (e.g., Qwen3‑1.7B: 53.70% vs 50.82% baseline).

Strategy 2: Selective Layer Training – Train only the top‑k layers by contribution and freeze the rest. Results: Qwen3‑8B top‑10 layers achieve 69.11% vs 66.43% baseline (+2.68%). Training the lowest‑contribution layers harms performance.

Strategy 3: Zero‑Analysis Heuristic – Without any contribution data, simply train the middle five layers (e.g., Layers 11‑15 for a 28‑layer model). This heuristic matches or exceeds contribution‑guided selection across model sizes.

Training strategies performance
Training strategies performance

Deeper Analysis: Parameter‑Space Quality vs. Magnitude

Two observations challenge the notion that weight‑change magnitude explains contribution:

Full‑parameter RL induces uniformly small L2 weight changes (0.5‑0.8) across layers, yet middle layers contribute disproportionately.

Single‑layer training causes similar L2 weight changes (0.8‑1.0) for both high‑ and low‑contribution layers, but only the former yield large performance gains.

Weight change analysis
Weight change analysis

Thus, layer contribution reflects the effectiveness of a layer’s parameter sub‑space for capturing RL improvements, not merely the amount of weight change.

Conclusion

The study uncovers a previously unrecognized structural property of RL post‑training: (1) RL gains are highly concentrated in a small set of middle layers; (2) training a single such layer can match or exceed full‑parameter RL; (3) simple, contribution‑guided strategies consistently outperform standard full‑parameter RL. These insights provide new directions for understanding and improving RL fine‑tuning of large language models.

Original Source

Signed-in readers can open the original source through BestHub's protected redirect.

Sign in to view source
Republication Notice

This article has been distilled and summarized from source material, then republished for learning and reference. If you believe it infringes your rights, please contactadmin@besthub.devand we will review it promptly.

model trainingreinforcement learningparameter-efficient fine-tuninglayer contribution
Machine Learning Algorithms & Natural Language Processing
Written by

Machine Learning Algorithms & Natural Language Processing

Focused on frontier AI technologies, empowering AI researchers' progress.

0 followers
Reader feedback

How this landed with the community

Sign in to like

Rate this article

Was this worth your time?

Sign in to rate
Discussion

0 Comments

Thoughtful readers leave field notes, pushback, and hard-won operational detail here.