Draft fine-tuning step

Reports Kimi K3 technical report @ 3cb39df · Kimi K2.5 report (arXiv 2602.02276v2), for the RL step · no training code is published · How to read

Draft fine-tuning step

Training sequences

×

tensor

What the draft is fine-tuned on is not disclosed.

shape
XX [B batch, T positions]
Sources1

Contextual evidence

§4.1.4, Draft Model Fine-Tuning: the pre-trained MTP layer is fine-tuned into an EAGLE-3-style draft model, with the target model frozen and only the draft layer and its feature-fusion projection updated; the draft is unrolled for seven steps and, beyond the first, consumes its own outputs. The input concatenates the outputs of the 1st, 4th and final AttnRes blocks and projects them by a bias-free matrix W_E3 initialized [0 0 I]. The loss is L_LK = −log Σ_x min(p(x), q(x)) (eq. 16), at temperature 1, with no cross-entropy term.

draft · MoonshotAI/Kimi-K3 / k3_tech_report.pdf
Pinned revision 3cb39dfd32e5 ↗ · p. 14