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

Salesforce 研究比较模型蒸馏与强化学习的训练成本

AI 导读

研究按训练 FLOP 对应的留出集准确率比较两种方法:使用外部后训练教师模型时,GRPO 通常更划算;将各领域经 GRPO 训练的专家模型蒸馏为一个学生模型,在数学和代码任务上达到联合 GRPO 的效果,总 FLOP 减少 42.0%。

原帖引用 · 原文

@rohanpaul_ai

New Salesforce paper finds that distilling from a big external teacher usually costs more compute than plain RL, while merging RL-trained domain experts saves 42%.

On-policy distillation lets a student learn from a teacher's per-token feedback on its own outputs, and many expect this dense signal to beat RL. But every training token also needs a teacher forward pass.

They compared distillation and GRPO by held-out accuracy per training FLOP, varying task difficulty, model sizes, domains, and training settings.

With an external post-trained teacher, GRPO generally won, and distillation often inflated response length. Distilling per-domain GRPO experts into 1 student matched joint GRPO on math plus code with 42.0% fewer total FLOPs.

To save compute, train per-domain experts with RL and distill them into 1 model, rather than distilling from a big external teacher.

来源:Rohan Paul · x.com

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