Improved brain tumor classification through DenseNet121 based transfer learning.
Authors
Affiliations (7)
Affiliations (7)
- Faculty of Computer Science and Information Technology, The Superior University, Lahore, 54600, Pakistan.
- Department of Computer Science, University of People, Pasadena, CA, 91101, USA.
- MLC Lab, Maharban House, House # 209, Zafar Colony, Okara, 56300, Pakistan.
- Information Technology Services, University of Okara, Okara, 56300, Pakistan.
- Department of Information Systems, College of Computer and Information Sciences, Princess Nourah bint Abdulrahman University, P.O. Box 84428, Riyadh, 11671, Saudi Arabia.
- School of Biochemistry and Biotechnology, University of the Punjab, Lahore, Pakistan.
- Department of Computer Science, Bahria University Lahore Campus, Lahore, 54000, Pakistan. [email protected].
Abstract
Brain tumors have a big effect on a person's health by letting abnormal cells grow unchecked in the brain. This means that early and correct diagnosis is very important for effective treatment. Many of the current diagnostic methods are time-consuming, rely primarily on hand interpretation, and frequently yield unsatisfactory results. This work finds brain tumors in MRI data using DenseNet121 architecture with transfer learning. Model training made use of the Kaggle dataset. By preprocessing the stage, resizing the MRI pictures to minimize noise would help the model perform better. From one MRI scan, the proposed approach divides brain tissues into four groups: benign tumors, gliomas, meningiomas, and pituitary gland malignancies. The designed DenseNet121 architecture precisely classifies brain cancers. We assessed the models' performance in terms of accuracy, precision, recall, and F1-score. The suggested approach proved successful in the multi-class categorization of brain tumors since it attained an average accuracy improvement of 96.90%. Unlike previous diagnostic techniques, such as eye inspection and other machine learning models, the proposed DenseNet121-based approach is more accurate, takes less time to analyze, and requires less human input. Although the automated method ensures consistent and predictable results, human error sometimes causes more unpredictability in conventional methods. Based on MRI-based detection and transfer learning, this paper proposes an automated method for the classification of brain cancers. The method improves the precision and speed of brain tumor diagnosis, which benefits both MRI-based classification research and clinical use. The development of deep-learning models may even further improve tumor identification and prognosis prediction.