Search papers, labs, and topics across Lattice.
This paper introduces Flux-OPD, an on-policy distillation method that leverages evolving contexts to enhance supervision in large language model training within open-ended domains. By analyzing the reverse KL objective, the authors identify how context-conditioned teachers influence the student model and propose a mechanism to stabilize the distillation process while addressing conflicts among teacher distributions. Experimental results demonstrate that Flux-OPD significantly outperforms traditional on-policy distillation approaches, effectively capturing task preferences through dynamic context integration.
Evolving contexts can transform the stability and effectiveness of on-policy distillation, leading to superior performance in open-ended tasks.
Large language model training in open-ended domains lacks verifiable rewards, making task preferences difficult to formalize as effective supervision. Contexts can convey such preferences, yet provide little additional supervision once distilled into the student, motivating contexts that evolve with student performance. However, directly using evolving contexts as in-training supervision results in an unstable distillation target and conflicting distributions, requiring mechanisms to stabilize target and downweight conflicts. In this paper, we analyze the effect of contexts through a decomposition of the reverse KL objective, revealing two findings: the student is distilled toward the geometric mean of context-conditioned teachers, and the objective contains a conflict term that measures conflicts among these teachers. Based on this decomposition, we propose Flux-OPD, an OPD paradigm that uses evolving contexts as in-training supervision to capture task preferences in open-ended domains. Flux-OPD treats the differences between context-conditioned and context-free teachers as contextual difference signals, injects them as contextual corrections into the context-free teacher anchor, and weights their correction strength using the conflict term as an indicator. Experiments on open-ended tasks show that Flux-OPD outperforms existing OPD paradigms, highlighting the potential to combine teacher supervision with evolving contexts.
Tsinghua AI