Iteratively Refined Image Reconstruction with Learned Attentive Regularizers

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Pourya, Mehrsa, Neumayer, Sebastian, Unser, Michael
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910519723556864
author Pourya, Mehrsa
Neumayer, Sebastian
Unser, Michael
author_facet Pourya, Mehrsa
Neumayer, Sebastian
Unser, Michael
contents We propose a regularization scheme for image reconstruction that leverages the power of deep learning while hinging on classic sparsity-promoting models. Many deep-learning-based models are hard to interpret and cumbersome to analyze theoretically. In contrast, our scheme is interpretable because it corresponds to the minimization of a series of convex problems. For each problem in the series, a mask is generated based on the previous solution to refine the regularization strength spatially. In this way, the model becomes progressively attentive to the image structure. For the underlying update operator, we prove the existence of a fixed point. As a special case, we investigate a mask generator for which the fixed-point iterations converge to a critical point of an explicit energy functional. In our experiments, we match the performance of state-of-the-art learned variational models for the solution of inverse problems. Additionally, we offer a promising balance between interpretability, theoretical guarantees, reliability, and performance.
format Preprint
id arxiv_https___arxiv_org_abs_2407_06608
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Iteratively Refined Image Reconstruction with Learned Attentive Regularizers
Pourya, Mehrsa
Neumayer, Sebastian
Unser, Michael
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
We propose a regularization scheme for image reconstruction that leverages the power of deep learning while hinging on classic sparsity-promoting models. Many deep-learning-based models are hard to interpret and cumbersome to analyze theoretically. In contrast, our scheme is interpretable because it corresponds to the minimization of a series of convex problems. For each problem in the series, a mask is generated based on the previous solution to refine the regularization strength spatially. In this way, the model becomes progressively attentive to the image structure. For the underlying update operator, we prove the existence of a fixed point. As a special case, we investigate a mask generator for which the fixed-point iterations converge to a critical point of an explicit energy functional. In our experiments, we match the performance of state-of-the-art learned variational models for the solution of inverse problems. Additionally, we offer a promising balance between interpretability, theoretical guarantees, reliability, and performance.
title Iteratively Refined Image Reconstruction with Learned Attentive Regularizers
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2407.06608