Development of a GAN architecture based on integrating global and local information for paired and unpaired medical image translation

被引:14
作者
Amirkolaee, Hamed Amini [1 ]
Bokov, Dmitry Olegovich [2 ]
Sharma, Himanshu [3 ]
机构
[1] Univ Tehran, Coll Engn, Sch Surveying & Geospatial Engn, Tehran, Iran
[2] Sechenov First Moscow State Med Univ, Inst Pharm, Moscow, Russia
[3] GLA Univ Mathura, Dept Comp Engn & Applicat, Mathura, India
关键词
Medical image translation; Paired and unpaired data; Joint attention block; Global guidance block; Edge detection network; MRI-ONLY RADIOTHERAPY; ATTENUATION CORRECTION; PSEUDO-CT; GENERATION; SEGMENTATION; NETWORK;
D O I
10.1016/j.eswa.2022.117421
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The purpose of this study is to develop and evaluate a novel end-to-end generative adversarial network for image-to-image translation of medical paired and unpaired data. The proposed deep framework is based on generative adversarial networks, in which blocks are designed and used to improve the estimated images and achieve more accurate results. A joint attention block is designed to capture and fuse the global and local information during the training process. A dual discriminator is employed to accurately determine whether the generated images are real or fake. A global guidance up-sampling block is proposed to reconstruct the local details of the estimated images with guidance of the low-level information of the shallow layers. Moreover, an edge detection network is utilized to make the tissue edges sharper. The loss function is composed of different criteria considering the geometric characteristics for obtaining more accurate results. Different translation tasks including PD/T2, CT/MR, and PET/CT are considered to evaluate the performance of the proposed deep framework. The evaluation results indicated averagely 28.28, 10.87, 31.30, 0.90, 0.92, 0.85, 24.80, and 0.042 for root-mean-squared error (RMSE), mean absolute error (MAE), peak signal to noise ratio (PSNR), dice similarity coefficient (DSC), normalized cross-correlation (NCC), structural similarity index measure (SSIM), learned perceptual image patch similarity (LPIPS) and frechet inception distance (FID) respectively. Evaluation of the results showed improvement of 106, 66, 12, 0.22, 0.17, 0.15, 58, and 0.39 for RMSE, MAE, PSNR, DSC, NCC, SSIM, LPIPS, and FID, respectively when the proposed blocks were used. Also, comparing the performance of the proposed network against other methods in the two modes of using paired and unpaired data showed the better performance of the proposed network.
引用
收藏
页数:17
相关论文
共 63 条
[1]  
Alkan C., 2016, GOOGL SCHOL
[2]   Patch-based generation of a pseudo CT from conventional MRI sequences for MRI-only radiotherapy of the brain [J].
Andreasen, Daniel ;
Van Leemput, Koen ;
Hansen, Rasmus H. ;
Andersen, Jon A. L. ;
Edmund, Jens M. .
MEDICAL PHYSICS, 2015, 42 (04) :1596-1605
[3]  
[Anonymous], 2016, 4 INT C LEARNING REP
[4]   Atlas-guided generation of pseudo-CT images for MRI-only and hybrid PET-MRI-guided radiotherapy treatment planning [J].
Arabi, Hossein ;
Koutsouvelis, Nikolaos ;
Rouzaud, Michel ;
Miralbell, Raymond ;
Zaidi, Habib .
PHYSICS IN MEDICINE AND BIOLOGY, 2016, 61 (17) :6531-6552
[5]   MedGAN: Medical image translation using GANs [J].
Armanious, Karim ;
Jiang, Chenming ;
Fischer, Marc ;
Kuestner, Thomas ;
Nikolaou, Konstantin ;
Gatidis, Sergios ;
Yang, Bin .
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2020, 79
[6]  
Ben-Cohen Avi, 2017, Simulation and Synthesis in Medical Imaging. Second International Workshop, SASHIMI 2017. Held in Conjunction with MICCAI 2017. Proceedings: LNCS 10557, P49, DOI 10.1007/978-3-319-68127-6_6
[7]   MRI-Based Attenuation Correction for Hybrid PET/MRI Systems: A 4-Class Tissue Segmentation Technique Using a Combined Ultrashort-Echo-Time/Dixon MRI Sequence [J].
Berker, Yannick ;
Franke, Jochen ;
Salomon, Andre ;
Palmowski, Moritz ;
Donker, Henk C. W. ;
Temur, Yavuz ;
Mottaghy, Felix M. ;
Kuhl, Christiane ;
Izquierdo-Garcia, David ;
Fayad, Zahi A. ;
Kiessling, Fabian ;
Schulz, Volkmar .
JOURNAL OF NUCLEAR MEDICINE, 2012, 53 (05) :796-804
[8]   A comprehensive survey on machine learning for networking: evolution, applications and research opportunities [J].
Boutaba, Raouf ;
Salahuddin, Mohammad A. ;
Limam, Noura ;
Ayoubi, Sara ;
Shahriar, Nashid ;
Estrada-Solano, Felipe ;
Caicedo, Oscar M. .
JOURNAL OF INTERNET SERVICES AND APPLICATIONS, 2018, 9 (01)
[9]   Iterative framework for the joint segmentation and CT synthesis of MR images: application to MRI-only radiotherapy treatment planning [J].
Burgos, Ninon ;
Guerreiro, Filipa ;
McClelland, Jamie ;
Presles, Benoit ;
Modat, Marc ;
Nill, Simeon ;
Dearnaley, David ;
deSouza, Nandita ;
Oelfke, Uwe ;
Knopf, Antje-Christin ;
Ourselin, Sebastien ;
Cardoso, M. Jorge .
PHYSICS IN MEDICINE AND BIOLOGY, 2017, 62 (11) :4237-4253
[10]   Brain-Computer Interface-Based Humanoid Control: A Review [J].
Chamola, Vinay ;
Vineet, Ankur ;
Nayyar, Anand ;
Hossain, Eklas .
SENSORS, 2020, 20 (13) :1-23