arXiv reaDer
ノイズの多い線形逆問題のための拡散モデルベースの事後サンプリング
Diffusion Model Based Posterior Sampling for Noisy Linear Inverse Problems
加法的ガウス ノイズを伴うユビキタスな線形逆問題を検討し、拡散モデル ベースの事後サンプリング (DMPS) と呼ばれる教師なしの汎用サンプリング アプローチを提案して、ノイズの多い線形測定値から未知の信号を再構築します。具体的には、未知の信号の事前確率は、事前にトレーニングされた 1 つの拡散モデル (DM) によって暗黙的にモデル化されます。事後サンプリングでは、正確なノイズ摂動尤度スコアの難しさに対処するために、単純で効果的なノイズ摂動疑似尤度スコアが、有益でない事前仮定の下で導入されます。 DMPS は適切な修正を加えることであらゆる種類の DM に適用されますが、特定の例としてアブレーション拡散モデル (ADM) に焦点を当て、画像の超解像、ノイズ除去、ブレ除去、カラー化などのさまざまな線形逆問題に対するその有効性を評価します。実験結果は、分布内サンプルと分布外サンプルの両方で、DMPS が主要な競合他社よりも 3 倍高速でありながら、さまざまなタスクで非常に競争力のある、またはそれ以上のパフォーマンスを達成することを示しています。結果を再現するコードは、https://github.com/mengxiangming/dmps で入手できます。
We consider the ubiquitous linear inverse problems with additive Gaussian noise and propose an unsupervised general-purpose sampling approach called diffusion model based posterior sampling (DMPS) to reconstruct the unknown signal from noisy linear measurements. Specifically, the prior of the unknown signal is implicitly modeled by one pre-trained diffusion model (DM). In posterior sampling, to address the intractability of exact noise-perturbed likelihood score, a simple yet effective noise-perturbed pseudo-likelihood score is introduced under the uninformative prior assumption. While DMPS applies to any kind of DM with proper modifications, we focus on the ablated diffusion model (ADM) as one specific example and evaluate its efficacy on a variety of linear inverse problems such as image super-resolution, denoising, deblurring, colorization. Experimental results demonstrate that, for both in-distribution and out-of-distribution samples, DMPS achieves highly competitive or even better performances on various tasks while being 3 times faster than the leading competitor. The code to reproduce the results is available at https://github.com/mengxiangming/dmps.
updated: Sun Nov 20 2022 01:09:49 GMT+0000 (UTC)
published: Sun Nov 20 2022 01:09:49 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト