arXiv reaDer
平均反転確率微分方程式による画像復元
Image Restoration with Mean-Reverting Stochastic Differential Equations
この論文では、汎用画像復元のための確率微分方程式 (SDE) アプローチを紹介します。主要な構造は、高品質画像を固定ガウス ノイズを持つ平均状態として劣化した画像に変換する平均値反転 SDE で構成されます。次に、対応する逆時間 SDE をシミュレートすることで、タスク固有の事前知識に依存せずに、低品質画像の原点を復元できます。重要なのは、提案された平均値反転 SDE には閉形式の解法があり、グラウンド トゥルースの時間依存スコアを計算し、ニューラル ネットワークで学習できることです。さらに、トレーニングを安定させ、復元結果を向上させる最適な逆軌道を学習するための最尤目標を提案します。実験では、私たちが提案した方法が、画像のディレイニング、ブレ除去、ノイズ除去の定量的比較において非常に競争力のあるパフォーマンスを達成し、2 つのディレイニング データセットで最先端の技術を確立したことが示されています。最後に、画像の超解像度、修復、かすみ除去に関する定性的な結果を通じて、私たちのアプローチの一般的な適用可能性をさらに実証します。コードは https://github.com/Algolzw/image-restoration-sde で入手できます。
This paper presents a stochastic differential equation (SDE) approach for general-purpose image restoration. The key construction consists in a mean-reverting SDE that transforms a high-quality image into a degraded counterpart as a mean state with fixed Gaussian noise. Then, by simulating the corresponding reverse-time SDE, we are able to restore the origin of the low-quality image without relying on any task-specific prior knowledge. Crucially, the proposed mean-reverting SDE has a closed-form solution, allowing us to compute the ground truth time-dependent score and learn it with a neural network. Moreover, we propose a maximum likelihood objective to learn an optimal reverse trajectory that stabilizes the training and improves the restoration results. The experiments show that our proposed method achieves highly competitive performance in quantitative comparisons on image deraining, deblurring, and denoising, setting a new state-of-the-art on two deraining datasets. Finally, the general applicability of our approach is further demonstrated via qualitative results on image super-resolution, inpainting, and dehazing. Code is available at https://github.com/Algolzw/image-restoration-sde.
updated: Wed May 31 2023 12:41:58 GMT+0000 (UTC)
published: Fri Jan 27 2023 13:20:48 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アソシエイト