Draft fine-tuning step
LK loss
×tensor
The number this step minimizes. Its gradient updates the draft layer and the fusion matrix, and nothing of the target.
- shape
- [1]
Sources
Contextual evidence from LK loss
§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