arXiv reaDer
画像のノイズ除去と CT 再構成のためのマルチコード ディープ画像事前ベースのプラグアンドプレイ ADMM
Multi-code deep image prior based plug-and-play ADMM for image denoising and CT reconstruction
逆問題のイメージングにおける事前確率に基づく畳み込みニューラル ネットワークの使用は、ますます一般的になっています。ただし、現在の最先端の方法では深刻なオーバーフィッティングが容易に発生する可能性があるため、オーバーフィッティングの問題を解消するには、多くの早期停止手法が必要になります。私たちの仕事をやる気にさせるために、事前に画像を作成するためのいくつかの既存のアプローチを確認します。ディープ イメージ プライアとハンドクラフト プライアの組み合わせは、解釈可能性と表現可能性の点で優れたパフォーマンスを発揮することがわかりました。マルチコードディープイメージプライア、ディープイメージプライアの複数潜在コードバリアントを提案します。これは、オーバーフィッティングを排除するために利用でき、さまざまな数の潜在コードに対してもロバストです。手作りの事前確率は微分可能でないため、代替方向乗数法 (ADMM) アルゴリズムを使用します。 PnP-DIP、DIP-VBTV、ADMM DIP-WTV メソッドを含む既存の最先端の方法に対して、画像のノイズ除去問題と高度に設定されていない CT 再構成問題に対する提案された方法のパフォーマンスを比較します。 CelebA データセットのノイズ除去では、比較したすべての方法に対して 1.46 dB のピーク S/N 比の改善が得られました。 CT 再構成の場合、3 つのテスト画像の対応する平均改善は、DIP で 4.3 dB、ADMM DIP-WTV で 1.7 dB、PnP-DIP で 1.2 dB であり、構造的類似性指数も大幅に改善されています。
The use of the convolutional neural network based prior in imaging inverse problems has become increasingly popular. Current state-of-the-art methods, however, can easily result in severe overfitting, which makes a number of early stopping techniques necessary to eliminate the overfitting problem. To motivate our work, we review some existing approaches to image priors. We find that the deep image prior in combined with the handcrafted prior has an outstanding performance in terms of interpretability and representability. We propose a multi-code deep image prior, a multiple latent codes variant of the deep image prior, which can be utilized to eliminate overfitting and is also robust to the different numbers of the latent codes. Due to the non-differentiability of the handcrafted prior, we use the alternative direction method of multipliers (ADMM) algorithm. We compare the performance of the proposed method on an image denoising problem and a highly ill-posed CT reconstruction problem against the existing state-of-the-art methods, including PnP-DIP, DIP-VBTV and ADMM DIP-WTV methods. For the CelebA dataset denoising, we obtain 1.46 dB peak signal to noise ratio improvement against all compared methods. For the CT reconstruction, the corresponding average improvement of three test images is 4.3 dB over DIP, and 1.7 dB over ADMM DIP-WTV, and 1.2 dB over PnP-DIP along with a significant improvement in the structural similarity index.
updated: Sat Apr 08 2023 03:10:19 GMT+0000 (UTC)
published: Sat Apr 08 2023 03:10:19 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アソシエイト