跳到正文
Aravind Srinivas· @AravSrinivas · X·· 4 小时前AI 评分51
AI 导读

Perplexity 与 NVIDIA 联合发布后训练研究 JOLT,让同一模型同时担任教师和学生,并训练教师从自身尝试的奖励中学习,同时惩罚偏离学生行为的漂移。在 MATH 基准上,JOLT 以 6.5 倍更少的训练 token 达到与 GRPO 相同的准确率;JOLT+ 在数学、编码、终端任务和工具使用基准上明显优于 GRPO。

正文

Sharing new post-training research on self-distillation + RL (JOLT), done together with @NVIDIAAI.

In self-distillation, the same model performs the role of both a teacher and a student. The teacher gets extra information that the student doesn't, and the student learns from feedback provided by the teacher. However, the teacher could still use shortcuts in ways the student has no way to follow.

Our proposed approach, JOLT, trains the teacher as well, to learn from rewards on its own attempts, while penalizing any drift that's far off from what the student model would do. JOLT+ further supplements JOLT by providing the student model outcome-based rewards in addition to the teacher supervision signals.

On the MATH benchmark, we found that JOLT matches GRPO with 6.5x fewer training tokens, while JOLT+ convincingly beats GRPO across math, coding, terminal tasks, and tool use benchmarks.

引用Perplexity@perplexity_ai
New research: We developed JOLT, a post-training method where one model acts as both teacher and student, with the teacher trained to give feedback the student can use. On MATH, it matched GRPO's accuracy with 6.5x fewer training tokens. https://www.perplexity.ai/hub/blog/a-good-teacher-meets-the-student-where-they-are
在 X 查看被引用的帖子

来源:Aravind Srinivas · x.com