arXiv reaDer
教師なし低線量CTノイズ除去用の可逆ジェネレーターを使用したサイクルフリーのCycleGAN
Cycle-free CycleGAN using Invertible Generator for Unsupervised Low-Dose CT Denoising
最近、CycleGANは、ペアのトレーニングデータセットを必要とせずに、低線量X線コンピューター断層撮影(CT)に高性能で超高速のノイズ除去を提供することが示されました。これはサイクルの一貫性のおかげで可能でしたが、CycleGANはサイクルの一貫性を強制するために2つのジェネレーターと2つのディスクリミネーターを必要とし、トレーニングのためにかなりのGPUリソースと技術スキルを要求します。アダプティブインスタンスノーマライゼーション(AdaIN)を使用した調整可能なCycleGANの最近の提案は、単一のジェネレーターを使用することで問題を部分的に軽減します。ただし、トレーニングには2つのディスクリミネーターと追加のAdaINコードジェネレーターが必要です。この問題を解決するために、ここでは、単一のジェネレーターとディスクリミネーターで構成され、サイクルの一貫性を保証する、新しいサイクルフリーのCycle-GANアーキテクチャを紹介します。主な革新は、可逆ジェネレーターの使用がサイクルの一貫性条件を自動的に満たし、CycleGAN定式化の追加の弁別子を排除するという観察から来ています。可逆ジェネレータをより効果的にするために、私たちのネットワークはウェーブレット残差領域に実装されています。さまざまなレベルの低線量CT画像を使用した広範な実験により、従来のCycleGANと比較して、学習可能なパラメーターのわずか10%とより速いトレーニング時間を使用して、私たちの方法がノイズ除去パフォーマンスを大幅に改善できることが確認されました。
Recently, CycleGAN was shown to provide high-performance, ultra-fast denoising for low-dose X-ray computed tomography (CT) without the need for a paired training dataset. Although this was possible thanks to cycle consistency, CycleGAN requires two generators and two discriminators to enforce cycle consistency, demanding significant GPU resources and technical skills for training. A recent proposal of tunable CycleGAN with Adaptive Instance Normalization (AdaIN) alleviates the problem in part by using a single generator. However, two discriminators and an additional AdaIN code generator are still required for training. To solve this problem, here we present a novel cycle-free Cycle-GAN architecture, which consists of a single generator and a discriminator but still guarantees cycle consistency. The main innovation comes from the observation that the use of an invertible generator automatically fulfills the cycle consistency condition and eliminates the additional discriminator in the CycleGAN formulation. To make the invertible generator more effective, our network is implemented in the wavelet residual domain. Extensive experiments using various levels of low-dose CT images confirm that our method can significantly improve denoising performance using only 10% of learnable parameters and faster training time compared to the conventional CycleGAN.
updated: Sat Apr 17 2021 13:23:36 GMT+0000 (UTC)
published: Sat Apr 17 2021 13:23:36 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アソシエイト