Self-Corrected Flow Distillation for Consistent One-Step and Few-Step Image Generation
Flow matching has emerged as a promising framework for training generative models, demonstrating impressive empirical performance while offering relative ease of training compared to diffusion-based models. However, this method still requires numerous function evaluations in the sampling process. To address these limitations, we introduce a self-corrected flow distillation method that effectively integrates consistency models and adversarial training within the flow-matching framework. This work is a pioneer in achieving consistent generation quality in both few-step and one-step sampling. Our extensive experiments validate the effectiveness of our method, yielding superior results both quantitatively and qualitatively on CelebA-HQ and zero-shot benchmarks on the COCO dataset.
Authors
- Hao Phung
- Dimitris Metaxas (ORCID: https://orcid.org/0000-0001-7142-7640)
- Anh Tran (ORCID: https://orcid.org/0000-0003-0123-9577)
- Quan Dao (ORCID: https://orcid.org/0009-0006-0996-0472)
- Trung Dao
Institutions
- Cornell University (US)
- Viña del Mar University (CL)
- Rutgers Sexual and Reproductive Health and Rights (NL)
Publication Details
- Journal
- Proceedings of the AAAI Conference on Artificial Intelligence
- Published
- 2025-04-11
- DOI
- https://doi.org/10.1609/aaai.v39i3.32269
- Primary Topic
- Innovative Microfluidic and Catalytic Techniques Innovation
- Type
- article
- Field-Weighted Citation Impact
- 0.00