Muse Spark training detail: pure KL logit distillation for 95% of pre-training

andrew_n_carr · x · 2026-10-20

Andrew Carr highlighted a training detail from the Muse Spark technical report: the base model is trained via logit distillation from Muse Spark using co-distillation, logging the top 256 logits during the teacher's pre-training.

Unlike prior formulations blending soft and hard targets, the objective uses only KL divergence for the first 95% of pre-training, then switches to cross-entropy-only loss for the final 5% to address truncation bias in top-k logit distillation. Data spans web, code, math, multilingual text and images, with a knowledge cutoff of 2026-01-04.

Original post →

More from Research

Research channel →