Department of Computer Science and Engineering (Artificial Intelligence and Machine Learning), Harsha Institute of Technology, Bengaluru, Karnataka, India.
Brain Tumor (BT) leads to disability in cognitive, motor, and social skills, and therefore, early diagnosis should be a milestone for treatment. In this work, a novel Federated Learning-based Convolutional Neural Network (FL-CNN) model is proposed for Brain Tumor Classification (BTC) with FL serving as the framework for the model. The model is trained to distinguish between four classes of brain conditions: glioma, meningioma, pituitary adenoma, and non-neoplastic growth. Through the use of Federated Learning (FL), this method allows multiple Decentralized clients to cooperate in training the model without exchanging the raw medical data belonging to the patients. The provided dataset is derived from a training set containing 5707 images and a testing set containing 1311 images, and both sets are labeled among four categories. The fully trained 2D-CNN model deals with pre-processed MRI images in dimensions of 128×128 pixels and internalizes key attributes for identifying all forms of BT. As for understanding the model’s performance, we compute accuracy, precision, recall, and F1-score. The model achieved a peak validation accuracy of 97.48% with a precision, recall, and F1 score of 97.48%. Early stopping was applied at round 12 due to performance stagnation, preventing overfitting. The final global accuracy reached 97.48%, with a loss of 0.1483, demonstrating strong classification performance. The results exhibit that the federated strategy yields comparable classification accuracy with the conventional approach for distributed data and minimizes the violation of individual data privacy. Moreover, this work discusses the applicability of FL to medical image analysis, indicating that collaborative models in this area can provide a highly accurate performance while avoiding data aggregation. The following paper is intended to contribute to the improvement of privacy-preserving ML pertaining to medical diagnosis with regard to BTs.
M. J. Sheller et al., “Federated learning in medicine: facilitating multi-institutional collaborations without sharing patient data,” Scientific Reports, vol. 10, no. 1, Jul. 2020, doi: 10.1038/s41598-020-69250-1.
E. Isik-Polat, G. Polat, A. Kocyigit, and A. Temizel, “Evaluation and Analysis of Different Aggregation and Hyperparameter Selection Methods for Federated Brain Tumor Segmentation,” Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries, pp. 405–419, 2022, doi: 10.1007/978-3-031-09002-8_36.
J. Amin, M. Sharif, A. Haldorai, M. Yasmin, and R. S. Nayak, “Brain tumor detection and classification using machine learning: a comprehensive survey,” Complex & Intelligent Systems, vol. 8, no. 4, pp. 3161–3183, Nov. 2021, doi: 10.1007/s40747-021-00563-y.
M. Aggarwal et al., “Privacy preserved collaborative transfer learning model with heterogeneous distributed data for brain tumor classification,” International Journal of Imaging Systems and Technology, vol. 34, no. 2, Nov. 2023, doi: 10.1002/ima.22994.
D. H. Mahlool and M. H. Abed, “Distributed brain tumor diagnosis using a federated learning environment,” Bulletin of Electrical Engineering and Informatics, vol. 11, no. 6, pp. 3313–3321, Dec. 2022, doi: 10.11591/eei.v11i6.4131.
W. Li et al., “Privacy-Preserving Federated Brain Tumour Segmentation,” Machine Learning in Medical Imaging, pp. 133–141, 2019, doi: 10.1007/978-3-030-32692-0_16.
O. Atef, M. A. Salam, and H. Abdelsalam, “Federated Learning Approach for Measuring the Response of Brain Tumors to Chemotherapy,” International Journal of Advanced Computer Science and Applications, vol. 13, no. 10, 2022, doi: 10.14569/ijacsa.2022.0131060.
S. Pati, "The federated tumor segmentation (FeTS) challenge," 2021. doi: 10.48550/arxiv.2105.05874.
P. Zhang, Y. Zhou, M. Hu, X. Fu, X. Wang, and M. Chen, "CyclicFL: A cyclic model pre-training approach to efficient FL," 2023. doi: 10.48550/arxiv.2301.12193.
A. A. Asiri et al., “Enhancing Brain Tumor Diagnosis: Transitioning From Convolutional Neural Network to Involutional Neural Network,” IEEE Access, vol. 11, pp. 123080–123095, 2023, doi: 10.1109/access.2023.3326421.
R. Richterová et al., “Most frequent molecular and immunohistochemical markers present in selected types of brain tumors,” General physiology and biophysics, vol. 33, no. 03, pp. 259–279, 2014, doi: 10.4149/gpb_2014007.
T. A. Jemimma and Y. J. Vetharaj, “Fractional probabilistic fuzzy clustering and optimization based brain tumor segmentation and classification,” Multimedia Tools and Applications, vol. 81, no. 13, pp. 17889–17918, Mar. 2022, doi: 10.1007/s11042-022-11969-2.
D. Rammurthy and P. K. Mahesh, “Whale Harris hawks optimization based deep learning classifier for brain tumor detection using MRI images,” Journal of King Saud University - Computer and Information Sciences, vol. 34, no. 6, pp. 3259–3272, Jun. 2022, doi: 10.1016/j.jksuci.2020.08.006.
R. Vankdothu and M. A. Hameed, “Brain tumor MRI images identification and classification based on the recurrent convolutional neural network,” Measurement: Sensors, vol. 24, p. 100412, Dec. 2022, doi: 10.1016/j.measen.2022.100412.
P. Agrawal, N. Katal, and N. Hooda, “Segmentation and classification of brain tumor using 3D-UNet deep neural networks,” International Journal of Cognitive Computing in Engineering, vol. 3, pp. 199–210, Jun. 2022, doi: 10.1016/j.ijcce.2022.11.001.
M. Islam, Md. T. Reza, M. Kaosar, and M. Z. Parvez, “Effectiveness of Federated Learning and CNN Ensemble Architectures for Identifying Brain Tumors Using MRI Images,” Neural Processing Letters, vol. 55, no. 4, pp. 3779–3809, Aug. 2022, doi: 10.1007/s11063-022-11014-1.
B. A. Kumar and N. Lakshmidevi, “A Multi Brain Tumor Classification Using a Deep Reinforcement Learning Model,” Meta Heuristic Techniques in Software Engineering and Its Applications, pp. 134–144, 2022, doi: 10.1007/978-3-031-11713-8_14.
S. Hossain, A. Chakrabarty, T. R. Gadekallu, M. Alazab, and Md. J. Piran, “Vision Transformers, Ensemble Model, and Transfer Learning Leveraging Explainable AI for Brain Tumor Detection and Classification,” IEEE Journal of Biomedical and Health Informatics, vol. 28, no. 3, pp. 1261–1272, Mar. 2024, doi: 10.1109/jbhi.2023.3266614.
S. Das, O. F. M. R. R. Aranya, and N. N. Labiba, “Brain Tumor Classification Using Convolutional Neural Network,” 2019 1st International Conference on Advances in Science, Engineering and Robotics Technology (ICASERT), pp. 1–5, May 2019, doi: 10.1109/icasert.2019.8934603.
Abiwinanda, N., Hanif, M., Hesaputra, S.T., Handayani, A., Mengko, T.R. (2019). BT Classification Using CNN. In: Lhotska, L., Sukupova, L., Lacković, I., Ibbott, G.S. (eds) World Congress on Medical Physics and Biomedical Engineering 2018. IFMBE Proceedings, vol 68/1. Springer, Singapore. https://doi.org/10.1007/978-981-10-9035-6_33
S. Bhadauriya, T. Merothiya, S. C. Yadav, and M. ChandraPrabha, “Detection of Brain Tumour using CNN in Federated Machine Learning,” 2023 5th International Conference on Advances in Computing, Communication Control and Networking (ICAC3N), pp. 653–658, Dec. 2023, doi: 10.1109/icac3n60023.2023.10541410.
S. Deepa, J. Janet, S. Sumathi, and J. P. Ananth, “Hybrid Optimization Algorithm Enabled Deep Learning Approach Brain Tumor Segmentation and Classification Using MRI,” Journal of Digital Imaging, vol. 36, no. 3, pp. 847–868, Jan. 2023, doi: 10.1007/s10278-022-00752-2.
N. Sivakumar et al., “A Hybrid Brain Tumor Classification Using FL With FedAvg and FedProx for Privacy and Robustness Across Heterogeneous Data Sources,” IEEE Access, vol. 13, pp. 57705–57719, 2025, doi: 10.1109/access.2025.3549440.
CRediT Author Statement
The authors confirm contribution to the paper as follows:
Conceptualization: Sivakumar N, Renukadevi S, Manujakshi B C and Shashidhar T M;
Methodology: Sivakumar N and Renukadevi S;
Writing- Original Draft Preparation: Sivakumar N, Renukadevi S, Manujakshi B C and Shashidhar T M;
Visualization: Sivakumar N and Renukadevi S;
Investigation: Manujakshi B C and Shashidhar T M;
Supervision: Sivakumar N and Renukadevi S;
Validation: Manujakshi B C and Shashidhar T M;
Writing- Reviewing and Editing: Sivakumar N, Renukadevi S, Manujakshi B C and Shashidhar T M; All authors reviewed the results and approved the final version of the manuscript.
Acknowledgements
Authors thank Reviewers for taking the time and effort necessary to review the manuscript.
Funding
No funding was received to assist with the preparation of this manuscript.
Ethics declarations
Conflict of interest
The authors have no conflicts of interest to declare that are relevant to the content of this article.
Availability of data and materials
Data sharing is not applicable to this article as no new data were created or analysed in this study.
Author information
Contributions
All authors have equal contribution in the paper and all authors have read and agreed to the published version of the manuscript.
Corresponding author
Sivakumar N
Department of Computer Engineering, Marwadi University, Rajkot, Gujarat, India.
Open Access This article is licensed under a Creative Commons Attribution NoDerivs is a more restrictive license. It allows you to redistribute the material commercially or non-commercially but the user cannot make any changes whatsoever to the original, i.e. no derivatives of the original work. To view a copy of this license, visit https://creativecommons.org/licenses/by-nc-nd/4.0/
Cite this article
Sivakumar N, Renukadevi S, Manujakshi B C and Shashidhar T M, “A Fusion Based Federated Learning Approach for Multi Class Brain Tumor Classification with Enhanced Privacy”, Journal of Machine and Computing, vol.5, no.4, pp. 2475-2494, October 2025, doi: 10.53759/7669/jmc202505190.
