Bidirectional brain image translation using transfer learning from generic pre-trained models

被引:2
作者
Haimour, Fatima [1 ]
Al-Sayyed, Rizik [2 ]
Mahafza, Waleed [3 ]
Al-Kadi, Omar S. [2 ]
机构
[1] Zarqa Univ, Fac Informat Technol, Zarqa 13110, Jordan
[2] Univ Jordan, King Abdullah 2 Sch Informat Technol, Amman 11942, Jordan
[3] Jordan Univ Hosp, Dept Diagnost Radiol, Amman 11942, Jordan
关键词
Image translation; Transfer learning; Pre-trained model; Brain tumor; Magnetic resonance imaging; Computed tomography; CycleGAN;
D O I
10.1016/j.cviu.2024.104100
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Brain imaging plays a crucial role in the diagnosis and treatment of various neurological disorders, providing valuable insights into the structure and function of the brain. Techniques such as magnetic resonance imaging (MRI) and computed tomography (CT) enable non-invasive visualization of the brain, aiding in the understanding of brain anatomy, abnormalities, and functional connectivity. However, cost and radiation dose may limit the acquisition of specific image modalities, so medical image synthesis can be used to generate required medical images without actual addition. CycleGAN and other GANs are valuable tools for generating synthetic images across various fields. In the medical domain, where obtaining labeled medical images is labor-intensive and expensive, addressing data scarcity is a major challenge. Recent studies propose using transfer learning to overcome this issue. This involves adapting pre-trained CycleGAN models, initially trained on non-medical data, to generate realistic medical images. In this work, transfer learning was applied to the task of MR-CT image translation and vice versa using 18 pre-trained non-medical models, and the models were fine-tuned to have the best result. The models' performance was evaluated using four widely used image quality metrics: Peak-signal-to-noise-ratio, Structural Similarity Index, Universal Quality Index, and Visual Information Fidelity. Quantitative evaluation and qualitative perceptual analysis by radiologists demonstrate the potential of transfer learning in medical imaging and the effectiveness of the generic pre-trained model. The results provide compelling evidence of the model's exceptional performance, which can be attributed to the high quality and similarity of the training images to actual human brain images. These results underscore the significance of carefully selecting appropriate and representative training images to optimize performance in brain image analysis tasks.
引用
收藏
页数:15
相关论文
共 55 条
[51]   Glioma Grading on Conventional MR Images: A Deep Learning Study With Transfer Learning [J].
Yang, Yang ;
Yan, Lin-Feng ;
Zhang, Xin ;
Han, Yu ;
Nan, Hai-Yan ;
Hu, Yu-Chuan ;
Hu, Bo ;
Yan, Song-Lin ;
Zhang, Jin ;
Cheng, Dong-Liang ;
Ge, Xiang-Wei ;
Cui, Guang-Bin ;
Zhao, Di ;
Wang, Wen .
FRONTIERS IN NEUROSCIENCE, 2018, 12
[52]   Seismic impedance inversion based on geophysical-guided cycle-consistent generative adversarial networks [J].
Zhang, Haihang ;
Zhang, Guangzhi ;
Gao, Jianhu ;
Li, Shengjun ;
Zhang, Jinmiao ;
Zhu, Zhenyu .
JOURNAL OF PETROLEUM SCIENCE AND ENGINEERING, 2022, 218
[53]   Unsupervised Conditional Consensus Adversarial Network for Brain Disease Identification with Structural MRI [J].
Zhang, Jing ;
Liu, Mingxia ;
Pan, Yongsheng ;
Shen, Dinggang .
MACHINE LEARNING IN MEDICAL IMAGING (MLMI 2019), 2019, 11861 :391-399
[54]   Translating and Segmenting Multimodal Medical Volumes with Cycle- and Shape-Consistency Generative Adversarial Network [J].
Zhang, Zizhao ;
Yang, Lin ;
Zheng, Yefeng .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :9242-9251
[55]   Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks [J].
Zhu, Jun-Yan ;
Park, Taesung ;
Isola, Phillip ;
Efros, Alexei A. .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :2242-2251