HyperAIHyperAI

Command Palette

Search for a command to run...

コインには裏表がある:大規模言語モデルのオンポリシー蒸留における汎化の二面性について

概要

オンポリシー蒸留(OPD)は、生徒モデル自身の方策からサンプリングされた軌跡を教師信号として用いることで教師の能力を転移させる手法であるが、その汎化挙動は十分に理解されていない。これは、多くの研究がOPDを単一のドメインや訓練データに近いベンチマークでのみ評価しているためである。本研究では、ドメイン内分布シフトからクロスドメイン転移、複数教師設定に至るまで、一度に一つの汎化要因を変化させる統制実験を実施する。その結果、OPDは特定の問題に対する解答そのものではなく、教師の推論行動を転移させることが明らかになった。すなわち、訓練の難易度はほとんど影響せず、教師が決して解けない問題でさえ有用である。転移は教師と生徒の出自関係に強く依存し、同一出自のペアでは、言語、推論の深さ、さらには他ドメインにわたって生徒が教師に近づく一方、異なる出自のペアでは、主に訓練分布への適合にとどまる。この広範な到達性は両刃の剣である。特定のプロンプトをドメインエキスパートに振り分けても各教師の影響を限定できないため、それらを組み合わせると、各能力の間で混合依存的なシーソー現象が生じる。これらの結果は、OPDがいつ汎化するかを明確にし、複数教師OPDを診断するための有用な視点を提供する。

One-sentence Summary

Researchers from the University of Science and Technology of China, Peking University, and other institutions demonstrate that on-policy distillation transfers a teacher's reasoning behavior rather than specific answers, with same-origin teacher-student pairs enabling broad cross-domain generalization while cross-origin pairs mainly fit the training distribution, and that multi-teacher distillation produces a mixture-dependent seesaw effect among capabilities, thereby clarifying when OPD generalizes.

Key Contributions

  • A controlled study systematically varies generalization factors in on-policy distillation (OPD) across math, code, science, and instruction-following domains, isolating in-domain shifts in problem difficulty, language, and reasoning horizon, cross-domain transfer, and multi-teacher settings.
  • OPD transfers a teacher’s reasoning behavior rather than its answers to specific problems: training problem difficulty barely matters, and even problems the teacher never solves are useful.
  • Generalization depends strongly on the teacher-student origin relationship: same-origin pairs achieve broad transfer across languages, reasoning horizons, and even other domains, while cross-origin pairs mostly fit the training distribution; in multi-teacher OPD, routing cannot confine each teacher’s influence, yielding a mixture-dependent seesaw among their capabilities.

Introduction

The authors investigate on-policy distillation (OPD), a technique that transfers capabilities from strong teacher models to smaller student models by supervising the student on its own generated trajectories. While prior work demonstrates OPD improves performance on target tasks, it fails to distinguish whether the student merely fits the training distribution or acquires broader reasoning patterns that generalize to new settings. This gap is critical for understanding how OPD works and for designing effective multi-teacher integration strategies. The authors conduct a controlled study that systematically varies one generalization factor at a time, revealing that OPD transfers the teacher’s reasoning behavior rather than specific problem solutions, that same-origin teacher-student pairs transfer broadly across languages, horizons, and even domains, and that this broad transfer becomes a double-edged sword in multi-teacher OPD, where prompt routing cannot isolate each teacher’s influence, leading to a mixture-dependent seesaw among capabilities.

Experiment

This work evaluates on-policy distillation with single and multiple teachers across math, code, science, and instruction following. It finds that training-problem difficulty has little effect, while dynamically discarding problems the student already solves gives small consistent gains. Same-origin teacher-student pairs transfer capabilities across language, reasoning horizon, and domains much more effectively than cross-origin pairs, which fit mainly the trained distribution. Because a teacher's influence reaches beyond its assigned domain, multi-teacher routing cannot isolate domain experts; changing teacher mixture ratios creates a seesaw effect, and same-origin teachers exert a stronger pull by aligning the student policy as a whole.

Dynamically discarding only the problems the student already solves (pass-rate in [0,1)) yields a small but consistent average improvement across six math benchmarks, while filtering to only unsolved or only solved problems does not help. This suggests that removing already-mastered problems prevents redundant realignment without harming generalization. For Polaris-7B → DS-distill-1.5B, discarding fully solved problems raised average accuracy from 41.4% to 42.0% (+0.6 pp). Restricting to only unsolved (pass-rate=0) or only solved (pass-rate=1) problems resulted in average accuracy equal to or below the no-filtering baseline. The benefit of discarding solved problems was consistent across both teacher-student pairs, with gains on most individual benchmarks.

Varying the prompt mixture ratio between a math-specialist teacher and a science/IF-specialist teacher in multi-teacher online policy distillation shifts the student's math accuracy toward the teacher that receives more prompts. When the math teacher JustRL-1.5B dominates, average math accuracy rises slightly; when the science/IF teacher Nemotron-1.5B dominates, math accuracy falls. This demonstrates the cross-domain seesaw effect, where a teacher's influence pulls performance in domains beyond its assigned teaching area. Increasing the share of prompts given to the math teacher improves the student's math accuracy, while increasing the science/IF teacher's share degrades it. The highest average math accuracy (27.1%) is achieved when the math teacher receives the largest prompt share (J/N=25/8), and the lowest (25.1%) when the science/IF teacher dominates (J/N=2/25). Relative to an equal mixture, a math-teacher-heavy ratio boosts math accuracy by 0.8 percentage points, whereas a science-teacher-heavy ratio reduces it by 1.2 points.

Filtering out problems the student already solves during online policy distillation yields a small but consistent accuracy improvement by preventing redundant realignment, while restricting to only unsolved or only solved problems does not help. Varying the prompt mixture between a math-specialist teacher and a science/instruction-following teacher demonstrates a cross-domain seesaw effect, where the student's math performance shifts toward the teacher that receives more prompts, even beyond that teacher's assigned domain. These results validate that selective problem filtering and teacher prompt allocation are effective levers for guiding student performance in multi-teacher distillation.


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています
コインには裏表がある:大規模言語モデルのオンポリシー蒸留における汎化の二面性について | 論文 | HyperAI超神経