arXiv:2502.19557cs.CLcs.AI2025-02被引 13

让小模型超越大模型,靠的是学答案也学评分标准。

Distill Not Only Data but Also Rewards: Can Smaller Language Models Surpass Larger Ones?

论文配图:Distill Not Only Data but Also Rewards: Can Smaller Language Models Surpass Larger Ones?
图 1 · 摘自论文原文
  • 用自监督方法从大模型生成内容中提取伪评分信号
  • 在GSM8K和MMLU-PRO上小模型性能超过教师模型
  • 无需外部评分,适合资源有限的模型压缩场景

大型语言模型(LLM)的蒸馏通常仅通过监督微调(SFT)转移教师模型的输出。然而,该方法忽略了同时蒸馏数据(输出内容)与奖励信号(质量评估)的潜力。直接从教师模型中提取可靠奖励信号具有挑战性,因为大模型针对生成优化而非评估,常导致偏差或不一致的判断。为此,我们提出一种新型蒸馏流程,同时传递响应与奖励。该方法通过自监督机制利用教师与学生响应的内在结构生成伪奖励信号,实现无需外部评估的奖励学习。奖励模型随后指导强化学习(RL),在完成SFT预热后迭代优化学生模型。在GSM8K和MMLU-PRO上的实验表明,该方法持续优于传统SFT方式,使学生模型性能超越其教师模型。本工作展示了通过结构化自监督奖励学习实现可扩展、高效的蒸馏,降低对外部奖励监督的依赖。

原文摘要 · Abstract (English)

Distilling large language models (LLMs) typically involves transferring the teacher model's responses through supervised fine-tuning (SFT). However, this approach neglects the potential to distill both data (output content) and reward signals (quality evaluations). Extracting reliable reward signals directly from teacher models is challenging, as LLMs are optimized for generation rather than evaluation, often resulting in biased or inconsistent assessments. To address this limitation, we propose a novel distillation pipeline that transfers both responses and rewards. Our method generates pseudo-rewards through a self-supervised mechanism that leverages the inherent structure of both teacher and student responses, enabling reward learning without explicit external evaluation. The reward model subsequently guides reinforcement learning (RL), allowing iterative refinement of the student model after an SFT warm-up phase. Experiments on GSM8K and MMLU-PRO demonstrate that our method consistently outperforms traditional SFT-based approaches, enabling student models to surpass the performance of their teachers. This work highlights the potential for scalable, efficient distillation through structured self-supervised reward learning, reducing dependence on external reward supervision.

模型蒸馏强化学习自监督小模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。