Inverse problem regularization with hierarchical variational autoencoders

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Prost, Jean, Houdard, Antoine, Almansa, Andrés, Papadakis, Nicolas
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911883980701696
author Prost, Jean
Houdard, Antoine
Almansa, Andrés
Papadakis, Nicolas
author_facet Prost, Jean
Houdard, Antoine
Almansa, Andrés
Papadakis, Nicolas
contents In this paper, we propose to regularize ill-posed inverse problems using a deep hierarchical variational autoencoder (HVAE) as an image prior. The proposed method synthesizes the advantages of i) denoiser-based Plug \& Play approaches and ii) generative model based approaches to inverse problems. First, we exploit VAE properties to design an efficient algorithm that benefits from convergence guarantees of Plug-and-Play (PnP) methods. Second, our approach is not restricted to specialized datasets and the proposed PnP-HVAE model is able to solve image restoration problems on natural images of any size. Our experiments show that the proposed PnP-HVAE method is competitive with both SOTA denoiser-based PnP approaches, and other SOTA restoration methods based on generative models.
format Preprint
id arxiv_https___arxiv_org_abs_2303_11217
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Inverse problem regularization with hierarchical variational autoencoders
Prost, Jean
Houdard, Antoine
Almansa, Andrés
Papadakis, Nicolas
Computer Vision and Pattern Recognition
Machine Learning
In this paper, we propose to regularize ill-posed inverse problems using a deep hierarchical variational autoencoder (HVAE) as an image prior. The proposed method synthesizes the advantages of i) denoiser-based Plug \& Play approaches and ii) generative model based approaches to inverse problems. First, we exploit VAE properties to design an efficient algorithm that benefits from convergence guarantees of Plug-and-Play (PnP) methods. Second, our approach is not restricted to specialized datasets and the proposed PnP-HVAE model is able to solve image restoration problems on natural images of any size. Our experiments show that the proposed PnP-HVAE method is competitive with both SOTA denoiser-based PnP approaches, and other SOTA restoration methods based on generative models.
title Inverse problem regularization with hierarchical variational autoencoders
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2303.11217