arXiv reaDer
REPNP:堅牢な画像復元のための事前の深い強化学習によるプラグアンドプレイ
REPNP: Plug-and-Play with Deep Reinforcement Learning Prior for Robust Image Restoration
事前にトレーニングされたディープモデルに基づく画像復元スキームは、さまざまな逆問題を解決するための独自の柔軟性により、大きな注目を集めています。特に、プラグアンドプレイ(PnP)フレームワークは、さまざまな画像復元タスク用の既製のディープデノイザーを既知の観測モデルと統合できる、人気のある強力なツールです。ただし、実際の観測モデルと完全に一致する観測モデルを取得することは、実際には困難な場合があります。したがって、従来のディープデノイザーを使用したPnPスキームでは、実際の画像復元タスクで満足のいく結果が得られない場合があります。 PnPフレームワークの堅牢性は、決定論的最適化によってトレーニングされた既製のディープデノイザーを使用することによって大幅に制限されると主張します。この目的のために、堅牢な画像復元タスクに軽量のDRLベースのデノイザーを活用することにより、RePNPと呼ばれる新しい深層強化学習(DRL)ベースのPnPフレームワークを提案します。実験結果は、提案されたRePNPが、実際のモデルから逸脱したPnPスキームで使用される観測モデルに対してロバストであることを示しています。したがって、RePNPは、画像のぼけ除去および超解像タスクに対して、より信頼性の高い復元結果を生成できます。いくつかの最先端の深部画像復元ベースラインと比較して、RePNPは、より少ないモデルパラメータでモデル偏差に主観的なより良い結果を達成します。
Image restoration schemes based on the pre-trained deep models have received great attention due to their unique flexibility for solving various inverse problems. In particular, the Plug-and-Play (PnP) framework is a popular and powerful tool that can integrate an off-the-shelf deep denoiser for different image restoration tasks with known observation models. However, obtaining the observation model that exactly matches the actual one can be challenging in practice. Thus, the PnP schemes with conventional deep denoisers may fail to generate satisfying results in some real-world image restoration tasks. We argue that the robustness of the PnP framework is largely limited by using the off-the-shelf deep denoisers that are trained by deterministic optimization. To this end, we propose a novel deep reinforcement learning (DRL) based PnP framework, dubbed RePNP, by leveraging a light-weight DRL-based denoiser for robust image restoration tasks. Experimental results demonstrate that the proposed RePNP is robust to the observation model used in the PnP scheme deviating from the actual one. Thus, RePNP can generate more reliable restoration results for image deblurring and super resolution tasks. Compared with several state-of-the-art deep image restoration baselines, RePNP achieves better results subjective to model deviation with fewer model parameters.
updated: Mon Jul 25 2022 10:56:10 GMT+0000 (UTC)
published: Mon Jul 25 2022 10:56:10 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アソシエイト