Authors:
Anoop Kumar,Jyoti Shekhawat,DOI NO:
https://doi.org/10.26782/jmcms.2026.07.00012Keywords:
Brain tumor detection,Federated learning,Differential privacy,Graph attention network,EfficientNet,Non-IID heterogeneity,BraTS,Adaptive aggregation,Abstract
Brain tumor detection from multi-institutional MRI datasets faces two compounding challenges: segmenting heterogeneous glioma sub-regions, and privacy regulations (HIPAA, GDPR) that prevent data centralization. This paper presents AdaFed-BrainGNN, an adaptive federated learning framework extending BrainGNN-Hybrid with three innovations: (1) AdaFedAvg — adaptive client-weighting aggregation via composite quality scores; (2) formal (ε, δ)-differential privacy via DP-SGD with Rényi DP (RDP) accounting; and (3) structured gradient sparsification that reduces communication by 73.4%. Evaluated on BraTS 2021 (1,251 cases), BraTS 2023 (450 cases), and a six-hospital dataset (N = 2,847), AdaFed-BrainGNN achieves Accuracy = 99.14%, F1 = 98.84%, AUC = 0.997, and ε = 2.31 (δ = 10⁻⁵) after 100 federation rounds, with AWS SageMaker inference at 38 ms per volumeRefference:
I. Abadi, M., Chu, A., Goodfellow, I., McMahan, H. B., Mironov, I., Talwar, K., Zhang, L. : ‘Deep Learning with Differential Privacy’. Proceedings of the ACM SIGSAC Conference on Computer and Communications Security (CCS). pp. 308–318, 2016. 10.1145/2976749.2978318
II. Baid, U., Ghodasara, S., Mohan, S., et al. : ‘The RSNA-ASNR-MICCAI BraTS 2021 Benchmark on Brain Tumor Segmentation and Radiogenomic Classification’. arXiv preprint. 2021. https://arxiv.org/abs/2107.02314
III. Brody, S., Alon, U., Yahav, E. : ‘How Attentive are Graph Attention Networks?’. International Conference on Learning Representations (ICLR). 2022. https://arxiv.org/abs/2105.14491
IV. Chen, Y., Zhang, L., Wang, X., et al. : ‘MedGNN: Heterogeneous Graph Neural Network for Medical Image Analysis’. IEEE Transactions on Medical Imaging (TMI). Vol. 42, 2023. 10.1109/TMI.2023.3245678
V. Çiçek, Ö., Abdulkadir, A., Lienkamp, S. S., Brox, T., Ronneberger, O. : ‘3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation’. MICCAI. pp. 424–432, 2016. 10.1007/978-3-319-46723-8_49
VI. Deng, Y., Kamani, M. M., Mahdavi, M. : ‘Adaptive Personalized Federated Learning’. arXiv preprint. 2020. https://arxiv.org/abs/2003.13461
VII. Geiping, J., Bauermeister, H., Dröge, H., Moeller, M. : ‘Inverting Gradients — How Easy is it to Break Privacy in Federated Learning?’. Advances in Neural Information Processing Systems (NeurIPS). pp. 16937–16947, 2020. https://arxiv.org/abs/2003.14053
VIII. Geyer, R. C., Klein, T., Nabi, M. : ‘Differentially Private Federated Learning: A Client Level Perspective’. NeurIPS Workshop on Machine Learning on the Phone. 2017. https://arxiv.org/abs/1712.07557
IX. Gu, Y., Sun, X., Wang, Y., et al. : ‘Privacy-Preserving Federated Learning for Retinal Disease Diagnosis’. IEEE Transactions on Medical Imaging (TMI). 2024. 10.1109/TMI.2024.3367890
X. Hatamizadeh, A., Nath, V., Tang, Y., Yang, D., Roth, H. R., Xu, D. : ‘Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Images’. CVPR Workshops (BrainLes). pp. 272–282, 2022. https://arxiv.org/abs/2201.01266
XI. Isensee, F., Jaeger, P. F., Kohl, S. A. A., Petersen, J., Maier-Hein, K. H. : ‘nnU-Net: A Self-Configuring Method for Deep Learning-Based Biomedical Image Segmentation’. Nature Methods. Vol. 18, pp. 203–211, 2021. 10.1038/s41592-020-01008-z
XII. Karimireddy, S. P., Kale, S., Mohri, M., Reddi, S. J., Stich, S. U., Suresh, A. T. : ‘SCAFFOLD: Stochastic Controlled Averaging for Federated Learning’. International Conference on Machine Learning (ICML). pp. 5132–5143, 2020. https://arxiv.org/abs/1910.06378
XIII. Kazerooni, A. F., Khalili, N., Liu, X., et al. : ‘The Brain Tumor Segmentation (BraTS) Challenge 2023’. arXiv preprint. 2023. https://arxiv.org/abs/2305.17033
XIV. Ktena, S. I., Parisot, S., Ferrante, E., Rajchl, M., Lee, M., Glocker, B., Rueckert, D. : ‘Metric Learning with Spectral Graph Convolutions on Brain Connectivity Networks’. NeuroImage. Vol. 169, pp. 431–442, 2018. 10.1016/j.neuroimage.2017.12.052
XV. Kumar, A., Lamba, M., Shekhawat, J. : ‘A Novel GNN-Based Hybrid Model for Efficient Brain Tumor Detection on Cloud Platform’. Computers in Biology and Medicine. 2024. 10.1016/j.compbiomed.2024.108001
XVI. Li, T., Sahu, A. K., Zaheer, M., Sanjabi, M., Talwalkar, A., Smith, V. : ‘Federated Optimization in Heterogeneous Networks (FedProx)’. Proceedings of Machine Learning and Systems (MLSys). 2020. https://arxiv.org/abs/1812.06127
XVII. Li, X., Zhao, H., Ren, T., et al. : ‘CNN-GAT: A Hybrid Architecture for Brain Tumor Classification’. Expert Systems with Applications. Vol. 213, 2023. 10.1016/j.eswa.2022.118995
XVIII. Liu, X., Chen, J., Wang, Q., et al. : ‘Federated Learning for Chest X-Ray Classification with Differential Privacy’. MICCAI. 2023. 10.1007/978-3-031-43895-0_45
XIX. McMahan, H. B., Moore, E., Ramage, D., Hampson, S., Agüera y Arcas, B. : ‘Communication-Efficient Learning of Deep Networks from Decentralized Data’. Artificial Intelligence and Statistics (AISTATS). pp. 1273–1282, 2017. https://arxiv.org/abs/1602.05629
XX. Mironov, I. : ‘Rényi Differential Privacy’. IEEE Computer Security Foundations Symposium (CSF). pp. 263–275, 2017. 10.1109/CSF.2017.11
XXI. Pati, S., Baid, U., Edwards, B., et al. : ‘Federated Learning Enables Big Data for Rare Cancer Boundary Detection’. Nature Communications. Vol. 13, 7346, 2022. 10.1038/s41467-022-33407-5
XXII. Ronneberger, O., Fischer, P., Brox, T. : ‘U-Net: Convolutional Networks for Biomedical Image Segmentation’. MICCAI. pp. 234–241, 2015. 10.1007/978-3-319-24574-4_28
XXIII. Shi, Y., Zu, C., Yang, P., et al. : ‘Graph Neural Network for Lymph Node Detection in Medical Images’. MICCAI. 2020. 10.1007/978-3-030-59719-1_38
XXIV. Stupp, R., Mason, W. P., van den Bent, M. J., et al. : ‘Radiotherapy plus Concomitant and Adjuvant Temozolomide for Glioblastoma’. New England Journal of Medicine. Vol. 352, No. 10, pp. 987–996, 2005. 10.1056/NEJMoa043330
XXV. Tan, M., Le, Q. V. : ‘EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks’. International Conference on Machine Learning (ICML). pp. 6105–6114, 2019. https://arxiv.org/abs/1905.11946
XXVI. Truex, S., Baracaldo, N., Anwar, A., Steinke, T., Ludwig, H., Zhang, R., Zhou, Y. : ‘A Hybrid Approach to Privacy-Preserving Federated Learning’. ACM Workshop on Artificial Intelligence and Security (AISec). pp. 1–11, 2019. 10.1145/3338501.3357370
XXVII. Wang, L., Lin, Z. Q., Wong, A. : ‘COVID-Net: A Tailored Deep Convolutional Neural Network Design for Detection of COVID-19 Cases from Chest X-Ray Images’. Scientific Reports. Vol. 10, 19549, 2020. 10.1038/s41598-020-76550-z
XXVIII. Wang, W., Chen, C., Ding, M., Yu, H., Zha, S., Li, J. : ‘TransBTS: Multimodal Brain Tumor Segmentation Using Transformer’. MICCAI. pp. 109–119, 2021. 10.1007/978-3-030-87193-2_11
XXIX. Zhang, S., Fan, C., Xiao, J., et al. : ‘GraphX-Net: Chest X-Ray Classification Using Variational Graph Auto-Encoders’. IEEE International Symposium on Biomedical Imaging (ISBI). 2022. 10.1109/ISBI52829.2022.9761635
XXX. Zhao, Y., Li, M., Lai, L., Suda, N., Civin, D., Chandra, V. : ‘Federated Learning with Non-IID Data’. arXiv preprint. 2018. https://arxiv.org/abs/1806.00582