IEEE Access (Jan 2022)
Improving Effectiveness of Different Deep Transfer Learning-Based Models for Detecting Brain Tumors From MR Images
Abstract
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.
Keywords
