Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
FlowMap-OPD addresses the challenge of on-policy distillation for few-step generative flow-map models like MeanFlow and consistency models by decoupling student trajectory rollouts from teacher-student distribution comparison kernels. Grounded in state-marginal formulation and flow-velocity consistency, the framework bridges local supervision with deployed long-range transport dynamics. Across ImageNet and text-to-image benchmarks, instantaneous-velocity supervision with decoupled student consistency achieves state-of-the-art distillation efficiency, significantly outperforming multi-reward Flow-Map GRPO in convergence speed and multi-specialist consolidation.
Decoupling student rollout generation from teacher supervision kernels enables stable on-policy distillation for few-step flow models, outperforming multi-reward GRPO while accelerating convergence.
Few-step flow-map generators, including MeanFlow and consistency models, enable efficient sampling through long-range transport, yet their on-policy distillation remains underexplored. We introduce FlowMap-OPD, an on-policy distillation framework that separates student-state acquisition from teacher--student distribution comparison. A formulation based on state marginals establishes this separation, while flow--velocity consistency connects local supervision to the deployed long-range map. Within this framework, we develop flow-map, induced-velocity, and instantaneous-velocity distribution supervision, each paired with a separately specified native flow-map rollout. Cross-capacity ImageNet experiments across three teacher rewards identify instantaneous-velocity distribution supervision with independently tunable student consistency as the most effective choice. In text-to-image experiments, FlowMap-OPD demonstrates strong multi-specialist consolidation capabilities and surpasses multi-reward Flow-Map GRPO in task performance and convergence speed.