Step distillation
A flow-matching action head integrates the velocity field over several denoising steps. We distill it into a single step with a Drift loss: each generated action is pulled toward the demonstration while sibling samples of the same input push each other apart, which keeps the action distribution diverse without a teacher at inference time.
- GR00T N1.7: the DiT action head runs once instead of 4 times.
- Cosmos3-Edge: the 4-step sampler with classifier-free guidance (8 tower forwards) becomes 1 forward at σ = 1 with guidance off.