Multi-teacher on-policy distillation
×Stage · Kimi K3
The nine experts are consolidated into one model: for a domain and a sampled effort, the expert of that pair scores the student’s own tokens.
- Teachers
- the expert of the domain and the sampled effort
- Reward
- clipped log-ratio of teacher to student, per token
- How it trains
- The reward enters the same RL framework, partial rollouts included
Not disclosed R_max · Where the student starts · Steps · Learning rate · Tokens
Why
Distillation consolidates the experts’ capabilities at every reasoning effort into a single model. Kimi K3 report §4.1.3, p. 13
The dense per-token reward fits the same RL framework, partial rollouts included. Kimi K3 report §4.1.3, p. 14
The training step →What one step of this stage computes, as a plate.
Architecture at this stageOne student model; the nine experts are its teachers.
All settings3
- Teachers
- For a domain d and a sampled effort e ∈ {low, high, max}, the expert of that pair among the nine
- Reward
- Per token: the clipped, stop-gradient log-ratio of the teacher’s probability to the student’s, within ±R_max (eq. 15)
- How it trains
- The reward enters the same RL framework, partial rollouts included
Sources3
- k3_tech_report.pdf, p. 13Teachers
- k3_tech_report.pdf, pp. 13–14Reward
- k3_tech_report.pdf, p. 14How it trains