New preprint: geometry-aware time reparameterization improves flow-map distillation

FrnkNlsn · x · 2026-10-08

An OIST team (Félix Dedek, Makoto Yamada) posted arXiv:2610.02427 on Geometry-Aware Time Reparameterization for Flow-Map Distillation. Idea: flow-map distillation learns finite-time transitions of a pretrained generative ODE for one/few-step generation; segments with large normal acceleration are harder to distill, so the paper allocates more student time to highly curved regions while preserving the teacher's geometric paths and terminal distribution. Key property: the shared clock is estimated once before distillation — no teacher retraining, no extra student parameters or inference cost. Results: on synthetic data, CIFAR-10 and CelebA-64, sample quality beats identity-time distillation at matched budgets, including one- and two-step generation; gains in one-step generation (where no intermediate times exist) highlight the value of time reparameterization during distillation.

Original post →

More from Research

Research channel →