arXiv reaDer
メモリ効率の高い高解像度3DGANのための階層的償却トレーニング
Hierarchical Amortized Training for Memory-efficient High Resolution 3D GAN
Generative Adversarial Networks(GAN)には、データ拡張、ドメイン適応、モデルの説明など、多くの潜在的な医用画像アプリケーションがあります。グラフィックプロセッシングユニット(GPU)のメモリが限られているため、現在のほとんどの3D GANモデルは低解像度の医用画像でトレーニングされており、これらのモデルは高解像度にスケーリングできないか、パッチ状のアーティファクトが発生しやすくなっています。この作業では、高解像度の3D画像を生成できる新しいエンドツーエンドのGANアーキテクチャを提案します。トレーニングと推論を分離することで、この目標を達成します。トレーニング中に、低解像度バージョンの画像とランダムに選択された高解像度画像のサブボリュームを同時に生成する階層構造を採用します。階層設計には2つの利点があります。1つは、高解像度画像のトレーニングに必要なメモリがサブボリューム間で償却されることです。さらに、高解像度のサブボリュームを単一の低解像度の画像に固定することで、サブボリューム間の解剖学的な一貫性が保証されます。推論中に、私たちのモデルは完全な高解像度画像を直接生成できます。また、同様の階層構造を持つエンコーダーをモデルに組み込んで、画像から特徴を抽出します。 3D胸部CTと脳MRIの実験は、私たちのアプローチが画像生成の最先端を上回っていることを示しています。また、データ拡張、画像の超解像、および臨床関連の特徴抽出における提案されたモデルの臨床応用を示します。
Generative Adversarial Networks (GAN) have many potential medical imaging applications, including data augmentation, domain adaptation, and model explanation. Due to the limited memory of Graphical Processing Units (GPUs), most current 3D GAN models are trained on low-resolution medical images, these models either cannot scale to high-resolution or are prone to patchy artifacts. In this work, we propose a novel end-to-end GAN architecture that can generate high-resolution 3D images. We achieve this goal by separating training and inference. During training, we adopt a hierarchical structure that simultaneously generates a low-resolution version of the image and a randomly selected sub-volume of the high-resolution image. The hierarchical design has two advantages: First, the memory demand for training on high-resolution images is amortized among subvolumes. Furthermore, anchoring the high-resolution subvolumes to a single low-resolution image ensures anatomical consistency between subvolumes. During inference, our model can directly generate full high-resolution images. We also incorporate an encoder with a similar hierarchical structure into the model to extract features from the images. Experiments on 3D thorax CT and brain MRI demonstrate that our approach outperforms state of the art in image generation. We also demonstrate clinical applications of the proposed model in data augmentation, image super-resolution and clinical-relevant feature extraction.
updated: Tue Aug 10 2021 20:27:09 GMT+0000 (UTC)
published: Wed Aug 05 2020 02:33:04 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アソシエイト