TumorGANet: A Transfer Learning and Generative Adversarial Network- Based Data Augmentation Model for Brain Tumor Classification
Nag, A., Mondal, H., Mehedi Hassan, M. 
ORCID: 0000-0002-9890-0968 , Al-Shehari, T. ORCID: 0000-0002-9783-919X, Kadrie, M. ORCID: 0009-0000-4591-5069, Al-Razgan, M. ORCID: 0000-0002-9705-3867, Alfakih, T. ORCID: 0000-0003-0366-5932, Biswas, S. ORCID: 0000-0002-6770-9845 & Kumar Bairagi, A. ORCID: 0009-0000-9132-8893 (2024).
TumorGANet: A Transfer Learning and Generative Adversarial Network- Based Data Augmentation Model for Brain Tumor Classification.
IEEE Access, 12,
pp. 103060-103081.
doi: 10.1109/access.2024.3429633
Abstract
Diagnosing brain tumors using magnetic resonance imaging (MRI) presents significant challenges due to the complexities of segmentation and the variability in tumor characteristics. To address the limitations inherent in traditional methods, this research employs an advanced deep learning approach, integrating ResNet50 for feature extraction and Generative Adversarial Networks (GANs) for data augmentation. A comprehensive evaluation of ten transfer learning algorithms, including GoogLeNet and VGG-16, was conducted for the classification of brain tumors. Model performance was assessed using precision, recall, and F1-score metrics, complemented by additional metrics such as Hamming loss and the Matthews correlation coefficient to provide a more comprehensive insight. To ensure transparency in image predictions, Explainable AI techniques, specifically Local Interpretable Model-Agnostic Explanations (LIME), were utilized. The study involved the analysis of 7023 MRI images, with TumorGANet being trained on a dataset encompassing gliomas, meningiomas, non-tumorous cases, and pituitary tumors. The results demonstrate the exceptional performance of proposed model named TumorGANet, achieving an accuracy of 99.53%, precision and recall rates of 100%, F1 scores of 99%, and a Hamming loss of 0.2%.
| Publication Type: | Article |
|---|---|
| Additional Information: | 2024 The Authors. This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 License. For more information, see https://creativecommons.org/licenses/by-nc-nd/4.0/ |
| Publisher Keywords: | Brain imaging, brain tumor, transfer learning, generative adversarial network, explainable AI |
| Subjects: | Q Science > QA Mathematics > QA75 Electronic computers. Computer science R Medicine > RC Internal medicine > RC0254 Neoplasms. Tumors. Oncology (including Cancer) R Medicine > RC Internal medicine > RC0321 Neuroscience. Biological psychiatry. Neuropsychiatry |
| Departments: | School of Science & Technology School of Science & Technology > Computer Science |
| SWORD Depositor: |
Available under License Creative Commons Attribution Non-commercial No Derivatives.
Download (6MB) | Preview
Export
Downloads
Downloads per month over past year
Metadata
MetadataActions (login required)
Admin Login