MAP-based Problem-Agnostic diffusion model for Inverse Problems

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tao, Pingping, Liu, Haixia, Su, Jing
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917323166711808
author Tao, Pingping
Liu, Haixia
Su, Jing
author_facet Tao, Pingping
Liu, Haixia
Su, Jing
contents Diffusion models have indeed shown great promise in solving inverse problems in image processing. In this paper, we propose a novel, problem-agnostic diffusion model called the maximum a posteriori (MAP)-based guided term estimation method for inverse problems. To leverage unconditionally pretrained diffusion models to address conditional generation tasks, we divide the conditional score function into two terms according to Bayes' rule: an unconditional score function (approximated by a pretrained score network) and a guided term, which is estimated using a novel MAP-based method that incorporates a Gaussian-type prior of natural images. This innovation allows us to better capture the intrinsic properties of the data, leading to improved performance. Numerical results demonstrate that our method preserves contents more effectively compared to state-of-the-art methods--for example, maintaining the structure of glasses in super-resolution tasks and producing more coherent results in the neighborhood of masked regions during inpainting.
format Preprint
id arxiv_https___arxiv_org_abs_2501_15128
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle MAP-based Problem-Agnostic diffusion model for Inverse Problems
Tao, Pingping
Liu, Haixia
Su, Jing
Image and Video Processing
Computer Vision and Pattern Recognition
Diffusion models have indeed shown great promise in solving inverse problems in image processing. In this paper, we propose a novel, problem-agnostic diffusion model called the maximum a posteriori (MAP)-based guided term estimation method for inverse problems. To leverage unconditionally pretrained diffusion models to address conditional generation tasks, we divide the conditional score function into two terms according to Bayes' rule: an unconditional score function (approximated by a pretrained score network) and a guided term, which is estimated using a novel MAP-based method that incorporates a Gaussian-type prior of natural images. This innovation allows us to better capture the intrinsic properties of the data, leading to improved performance. Numerical results demonstrate that our method preserves contents more effectively compared to state-of-the-art methods--for example, maintaining the structure of glasses in super-resolution tasks and producing more coherent results in the neighborhood of masked regions during inpainting.
title MAP-based Problem-Agnostic diffusion model for Inverse Problems
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2501.15128