Minutes to Seconds: Speeded-up DDPM-based Image Inpainting with Coarse-to-Fine Sampling

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Lintao, Du, Xiangcheng, TomyEnrique, LeoWu, Wang, Yiqun, Zheng, Yingbin, Jin, Cheng
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866907971665002496
author Zhang, Lintao
Du, Xiangcheng
TomyEnrique, LeoWu
Wang, Yiqun
Zheng, Yingbin
Jin, Cheng
author_facet Zhang, Lintao
Du, Xiangcheng
TomyEnrique, LeoWu
Wang, Yiqun
Zheng, Yingbin
Jin, Cheng
contents For image inpainting, the existing Denoising Diffusion Probabilistic Model (DDPM) based method i.e. RePaint can produce high-quality images for any inpainting form. It utilizes a pre-trained DDPM as a prior and generates inpainting results by conditioning on the reverse diffusion process, namely denoising process. However, this process is significantly time-consuming. In this paper, we propose an efficient DDPM-based image inpainting method which includes three speed-up strategies. First, we utilize a pre-trained Light-Weight Diffusion Model (LWDM) to reduce the number of parameters. Second, we introduce a skip-step sampling scheme of Denoising Diffusion Implicit Models (DDIM) for the denoising process. Finally, we propose Coarse-to-Fine Sampling (CFS), which speeds up inference by reducing image resolution in the coarse stage and decreasing denoising timesteps in the refinement stage. We conduct extensive experiments on both faces and general-purpose image inpainting tasks, and our method achieves competitive performance with approximately 60 times speedup.
format Preprint
id arxiv_https___arxiv_org_abs_2407_05875
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Minutes to Seconds: Speeded-up DDPM-based Image Inpainting with Coarse-to-Fine Sampling
Zhang, Lintao
Du, Xiangcheng
TomyEnrique, LeoWu
Wang, Yiqun
Zheng, Yingbin
Jin, Cheng
Computer Vision and Pattern Recognition
For image inpainting, the existing Denoising Diffusion Probabilistic Model (DDPM) based method i.e. RePaint can produce high-quality images for any inpainting form. It utilizes a pre-trained DDPM as a prior and generates inpainting results by conditioning on the reverse diffusion process, namely denoising process. However, this process is significantly time-consuming. In this paper, we propose an efficient DDPM-based image inpainting method which includes three speed-up strategies. First, we utilize a pre-trained Light-Weight Diffusion Model (LWDM) to reduce the number of parameters. Second, we introduce a skip-step sampling scheme of Denoising Diffusion Implicit Models (DDIM) for the denoising process. Finally, we propose Coarse-to-Fine Sampling (CFS), which speeds up inference by reducing image resolution in the coarse stage and decreasing denoising timesteps in the refinement stage. We conduct extensive experiments on both faces and general-purpose image inpainting tasks, and our method achieves competitive performance with approximately 60 times speedup.
title Minutes to Seconds: Speeded-up DDPM-based Image Inpainting with Coarse-to-Fine Sampling
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.05875