Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Wasserstein-based Projections with Applications to Inverse Problems
(Submitted on 5 Aug 2020 (v1), last revised 14 Apr 2021 (this version, v3))
Abstract: Inverse problems consist of recovering a signal from a collection of noisy measurements. These are typically cast as optimization problems, with classic approaches using a data fidelity term and an analytic regularizer that stabilizes recovery. Recent Plug-and-Play (PnP) works propose replacing the operator for analytic regularization in optimization methods by a data-driven denoiser. These schemes obtain state of the art results, but at the cost of limited theoretical guarantees. To bridge this gap, we present a new algorithm that takes samples from the manifold of true data as input and outputs an approximation of the projection operator onto this manifold. Under standard assumptions, we prove this algorithm generates a learned operator, called Wasserstein-based projection (WP), that approximates the true projection with high probability. Thus, WPs can be inserted into optimization methods in the same manner as PnP, but now with theoretical guarantees. Provided numerical examples show WPs obtain state of the art results for unsupervised PnP signal recovery.
Submission history
From: Howard Heaton [view email][v1] Wed, 5 Aug 2020 15:58:55 GMT (3450kb,D)
[v2] Fri, 16 Oct 2020 17:29:35 GMT (3343kb,D)
[v3] Wed, 14 Apr 2021 16:25:43 GMT (4280kb,D)
Link back to: arXiv, form interface, contact.