Language Models Learn to Improve Language Models
Abstract
LM2LM uses a frozen research world model to provide scalable reinforcement learning rewards for generating language model improvements, maintaining reliability through periodic real execution and outperforming larger baselines at lower cost.
Large language models (LLMs) are increasingly used to automate AI research, including language-model (LM) development, offering a potential path toward recursive self-improvement. However, turning experimental feedback into a learned ability to propose effective improvements across models faces two challenges. First, training such an ability with real feedback is expensive: reinforcement learning (RL) requires evaluating tens of thousands of proposals, each potentially requiring a full model-training run. Even at ten GPU-hours per run, reward collection alone would consume hundreds of thousands of GPU-hours, motivating a cheaper surrogate. Second, surrogate feedback can become unreliable as policy optimization favors overestimated proposals when the proposal distribution shifts. To address these challenges, we propose LM2LM, which trains LLMs with RL to generate effective LM improvement proposals. To provide scalable feedback, we use a research world model (RWM), a frozen LLM that predicts proposal gains from real experimental records provided in context, as the reward model for RL. To maintain feedback reliability, we periodically execute current-policy proposals ranked highly by the RWM and add their measured outcomes to its context. Across eight autoregressive and diffusion LM development environments, real-execution evaluations on held-out tasks show that LM2LM enables Qwen3.8-27B to outperform GPT-6-sol, even when GPT-6-sol is given 300 real experimental records in context, while requiring 33% lower data-generation cost. LM2LM also generalizes to a model family unseen during training.
Community
What we do: We study learning to improve language models. Recursive self-improvement needs a model whose ability to improve models grows with experience. We ask whether this ability can be trained into the model with RL, scalably.
How we do it: LM2LM trains an LLM proposer with RL, with two components: (i) Cheap, scalable feedback from a research world model (RWM). Instead of running a training job per proposal, a frozen LLM grounded in real experimental records in its context predicts each proposal's gain, serving as the RL reward model. (ii) On-policy calibration. To prevent the policy from exploiting RWM overestimation, we periodically execute the policy's top-rated proposals and append the results to the RWM's context, calibrating its predictions.
Key results:
- Beats frontier models. After LM2LM training, Qwen3.8-27B outperforms GPT-6-sol equipped with 300 real experimental records in context, and transfers to an unseen model family.
- 200 real runs instead of ~25k, and 33% cheaper than ICL. Training the proposer with RL requires rewards for tens of thousands of proposals; executing each for real (~10 B200 GPU-hours) would cost hundreds of thousands of GPU-hours. LM2LM scores them with RWM tokens instead and executes only 200 runs for calibration. Those 200 runs ($40.1k, 343 h) are also fewer than the 300 records the ICL baseline needs ($60.0k, 510 h), a 33% saving while beating it.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Language Models as AI Research World Models (2026)
- Verify Less, Evolve More: Training Idea-Level Critics for Verification-Efficient ML Evolving Agents (2026)
- Dyad: Extending Large Language Models with Native Typed Decision-Making (2026)
- Teacher-Guided Tool Learning for Small Language Models via Bidirectional KL Constraints (2026)
- CanvasAnneal: Curriculum Reinforcement Learning for Diffusion Language Models (2026)
- Learning to Optimize through Solver-Grounded Self-Play (2026)
- Pretraining Latent Information Feedback Transformers with Teacher Supervision (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper