Employing deep learning and transfer learning for accurate brain tumor detection

Sandeep Kumar Mathivanan,Sridevi Sonaimuthu,Sankar Murugesan,Hariharan Rajadurai,Basu Dev Shivahare,Mohd Asif Shah
DOI: https://doi.org/10.1038/s41598-024-57970-7
IF: 4.6
2024-03-29
Scientific Reports
Abstract:Artificial intelligence-powered deep learning methods are being used to diagnose brain tumors with high accuracy, owing to their ability to process large amounts of data. Magnetic resonance imaging stands as the gold standard for brain tumor diagnosis using machine vision, surpassing computed tomography, ultrasound, and X-ray imaging in its effectiveness. Despite this, brain tumor diagnosis remains a challenging endeavour due to the intricate structure of the brain. This study delves into the potential of deep transfer learning architectures to elevate the accuracy of brain tumor diagnosis. Transfer learning is a machine learning technique that allows us to repurpose pre-trained models on new tasks. This can be particularly useful for medical imaging tasks, where labelled data is often scarce. Four distinct transfer learning architectures were assessed in this study: ResNet152, VGG19, DenseNet169, and MobileNetv3. The models were trained and validated on a dataset from benchmark database: Kaggle. Five-fold cross validation was adopted for training and testing. To enhance the balance of the dataset and improve the performance of the models, image enhancement techniques were applied to the data for the four categories: pituitary, normal, meningioma, and glioma. MobileNetv3 achieved the highest accuracy of 99.75%, significantly outperforming other existing methods. This demonstrates the potential of deep transfer learning architectures to revolutionize the field of brain tumor diagnosis.
multidisciplinary sciences
What problem does this paper attempt to address?
The problem that this paper attempts to solve is **to improve the accuracy of brain tumor diagnosis**, especially by using deep learning and transfer learning techniques in the field of medical image analysis to achieve this goal. Specifically, the paper explores how to improve the performance of brain tumor classification through transfer learning architectures (such as ResNet152, VGG19, DenseNet169 and MobileNetv3). ### Core problems of the paper: 1. **Challenges in brain tumor diagnosis**: Although magnetic resonance imaging (MRI) is the gold standard for brain tumor diagnosis, accurate classification of brain tumors remains challenging due to the complex structure of the brain. In addition, the scarcity of labeled medical data further limits the training effect of the model. 2. **Limitations of existing methods**: Traditional methods have limited performance in handling large - scale data and extracting complex features, while deep learning methods can make up for these deficiencies by processing large amounts of data and extracting high - dimensional features. However, directly training deep learning models requires a large amount of labeled data, which is usually not feasible in the medical field. 3. **Application of transfer learning**: Transfer learning is an effective solution that allows researchers to use the knowledge of pre - trained models to solve new tasks, especially when the labeled data is limited. The goal of the paper is to evaluate the performance of four transfer learning architectures in brain tumor classification and find the optimal model. --- ### Specific research objectives: - **Compare the performance of different transfer learning models**: Evaluate the performance of four transfer learning models, ResNet152, VGG19, DenseNet169 and MobileNetv3, in the brain tumor classification task through experiments. - **Optimize model performance**: Use image enhancement techniques (such as rotation, scaling, flipping, etc.) to expand the data set, alleviate the data imbalance problem, and improve the generalization ability of the model. - **Verify the optimal model**: Identify the transfer learning model most suitable for the brain tumor classification task and verify its performance in practical applications. The experimental results show that MobileNetv3 achieved **an accuracy rate of 99.75%** on the test set, significantly outperforming other methods. --- ### Summary: The main purpose of the paper is to solve the problems of data scarcity and model complexity in brain tumor classification by introducing transfer learning techniques and combining deep neural network architectures. The ultimate goal is to provide an efficient and accurate tool for clinical diagnosis to improve the treatment effect and quality of life of patients.