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 条
[11]  
Bi YD, 2023, Arxiv, DOI arXiv:2303.09012
[12]   Transfer Learning Using Convolutional Neural Network Architectures for Brain Tumor Classification from MRI Images [J].
Chelghoum, Rayene ;
Ikhlef, Ameur ;
Hameurlaine, Amina ;
Jacquir, Sabir .
ARTIFICIAL INTELLIGENCE APPLICATIONS AND INNOVATIONS, AIAI 2020, PT I, 2020, 583 :189-200
[13]   Not-so-supervised: A survey of semi-supervised, multi-instance, and transfer learning in medical image analysis [J].
Cheplygina, Veronika ;
de Bruijne, Marleen ;
Pluim, Josien P. W. .
MEDICAL IMAGE ANALYSIS, 2019, 54 :280-296
[14]   Detection of tumors on brain MRI images using the hybrid convolutional neural network architecture [J].
Cinar, Ahmet ;
Yildirim, Muhammed .
MEDICAL HYPOTHESES, 2020, 139
[15]   The Cityscapes Dataset for Semantic Urban Scene Understanding [J].
Cordts, Marius ;
Omran, Mohamed ;
Ramos, Sebastian ;
Rehfeld, Timo ;
Enzweiler, Markus ;
Benenson, Rodrigo ;
Franke, Uwe ;
Roth, Stefan ;
Schiele, Bernt .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :3213-3223
[16]   Automated Categorization of Brain Tumor from MRI Using CNN features and SVM [J].
Deepak, S. ;
Ameer, P. M. .
JOURNAL OF AMBIENT INTELLIGENCE AND HUMANIZED COMPUTING, 2021, 12 (08) :8357-8369
[17]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[18]  
Depeursinge A, 2017, ELS MIC SOC BOOK SER, P1
[19]  
Di Ieva Antonio, 2024, Adv Neurobiol, V36, P525, DOI 10.1007/978-3-031-47606-8_27
[20]   On Handling Negative Transfer and Imbalanced Distributions in Multiple Source Transfer Learning [J].
Ge, Liang ;
Gao, Jing ;
Ngo, Hung ;
Li, Kang ;
Zhang, Aidong .
STATISTICAL ANALYSIS AND DATA MINING, 2014, 7 (04) :254-271