arXiv reaDer
DriftRec: 拡散モデルをブラインド JPEG 復元に適応させる
DriftRec: Adapting diffusion models to blind JPEG restoration
この作業では、拡散モデルの高忠実度生成機能を利用して、高圧縮レベルでのブラインド JPEG 復元を解決します。拡散モデルの順確率微分方程式をこの復元タスクに適応させるための洗練された修正を提案し、メソッドを DriftRec と名付けます。同じネットワーク アーキテクチャと JPEG 復元用の 2 つの最先端技術を使用して、DriftRec を L_2 回帰ベースラインと比較することで、私たちのアプローチがぼやけた画像を生成する他の方法の傾向を回避し、きれいな画像の分布を回復できることを示します。かなり忠実に。このためには、クリーン/破損イメージペアのデータセットのみが必要であり、破損操作に関する知識は必要ないため、他の復元タスクへの幅広い適用が可能になります。他の条件付きおよび無条件の拡散モデルとは対照的に、きれいな画像と破損した画像の分布は、拡散モデルの逆プロセスの前の通常のガウス分布よりも互いにはるかに近いという考えを利用します。したがって、私たちのアプローチでは、低レベルの追加ノイズのみが必要であり、さらなる最適化を行わなくても、比較的少数のサンプリングステップしか必要としません。 DriftRec は、トレーニング中にそのような例に遭遇することなく、オンラインで見つかった位置合わせされていない二重 JPEG 圧縮や JPEG のブラインド復元などの現実的で困難なシナリオに自然に一般化することを示します。
In this work, we utilize the high-fidelity generation abilities of diffusion models to solve blind JPEG restoration at high compression levels. We propose an elegant modification of the forward stochastic differential equation of diffusion models to adapt them to this restoration task and name our method DriftRec. Comparing DriftRec against an L_2 regression baseline with the same network architecture and two state-of-the-art techniques for JPEG restoration, we show that our approach can escape the tendency of other methods to generate blurry images, and recovers the distribution of clean images significantly more faithfully. For this, only a dataset of clean/corrupted image pairs and no knowledge about the corruption operation is required, enabling wider applicability to other restoration tasks. In contrast to other conditional and unconditional diffusion models, we utilize the idea that the distributions of clean and corrupted images are much closer to each other than each is to the usual Gaussian prior of the reverse process in diffusion models. Our approach therefore requires only low levels of added noise, and needs comparatively few sampling steps even without further optimizations. We show that DriftRec naturally generalizes to realistic and difficult scenarios such as unaligned double JPEG compression and blind restoration of JPEGs found online, without having encountered such examples during training.
updated: Mon Jul 03 2023 18:24:44 GMT+0000 (UTC)
published: Sat Nov 12 2022 22:29:42 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アソシエイト