aifollow.news
返回 Rohan Paul
Rohan Paul· @rohanpaul_ai · X· · 原发布时间 AI 评分54

卡内基梅隆大学论文提出用同一训练规则贯通大模型训练阶段

自动核验发布 · 本文由系统生成并完成证据核验,未经人工审稿。

AI 导读

这篇论文将一种强化学习方法改写为基于模型自身正确答案的似然训练,使预训练、微调和强化学习数据能按比例混合。在124M参数模型的推理题实验中,最佳混合训练结果优于最佳分阶段训练;另一项编程测试中,0.76B参数模型取得32.69%,分阶段训练为29.81%。

原帖引用 · 原文

@rohanpaul_ai

The separate stages of AI training can collapse into a single method.

New Carnegie Mellon paper shows that pretraining, fine-tuning, and RL can share a single training rule, and blending their data often beats separate stages.

LLMs are usually trained in stages: pretraining, midtraining, fine-tuning, then RL, each with its own algorithm. Hard switches between stages can also erase earlier skills.

The authors recast a form of RL called MaxRL as likelihood training on the model's own correct answers. That lets one trainer mix pretraining text, curated data, and RL samples in any proportion over time.

On reasoning puzzles with a 124M model, the best blended runs beat the best staged run and kept general skills better. After later coding RL, a 0.76B model that blended in 5% midtraining data during math RL reached 32.69% on MBPP+, versus 29.81% for staged training.

Treat your training pipeline as a data-mixing schedule, and try blending earlier data into RL instead of switching stages abruptly.

来源:Rohan Paul · x.com

论文
发现内容有误?提交纠错