arXiv reaDer
プラグアンドプレイ画像復元のためのノイズ除去拡散モデル
Denoising Diffusion Models for Plug-and-Play Image Restoration
プラグアンドプレイ画像復元 (IR) は、既製のデノイザーを暗黙的な画像事前処理として利用することで、さまざまな逆問題を解決するための柔軟で解釈可能な方法として広く認識されています。ただし、既存の手法のほとんどは、識別ガウス デノイザーに焦点を当てています。拡散モデルは高品質の画像合成において優れたパフォーマンスを示していますが、プラグアンドプレイ IR 手法に先立って生成ノイズ除去装置として機能する可能性については、さらに検討する必要があります。画像復元に拡散モデルを採用する試みは他にもいくつかありますが、満足のいく結果が得られないか、推論中に許容できない数の神経機能評価 (NFE) が必要になるのが一般的です。この論文では、従来のプラグアンドプレイ手法を拡散サンプリング フレームワークに統合する DiffPIR を提案します。識別ガウス デノイザーに依存するプラグ アンド プレイ IR 手法と比較して、DiffPIR は拡散モデルの生成能力を継承すると予想されます。超解像度、画像のブレ除去、修復を含む 3 つの代表的な IR タスクの実験結果は、DiffPIR が FFHQ データセットと ImageNet データセットの両方で、再構築の忠実性と知覚品質の点で最先端のパフォーマンスを達成できることを示しています。 100 NFE。ソース コードは https://github.com/yuanzhi-zhu/DiffPIR で入手できます。
Plug-and-play Image Restoration (IR) has been widely recognized as a flexible and interpretable method for solving various inverse problems by utilizing any off-the-shelf denoiser as the implicit image prior. However, most existing methods focus on discriminative Gaussian denoisers. Although diffusion models have shown impressive performance for high-quality image synthesis, their potential to serve as a generative denoiser prior to the plug-and-play IR methods remains to be further explored. While several other attempts have been made to adopt diffusion models for image restoration, they either fail to achieve satisfactory results or typically require an unacceptable number of Neural Function Evaluations (NFEs) during inference. This paper proposes DiffPIR, which integrates the traditional plug-and-play method into the diffusion sampling framework. Compared to plug-and-play IR methods that rely on discriminative Gaussian denoisers, DiffPIR is expected to inherit the generative ability of diffusion models. Experimental results on three representative IR tasks, including super-resolution, image deblurring, and inpainting, demonstrate that DiffPIR achieves state-of-the-art performance on both the FFHQ and ImageNet datasets in terms of reconstruction faithfulness and perceptual quality with no more than 100 NFEs. The source code is available at https://github.com/yuanzhi-zhu/DiffPIR
updated: Mon May 15 2023 20:24:38 GMT+0000 (UTC)
published: Mon May 15 2023 20:24:38 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アソシエイト