プラグアンドプレイ(PnP)は、ADMMまたは他の近接アルゴリズムと高度なノイズ除去機能を組み合わせた非凸型フレームワークです。最近、PnPは、特に深層学習ベースのノイズ除去装置の統合により、経験的に大きな成功を収めました。ただし、PnPベースのアプローチの重要な問題は、手動のパラメーター調整が必要なことです。イメージング条件とさまざまなシーンコンテンツの点で、高い不一致を越えて高品質の結果を取得する必要があります。この作業では、ペナルティパラメータ、ノイズ除去強度、および終端時間を含む内部パラメータを自動的に決定できる、チューニングのないPnP近位アルゴリズムを提示します。私たちのアプローチの重要な部分は、パラメータの自動検索のためのポリシーネットワークを開発することです。これは、モデルフリーとモデルベースの深層強化学習を組み合わせて効果的に学習できます。数値的および視覚的な実験を通じて、学習したポリシーがさまざまな状態のさまざまなパラメーターをカスタマイズでき、多くの場合、既存の手作りの基準よりも効率的かつ効果的であることを実証します。さらに、プラグ付きノイズ除去装置の実用的な考慮事項について説明します。これは、学習したポリシーとともに最新の結果をもたらします。これは、線形および非線形の例示的な逆イメージング問題の両方で一般的であり、特に、圧縮センシングMRIおよび位相回復に関する有望な結果を示しています。
Plug-and-play (PnP) is a non-convex framework that combines ADMM or other proximal algorithms with advanced denoiser priors. Recently, PnP has achieved great empirical success, especially with the integration of deep learning-based denoisers. However, a key problem of PnP based approaches is that they require manual parameter tweaking. It is necessary to obtain high-quality results across the high discrepancy in terms of imaging conditions and varying scene content. In this work, we present a tuning-free PnP proximal algorithm, which can automatically determine the internal parameters including the penalty parameter, the denoising strength and the terminal time. A key part of our approach is to develop a policy network for automatic search of parameters, which can be effectively learned via mixed model-free and model-based deep reinforcement learning. We demonstrate, through numerical and visual experiments, that the learned policy can customize different parameters for different states, and often more efficient and effective than existing handcrafted criteria. Moreover, we discuss the practical considerations of the plugged denoisers, which together with our learned policy yield state-of-the-art results. This is prevalent on both linear and nonlinear exemplary inverse imaging problems, and in particular, we show promising results on Compressed Sensing MRI and phase retrieval.