Empirical Bayesian image restoration by Langevin sampling with a denoising diffusion implicit prior

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Mbakam, Charlesquin Kemajou, Giovannelli, Jean-Francois, Pereyra, Marcelo
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913492996456448
author Mbakam, Charlesquin Kemajou
Giovannelli, Jean-Francois
Pereyra, Marcelo
author_facet Mbakam, Charlesquin Kemajou
Giovannelli, Jean-Francois
Pereyra, Marcelo
contents Score-based diffusion methods provide a powerful strategy to solve image restoration tasks by flexibly combining a pre-trained foundational prior model with a likelihood function specified during test time. Such methods are predominantly derived from two stochastic processes: reversing Ornstein-Uhlenbeck, which underpins the celebrated denoising diffusion probabilistic models (DDPM) and denoising diffusion implicit models (DDIM), and the Langevin diffusion process. The solutions delivered by DDPM and DDIM are often remarkably realistic, but they are not always consistent with measurements because of likelihood intractability issues and the associated required approximations. Alternatively, using a Langevin process circumvents the intractable likelihood issue, but usually leads to restoration results of inferior quality and longer computing times. This paper presents a novel and highly computationally efficient image restoration method that carefully embeds a foundational DDPM denoiser within an empirical Bayesian Langevin algorithm, which jointly calibrates key model hyper-parameters as it estimates the model's posterior mean. Extensive experimental results on three canonical tasks (image deblurring, super-resolution, and inpainting) demonstrate that the proposed approach improves on state-of-the-art strategies both in image estimation accuracy and computing time.
format Preprint
id arxiv_https___arxiv_org_abs_2409_04384
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Empirical Bayesian image restoration by Langevin sampling with a denoising diffusion implicit prior
Mbakam, Charlesquin Kemajou
Giovannelli, Jean-Francois
Pereyra, Marcelo
Computer Vision and Pattern Recognition
Computation and Language
Image and Video Processing
Score-based diffusion methods provide a powerful strategy to solve image restoration tasks by flexibly combining a pre-trained foundational prior model with a likelihood function specified during test time. Such methods are predominantly derived from two stochastic processes: reversing Ornstein-Uhlenbeck, which underpins the celebrated denoising diffusion probabilistic models (DDPM) and denoising diffusion implicit models (DDIM), and the Langevin diffusion process. The solutions delivered by DDPM and DDIM are often remarkably realistic, but they are not always consistent with measurements because of likelihood intractability issues and the associated required approximations. Alternatively, using a Langevin process circumvents the intractable likelihood issue, but usually leads to restoration results of inferior quality and longer computing times. This paper presents a novel and highly computationally efficient image restoration method that carefully embeds a foundational DDPM denoiser within an empirical Bayesian Langevin algorithm, which jointly calibrates key model hyper-parameters as it estimates the model's posterior mean. Extensive experimental results on three canonical tasks (image deblurring, super-resolution, and inpainting) demonstrate that the proposed approach improves on state-of-the-art strategies both in image estimation accuracy and computing time.
title Empirical Bayesian image restoration by Langevin sampling with a denoising diffusion implicit prior
topic Computer Vision and Pattern Recognition
Computation and Language
Image and Video Processing
url https://arxiv.org/abs/2409.04384