Improving Effectiveness of Different Deep Transfer Learning-Based Models for Detecting Brain Tumors From MR Images

被引:75
作者
Asif, Sohaib [1 ,2 ]
Yi, Wenhui [1 ]
Ul Ain, Qurrat [3 ]
Hou, Jin [4 ]
Yi, Tao [5 ]
Si, Jinhai [1 ]
机构
[1] Xi An Jiao Tong Univ, Fac Elect & Informat Engn, Sch Elect Sci & Engn, Xian 710049, Shaanxi, Peoples R China
[2] Cent South Univ, Sch Comp Sci & Engn, Changsha 410083, Hunan, Peoples R China
[3] Cent South Univ, Sch Publ Hlth, Changsha 410007, Hunan, Peoples R China
[4] Xian Med Univ, Sch Basic Med Sci, Dept Pharmacol, Xian 710021, Shaanxi, Peoples R China
[5] Xi An Jiao Tong Univ, Sch Comp Sci & Engn, Xian 710049, Shaanxi, Peoples R China
基金
中国国家自然科学基金;
关键词
Tumors; Brain modeling; Magnetic resonance imaging; Feature extraction; Transfer learning; Solid modeling; Deep learning; Brain tumor classification; transfer learning; deep learning; magnetic resonance imaging; convolutional neural network; CLASSIFICATION; SEGMENTATION; FEATURES;
D O I
10.1109/ACCESS.2022.3153306
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Early classification of brain tumors from magnetic resonance imaging (MRI) plays an important role in the diagnosis of such diseases. There are many diagnostic imaging methods used to identify tumors in the brain. MRI is commonly used for such tasks because of its unmatched image quality. The relevance of artificial intelligence (AI) in the form of deep learning (DL) has revolutionized new methods of automated medical image diagnosis. This study aimed to develop a robust and efficient method based on transfer learning technique for classifying brain tumors using MRI. In this article, the popular deep learning architectures are utilized to develop brain tumor diagnostic system. The pre-trained models such as Xception, NasNet Large, DenseNet121 and InceptionResNetV2 are used to extract the deep features from brain MRI. The experiment was performed using two benchmark datasets that are openly accessible from the web. Images from the dataset were first cropped, preprocessed, and augmented for accurate and fast training. Deep transfer learning models are trained and tested on a brain MRI dataset using three different optimization algorithms (ADAM, SGD, and RMSprop). The performance of the transfer learning models is evaluated using performance metrics such as accuracy, sensitivity, precision, specificity and F1-score. From the experimental results, our proposed CNN model based on the Xception architecture using ADAM optimizer is better than the other three proposed models. The Xception model achieved accuracy, sensitivity, precision specificity, and F1-score values of 99.67%, 99.68%, 99.68%, 99.66%, and 99.68% on the MRI-large dataset, and 91.94%, 96.55%, 87.50%, 87.88%, and 91.80% on the MRI-small dataset, respectively. The proposed method is superior to the existing literature, indicating that it can be used to quickly and accurately classify brain tumors.
引用
收藏
页码:34716 / 34730
页数:15
相关论文
共 45 条
[1]   Brain Tumor Classification Using Convolutional Neural Network [J].
Abiwinanda, Nyoman ;
Hanif, Muhammad ;
Hesaputra, S. Tafwida ;
Handayani, Astri ;
Mengko, Tati Rajab .
WORLD CONGRESS ON MEDICAL PHYSICS AND BIOMEDICAL ENGINEERING 2018, VOL 1, 2019, 68 (01) :183-189
[2]  
Afshar P, 2019, INT CONF ACOUST SPEE, P1368, DOI [10.1109/ICASSP.2019.8683759, 10.1109/icassp.2019.8683759]
[3]   Hybrid gray wolf optimizer-artificial neural network classification approach for magnetic resonance brain images [J].
Ahmed, Heba M. ;
Youssef, Bayumy A. B. ;
Elkorany, Ahmed S. ;
Saleeb, Adel A. ;
Abd El-Samie, Fathi .
APPLIED OPTICS, 2018, 57 (07) :B25-B31
[4]   A distinctive approach in brain tumor detection and classification using MRI [J].
Amin, Javeria ;
Sharif, Muhammad ;
Yasmin, Mussarat ;
Fernandes, Steven Lawrence .
PATTERN RECOGNITION LETTERS, 2020, 139 :118-127
[5]   Classification of Brain Tumors from MRI Images Using a Convolutional Neural Network [J].
Badza, Milica M. ;
Barjaktarovic, Marko C. .
APPLIED SCIENCES-BASEL, 2020, 10 (06)
[6]  
Chakrabarty N., 2019, BRAIN MRI IMAGES DAT
[7]  
Chelghoum R., 2020, PROC IFIP INT C ARTI, V583, P189, DOI DOI 10.1007/978-3-030-49161-1_17
[8]   Enhanced Performance of Brain Tumor Classification via Tumor Region Augmentation and Partition [J].
Cheng, Jun ;
Huang, Wei ;
Cao, Shuangliang ;
Yang, Ru ;
Yang, Wei ;
Yun, Zhaoqiang ;
Wang, Zhijian ;
Feng, Qianjin .
PLOS ONE, 2015, 10 (10)
[9]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[10]   Detection of tumors on brain MRI images using the hybrid convolutional neural network architecture [J].
Cinar, Ahmet ;
Yildirim, Muhammed .
MEDICAL HYPOTHESES, 2020, 139