Apple Machine Learning ResearchT1·78 pts
NTM: Exact Likelihood Four-Step Image Generation via Normalizing Flows
Original: Normalizing Trajectory Models
- NTM models reverse steps as expressive conditional normalizing flows, maintaining exact likelihood training even when compressed to four coarse transitions.
- The architecture combines shallow invertible blocks with a deep parallel predictor, supporting both end-to-end training from scratch and initialization from pretrained flow-matching models.
- A self-distillation mechanism based on exact trajectory likelihood allows lightweight denoisers to generate high-quality samples using the model's own induced score function.