Removing Structured Noise with Diffusion Models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Stevens, Tristan S. W., van Gorp, Hans, Meral, Faik C., Shin, Junseob, Yu, Jason, Robert, Jean-Luc, van Sloun, Ruud J. G.
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915442439749632
author Stevens, Tristan S. W.
van Gorp, Hans
Meral, Faik C.
Shin, Junseob
Yu, Jason
Robert, Jean-Luc
van Sloun, Ruud J. G.
author_facet Stevens, Tristan S. W.
van Gorp, Hans
Meral, Faik C.
Shin, Junseob
Yu, Jason
Robert, Jean-Luc
van Sloun, Ruud J. G.
contents Solving ill-posed inverse problems requires careful formulation of prior beliefs over the signals of interest and an accurate description of their manifestation into noisy measurements. Handcrafted signal priors based on e.g. sparsity are increasingly replaced by data-driven deep generative models, and several groups have recently shown that state-of-the-art score-based diffusion models yield particularly strong performance and flexibility. In this paper, we show that the powerful paradigm of posterior sampling with diffusion models can be extended to include rich, structured, noise models. To that end, we propose a joint conditional reverse diffusion process with learned scores for the noise and signal-generating distribution. We demonstrate strong performance gains across various inverse problems with structured noise, outperforming competitive baselines that use normalizing flows and adversarial networks. This opens up new opportunities and relevant practical applications of diffusion modeling for inverse problems in the context of non-Gaussian measurement models.
format Preprint
id arxiv_https___arxiv_org_abs_2302_05290
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Removing Structured Noise with Diffusion Models
Stevens, Tristan S. W.
van Gorp, Hans
Meral, Faik C.
Shin, Junseob
Yu, Jason
Robert, Jean-Luc
van Sloun, Ruud J. G.
Machine Learning
Image and Video Processing
Signal Processing
Solving ill-posed inverse problems requires careful formulation of prior beliefs over the signals of interest and an accurate description of their manifestation into noisy measurements. Handcrafted signal priors based on e.g. sparsity are increasingly replaced by data-driven deep generative models, and several groups have recently shown that state-of-the-art score-based diffusion models yield particularly strong performance and flexibility. In this paper, we show that the powerful paradigm of posterior sampling with diffusion models can be extended to include rich, structured, noise models. To that end, we propose a joint conditional reverse diffusion process with learned scores for the noise and signal-generating distribution. We demonstrate strong performance gains across various inverse problems with structured noise, outperforming competitive baselines that use normalizing flows and adversarial networks. This opens up new opportunities and relevant practical applications of diffusion modeling for inverse problems in the context of non-Gaussian measurement models.
title Removing Structured Noise with Diffusion Models
topic Machine Learning
Image and Video Processing
Signal Processing
url https://arxiv.org/abs/2302.05290