Wang Chengjia, Yang Guang, Papanastasiou Giorgos, Tsaftaris Sotirios A, Newby David E, Gray Calum, Macnaught Gillian, MacGillivray Tom J
BHF Centre for Cardiovascular Science, University of Edinburgh, Edinburgh, UK.
National Heart and Lung Institute, Imperial College London, London, UK.
Inf Fusion. 2021 Mar;67:147-160. doi: 10.1016/j.inffus.2020.10.015.
Cycle-consistent generative adversarial network (CycleGAN) has been widely used for cross-domain medical image synthesis tasks particularly due to its ability to deal with unpaired data. However, most CycleGAN-based synthesis methods cannot achieve good alignment between the synthesized images and data from the source domain, even with additional image alignment losses. This is because the CycleGAN generator network can encode the relative deformations and noises associated to different domains. This can be detrimental for the downstream applications that rely on the synthesized images, such as generating pseudo-CT for PET-MR attenuation correction. In this paper, we present a deformation invariant cycle-consistency model that can filter out these domain-specific deformation. The deformation is globally parameterized by thin-plate-spline (TPS), and locally learned by modified deformable convolutional layers. Robustness to domain-specific deformations has been evaluated through experiments on multi-sequence brain MR data and multi-modality abdominal CT and MR data. Experiment results demonstrated that our method can achieve better alignment between the source and target data while maintaining superior image quality of signal compared to several state-of-the-art CycleGAN-based methods.
循环一致生成对抗网络(CycleGAN)已被广泛用于跨域医学图像合成任务,特别是由于其处理未配对数据的能力。然而,大多数基于CycleGAN的合成方法即使添加了额外的图像对齐损失,也无法在合成图像和源域数据之间实现良好的对齐。这是因为CycleGAN生成器网络可以对与不同域相关的相对变形和噪声进行编码。这对于依赖合成图像的下游应用可能是有害的,例如为PET-MR衰减校正生成伪CT。在本文中,我们提出了一种变形不变循环一致性模型,该模型可以滤除这些特定于域的变形。变形通过薄板样条(TPS)进行全局参数化,并通过修改后的可变形卷积层进行局部学习。通过对多序列脑MR数据以及多模态腹部CT和MR数据进行实验,评估了对特定于域的变形的鲁棒性。实验结果表明,与几种基于CycleGAN的最新方法相比,我们的方法可以在源数据和目标数据之间实现更好的对齐,同时保持信号的卓越图像质量。