On-Policy Distillation with Best-of-N Teacher Rollout Selection
基于N最佳教师轨迹选择的在线策略蒸馏
机构 * Johns Hopkins University(约翰霍普金斯大学) ; TikTok ; University of California, San Diego(加州大学圣地亚哥分校) ; Fudan University(复旦大学)
AI总结 本文提出BRTS框架,通过选择更准确且与学生行为一致的教师轨迹提升在线策略蒸馏效果,在AIME和AMC基准测试中取得显著改进。
Comments 10 pages, 5 figures