arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.01910eess.IVcs.CV
收藏

高分辨率三维GAN的分级分期记忆训练

Hierarchical Amortized Training for Memory-efficient High Resolution 3D GAN

University of Pittsburgh(匹兹堡大学);The University of Melbourne(墨尔本大学)

Li Sun, Junxiang Chen, Yanwu Xu, Mingming Gong, Ke Yu, Kayhan Batmanghelich

摘要:

生成对抗网络(GAN)具有许多潜在的医学成像应用,包括数据增强,域自适应和模型解释。由于图形处理单元(GPU)的内存有限,目前大多数3D GAN模型都是在低分辨率医学图像上训练的,这些模型要么不能扩展到高分辨率,要么容易出现斑块状伪影。在这项工作中,我们提出了一种新的端到端的GAN架构,可以生成高分辨率的3D图像。我们通过在训练和推理之间使用不同的配置来实现这一目标。在训练过程中,我们采用分层结构,同时生成图像的低分辨率版本和随机选择的高分辨率图像的子体积。分层设计有两个优点:首先,在高分辨率图像上训练的存储器需求在子卷之间分摊。此外,将高分辨率子体积锚定到单个低分辨率图像确保子体积之间的解剖一致性。在推理过程中,我们的模型可以直接生成完整的高分辨率图像。我们还将一个编码器与一个类似的分层结构的模型,从图像中提取特征。在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 using different configurations between 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 sub-volumes. Furthermore, anchoring the high-resolution sub-volumes to a single low-resolution image ensures anatomical consistency between sub-volumes. 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 and clinical-relevant feature extraction.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑