How Does RL Post-training Induce Skill Composition? A Case Study on Countdown
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
While reinforcement learning (RL) successfully enhances reasoning in large language models, its role in fostering compositional generalization (the ability to synthesize novel skills from known components) is often conflated with mere length generalization. To this end, we study what RL post-training teaches about skill composition and how the structure of the composition affects the skill transfer. We focus on the Countdown task (given n numbers and a target, form an expression that evaluates to the target) and analyze model solutions as expression trees, where each subtree corresponds to a reusable subtask and thus can be viewed as a ``skill.'' Tracking tree shapes and their success rates over training, we find: (i) out-of-distribution (OOD) generalization to larger n and to unseen tree shapes, indicating compositional reuse of subtasks; (ii) a structure-dependent hierarchy of learnability -- models master shallow balanced trees (workload is balanced between subtasks) before deep unbalanced ones, with persistent fragility on right-heavy structures (even when the composition depth is the same as some left-heavy structures). Our diagnostic reveals what is learned, in what order, and where generalization fails, clarifying how RL-only post-training induces OOD generalization beyond what standard metrics such as pass@k reveal.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 26 of its 35 references that are in the map. First posted on arXiv 1 Dec 2025.
Builds on
26 of 35 references.
Closest ancestors · 5
Same direction, newest first
- 2025Sample More to Think Less: Group Filtered Policy Optimization for Concise Reasoning13 Aug
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?18 Apr
- 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 21
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,155
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2022Exploring Length Generalization in Large Language Modelscited by 204
- 2025REASONING GYM: Reasoning Environments for Reinforcement Learning with Verifiable Rewardscited by 121
- 2023Skill-it! A Data-Driven Skills Framework for Understanding and Training Language Modelscited by 120
- 2023A Theory for Emergence of Complex Skills in Language Modelscited by 111
- 2024Self-Improvement in Language Models: The Sharpening Mechanismcited by 102
- 2024Transformers Can Achieve Length Generalization But Not Robustlycited by 82
- 2023Skill-Mix: a Flexible and Expandable Family of Evaluations for AI modelscited by 64
9 more
- 2025Learning Adaptive Parallel Reasoning with Language Modelscited by 58
- 2025e3: Learning to Explore Enables Extrapolation of Test-Time Compute for LLMscited by 56
- 2025OMEGA: Can LLMs Reason Outside the Box in Math? Evaluating Exploratory, Compositional, and Transformative Generalizationcited by 48
- 2025From f(x) and g(x) to f(g(x)): LLMs Learn New Skills in RL by Composing Old Onescited by 40
- 2025Weight Ensembling Improves Reasoning in Language Modelscited by 36
- 2025Self-Improving Transformers Overcome Easy-to-Hard and Length Generalization Challengescited by 30
- 2022How to Reuse and Compose Knowledge for a Lifetime of Tasks: A Survey on Continual Learning and Functional Compositioncited by 27
- 2024Can Models Learn Skill Composition from Examples?cited by 17
- 2025LLM-First Search: Self-Guided Exploration of the Solution Spacecited by 5
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026The Unlearnability Phenomenon in RLVR for Language Modelscited by 2
- 2026Zipping the Thought: When and How Compressed Reasoning Data Works in LLM Post-Trainingcited by 2
- 2026New Skills or Sharper Primitives? A Probabilistic Perspective on the Emergence of Reasoning in RLVRcited by 1
- 2026Characterizing Model-Native Skillscited by 0
- 2026Provable Benefits of RLVR over SFT for Reasoning Models: Learning to Backtrack Efficientlycited by 0
- 2026Toward Latent Language Model Skills Steering and Optimization: An Empirical Studycited by 0
- 2026Toward Latent Language Model Skills Steering and Optimization: An Empirical Study29 Aug 2026
- 2026Provable Benefits of RLVR over SFT for Reasoning Models: Learning to Backtrack Efficiently22 Jun 2026
- 2026Zipping the Thought: When and How Compressed Reasoning Data Works in LLM Post-Training27 May 2026
- 2026The Unlearnability Phenomenon in RLVR for Language Models16 May 2026
- 2026Characterizing Model-Native Skills19 Apr 2026
- 2026New Skills or Sharper Primitives? A Probabilistic Perspective on the Emergence of Reasoning in RLVR9 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2026Compositional Reasoning in Language Models under Reinforcement Learning Post-Training
- 2026Demystifying Reinforcement Learning Post-Training of Language Models
- 2026Why Does Reinforcement Learning Generalize? A Feature-Level Mechanistic Study of Post-Training in Large Language Models
- 2025Mirage or Method? How Model-Task Alignment Induces Divergent RL Conclusions
- 2026Is One Layer Enough? Training A Single Transformer Layer Can Match Full-Parameter RL Training
- 2025Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretraining
- 2025On the Interplay of Pre-Training, Mid-Training, and RL on Reasoning Language Models
- 2025Breaking Barriers: Do Reinforcement Post Training Gains Transfer To Unseen Domains?