Phased Consistency Models

Fu-Yun Wang, Zhaoyang Huang, Alexander William Bergman, Dazhong Shen, Peng Gao +7 more
5/28/2024
cs.LGcs.CV

Official code: G-U-N/Phased-Consistency-Model (519 stars, Apache-2.0, last commit 2024-12-11).

Abstract

Consistency Models (CMs) have made significant progress in accelerating the generation of diffusion models. However, their application to high-resolution, text-conditioned image generation in the latent space remains unsatisfactory. In this paper, we identify three key flaws in the current design of Latent Consistency Models (LCMs). We investigate the reasons behind these limitations and propose Phased Consistency Models (PCMs), which generalize the design space and address the identified limitations. Our evaluations demonstrate that PCMs outperform LCMs across 1--16 step generation settings. While PCMs are specifically designed for multi-step refinement, they achieve comparable 1-step generation results to previously state-of-the-art specifically designed 1-step methods. Furthermore, we show the methodology of PCMs is versatile and applicable to video generation, enabling us to train the state-of-the-art few-step text-to-video generator. Our code is available at https://github.com/G-U-N/Phased-Consistency-Model.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

[NeurIPS 2024] Boosting the performance of consistency models with PCM!

51919Shell, PythonMay 28, 20241 years agoApache-2.0
consistency-modelsdiffusion-models

Similar Papers

Cite this paper

@article{wang2024phased,
  title  = {Phased Consistency Models},
  author = {Fu-Yun Wang and Zhaoyang Huang and Alexander William Bergman and Dazhong Shen and Peng Gao and Michael Lingelbach and Keqiang Sun and Weikang Bian and Guanglu Song and Yu Liu and Xiaogang Wang and Hongsheng Li},
  year   = {2024},
  eprint = {2405.18407},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.18407v2},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.18407.svg)](https://codeofpaper.com/paper/2405.18407)

Discussion