Mid-Training with Self-Generated Data Improves Reinforcement Learning in Language Models

arXiv cs.AI Papers

Summary

This paper investigates how using diverse self-generated data during mid-training improves the effectiveness of Reinforcement Learning in Large Language Models, particularly for reasoning tasks.

arXiv:2605.08472v1 Announce Type: new Abstract: The effectiveness of Reinforcement Learning (RL) in Large Language Models (LLMs) depends on the nature and diversity of the data used before and during RL. In particular, reasoning problems can often be approached in multiple ways that rely on different forms of reasoning, and exposure to only a limited range of such approaches in the training data may limit the effectiveness of RL. Motivated by this, we investigate using diverse self-generated data during mid-training as an intermediate step before RL training. Specifically, we adopt a bootstrapped data-generation framework guided by George Polya's problem-solving approaches for generating multiple variants of correct answers for each question in the training data, and then perform fine-tuning. We first provide a theoretical perspective on how mid-training on such data improves RL and explain how policy-gradient updates can incentivize combining multiple approaches. We then empirically demonstrate that RL-trained models initialized with our mid-training data achieve consistent improvements across various mathematical reasoning benchmarks and other OOD tasks like code generation and narrative reasoning. Overall, our investigative study shows that a language model learning multiple problem-solving approaches, through self-generated data helps subsequent RL.
Original Article
View Cached Full Text

Cached at: 05/12/26, 07:15 AM

# Mid-Training with Self-Generated Data Improves Reinforcement Learning in Language Models
Source: [https://arxiv.org/abs/2605.08472](https://arxiv.org/abs/2605.08472)
[View PDF](https://arxiv.org/pdf/2605.08472)

> Abstract:The effectiveness of Reinforcement Learning \(RL\) in Large Language Models \(LLMs\) depends on the nature and diversity of the data used before and during RL\. In particular, reasoning problems can often be approached in multiple ways that rely on different forms of reasoning, and exposure to only a limited range of such approaches in the training data may limit the effectiveness of RL\. Motivated by this, we investigate using diverse self\-generated data during mid\-training as an intermediate step before RL training\. Specifically, we adopt a bootstrapped data\-generation framework guided by George Polya's problem\-solving approaches for generating multiple variants of correct answers for each question in the training data, and then perform fine\-tuning\. We first provide a theoretical perspective on how mid\-training on such data improves RL and explain how policy\-gradient updates can incentivize combining multiple approaches\. We then empirically demonstrate that RL\-trained models initialized with our mid\-training data achieve consistent improvements across various mathematical reasoning benchmarks and other OOD tasks like code generation and narrative reasoning\. Overall, our investigative study shows that a language model learning multiple problem\-solving approaches, through self\-generated data helps subsequent RL\.

## Submission history

From: Aswin Ravikumar Rangasamy Veerasamy \[[view email](https://arxiv.org/show-email/ca2f8806/2605.08472)\] **\[v1\]**Fri, 8 May 2026 20:46:35 UTC \(1,820 KB\)

Similar Articles

@neural_avb: https://x.com/neural_avb/status/2072294078805684613

X AI KOLs Timeline

This paper introduces Autodata, a method that uses an agentic 'data scientist' AI to automate the creation of high-quality synthetic datasets through iterative generation, verification, and refinement, specifically optimized for reinforcement learning (GRPO) to improve reasoning in language models.

Improving language model behavior by training on a curated dataset

OpenAI Blog

OpenAI research demonstrates that language model behavior can be significantly improved through fine-tuning on small, curated datasets (<100 examples) targeting specific behavioral values, with effectiveness increasing at larger model scales. The approach provides users with tools to align models with Charter-compatible values for their specific applications.