arXiv reaDer
MCDIP-ADMM: DIP ベースの CT 再構成におけるオーバーフィッティングの克服
MCDIP-ADMM: Overcoming Overfitting in DIP-based CT reconstruction
この論文では、コンピュータ断層撮影 (CT) 再構成に対する教師なし学習法の適用について調査します。私たちの研究を動機付けるために、いくつかの既存の事前分布、すなわち、切り捨てガウス事前分布、l_1 事前分布、全変動事前分布、および深層画像事前分布 (DIP) をレビューします。 DIP は、表現能力と視覚的パフォーマンスの点で他の 3 つの以前の製品よりも優れていることがわかりました。ただし、DIP は反復回数が一定のしきい値を超えると、過学習によりパフォーマンスが低下します。この問題に対処するために、我々は、マルチコードディープイメージプライアとプラグアンドプレイの乗算器の代替方向法に基づく新しい方法(MCDIP-ADMM)を提案します。具体的には、MCDIP は複数の潜在コードを利用して、ジェネレーター モデル内の中間層で一連の特徴マップを生成します。これらのマップはトレーニング可能な重みを使用して構成され、事前の完全なイメージを表します。実験結果は、提案された MCDIP-ADMM が既存の 3 つの競合製品と比較して優れたパフォーマンスを示していることを示しています。ガウス ノイズを伴う平行ビーム投影の場合、MCDIP-ADMM は、PSNR に関して、DIP に対して 4.3 dB、ADMM DIP-WTV に対して 1.7 dB、PnP-DIP に対して 1.2 dB の平均改善を達成します。同様に、ポアソン ノイズを伴うファンビーム投影の場合、MCDIP-ADMM は、PSNR の点で、DIP に対して 3.09 dB、ADMM DIP-WTV に対して 1.86 dB、PnP-DIP に対して 0.84 dB の平均改善を達成します。
This paper investigates the application of unsupervised learning methods for computed tomography (CT) reconstruction. To motivate our work, we review several existing priors, namely the truncated Gaussian prior, the l_1 prior, the total variation prior, and the deep image prior (DIP). We find that DIP outperforms the other three priors in terms of representational capability and visual performance. However, the performance of DIP deteriorates when the number of iterations exceeds a certain threshold due to overfitting. To address this issue, we propose a novel method (MCDIP-ADMM) based on Multi-Code Deep Image Prior and plug-and-play Alternative Direction Method of Multipliers. Specifically, MCDIP utilizes multiple latent codes to generate a series of feature maps at an intermediate layer within a generator model. These maps are then composed with trainable weights, representing the complete image prior. Experimental results demonstrate the superior performance of the proposed MCDIP-ADMM compared to three existing competitors. In the case of parallel beam projection with Gaussian noise, MCDIP-ADMM achieves an average improvement of 4.3 dB over DIP, 1.7 dB over ADMM DIP-WTV, and 1.2 dB over PnP-DIP in terms of PSNR. Similarly, for fan-beam projection with Poisson noise, MCDIP-ADMM achieves an average improvement of 3.09 dB over DIP, 1.86 dB over ADMM DIP-WTV, and 0.84 dB over PnP-DIP in terms of PSNR.
updated: Thu Jun 01 2023 09:14:20 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アソシエイト