Integrated approach of federated learning with transfer learning for classification and diagnosis of brain tumor

Albalawi, Eid; T.R., Mahesh; Thakur, Arastu; Kumar, V. Vinoth; Gupta, Muskan; Khan, Surbhi Bhatia; Almusharraf, Ahlam

Abstract

Brain tumor classification using MRI images is a crucial yet challenging task in medical imaging. Accurate diagnosis is vital for effective treatment planning but is often hindered by the complex nature of tumor morphology and variations in imaging. Traditional methodologies primarily rely on manual interpretation of MRI images, supplemented by conventional machine learning techniques. These approaches often lack the robustness and scalability needed for precise and automated tumor classification. The major limitations include a high degree of manual intervention, potential for human error, limited ability to handle large datasets, and lack of generalizability to diverse tumor types and imaging conditions.To address these challenges, we propose a federated learning-based deep learning model that leverages the power of Convolutional Neural Networks (CNN) for automated and accurate brain tumor classification. This innovative approach not only emphasizes the use of a modified VGG16 architecture optimized for brain MRI images but also highlights the significance of federated learning and transfer learning in the medical imaging domain. Federated learning enables decentralized model training across multiple clients without compromising data privacy, addressing the critical need for confidentiality in medical data handling. This model architecture benefits from the transfer learning technique by utilizing a pre-trained CNN, which significantly enhances its ability to classify brain tumors accurately by leveraging knowledge gained from vast and diverse datasets.Our model is trained on a diverse dataset combining figshare, SARTAJ, and Br35H datasets, employing a federated learning approach for decentralized, privacy-preserving model training. The adoption of transfer learning further bolsters the model’s performance, making it adept at handling the intricate variations in MRI images associated with different types of brain tumors. The model demonstrates high precision (0.99 for glioma, 0.95 for meningioma, 1.00 for no tumor, and 0.98 for pituitary), recall, and F1-scores in classification, outperforming existing methods. The overall accuracy stands at 98%, showcasing the model’s efficacy in classifying various tumor types accurately, thus highlighting the transformative potential of federated learning and transfer learning in enhancing brain tumor classification using MRI images.

Citation

Albalawi, E., T.R., M., Thakur, A., Kumar, V. V., Gupta, M., Khan, S. B., & Almusharraf, A. (in press). Integrated approach of federated learning with transfer learning for classification and diagnosis of brain tumor. BMC Medical Imaging, 24(1), 110. https://doi.org/10.1186/s12880-024-01261-0

• Journal Article Type: Article
• Acceptance Date: Mar 27, 2024
• Online Publication Date: May 15, 2024
• Deposit Date: May 20, 2024
• Publicly Available Date: May 20, 2024
• Journal: BMC Medical Imaging
• Publisher: Springer Verlag
• Peer Reviewed: Peer Reviewed
• Volume: 24
• Issue: 1
• Pages: 110
• DOI: https://doi.org/10.1186/s12880-024-01261-0
• Keywords: Convolutional neural networks, Brain tumor classification, MRI imaging, VGG16, Medical image analysis, Federated learning