LM2LM:用 RL 训练 LLM 提出语言模型改进方案
Language Models Learn to Improve Language Models
用世界模型当奖励模型训练 AI 自己改进模型,Qwen3.8-27B 训完居然超过 GPT-6-sol,成本还降了 33%。
这篇 arXiv 论文提出 LM2LM 方法,用强化学习训练 LLM 生成有效的语言模型改进提案。为解决真实反馈昂贵的难题,团队使用研究世界模型(RWM)作为 RL 的奖励模型,从上下文中的真实实验记录预测提案收益,并定期将当前策略提案的真实执行结果回填到上下文以保持反馈可靠。在八个自回归和扩散语言模型开发环境中,Qwen3.8-27B 经 LM2LM 训练后在留出任务的真实执行评测中超越了 GPT-6-sol,即使后者可参考 300 条真实实验记录,同时数据生成成本降低 33%。该方法还能泛化到训练时未见过的模型家族。
Language Models Learn to Improve Language Models
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.
- IT之家10-17 10:45原文