Abstract
This research work describes a lightweight, secure, and interpretable federated learning framework for automatic leukemia classification, which identifies and addresses various problems regarding clinical data security and collaborative model building among partnering healthcare organizations. This framework employs a distributed learning paradigm that allows a number of healthcare facilities to work together to build a high predictive performance classification model while training the model without exchanging sensitive information about patient data, thus ensuring data privacy and methodological reproducibility. The proposed framework employs a lightweight attention-enhanced convolutional neural network (CNN) for the automated classification of leukemia cells to one of the four categories: benign, early, pre-leukemic, and pro-leukemic at only 0.14 s/batch. The global model at 3 clients achieves 95.70% test accuracy while at 5 clients and increased training rounds achieve 96.56% on test set on a weighted aggregation method. Additionally, for increased clinical interpretability and transparency explainable methods are used in this study.
Data availability
The dataset used in this study is publicly available on Kaggle and was provided by Aria, M., Ghaderzadeh, M., Bashash, D., Abolghasemi, H., Asadi, F., and Hosseini, A. (2021). It is titled Acute Lymphoblastic Leukemia (ALL) Image Dataset and can be accessed via the following DOI:10.34740/KAGGLE/DSV/2175623.
References
Obeagu, E. I. Revolutionizing hematological disorder diagnosis: unraveling the role of artificial intelligence. Ann. Med. Surg. 87(6), 3445–3457 (2025).
Haque, R. et al. Advancing early leukemia diagnostics: a comprehensive study incorporating image processing and transfer learning. Bio. Med. Inf. 4(2), 966–991 (2024).
Kaissis, G. A., Makowski, M. R., Rückert, D. & Braren, R. F. Secure, privacy-preserving and federated machine learning in medical imaging. Nat. Mach. Intell. 2(6), 305–311 (2020).
Li, T., Sahu, A. K., Talwalkar, A. & Smith, V. Federated learning: challenges, methods, and future directions. IEEE Signal Process. Mag. 37(3), 50–60 (2020).
Sam, K., Nawaz, S. & Vavekanand, R. CardioMix: a multimodal image-based classification pipeline for enhanced ECG diagnosis. Med. Data Min. 8(1), 6 (2025).
Vavekanand, R., Sam, K., Kumar, S. & Kumar, T. Cardiacnet: a neural networks based heartbeat classifications using ecg signals. Stud. Med. Health Sci. 1(2), 1–17 (2024).
Vavekanand, R., Kumar, G. & Kurbanova, S. A lightweight physics-conditioned diffusion multi-model for medical image reconstruction. Biomed. Eng. Commun. 5, 12 (2026).
Rieke, N. et al. The future of digital health with federated learning. NPJ Digital Med. 3(1), 119 (2020).
Xu, J. et al. Federated learning for healthcare informatics. J. Healthcare Inf. Res. 5(1), 1–19 (2021).
Tong, S., Zuo, Z., Liu, Z., Sun, D. & Zhou, T. Hybrid attention mechanism of feature fusion for medical image segmentation. IET Image Proc. 18(1), 77–87 (2024).
Chen, J. et al. TransUNet: transformers make strong encoders for medical image segmentation (2021). arXiv preprint arXiv:2102.04306.
Al-Razgan, M., Ali, Y. A. & Awwad, E. M. Enhancing Fetal Medical Image Analysis through attention-guided convolution: a comparative study with established models. J. Disability Res. 3(2), 20240005 (2024).
Li, H., Nan, Y., Del Ser, J. & Yang, G. Large-kernel attention for 3D medical image segmentation. Cogn. Comput. 16(4), 2063–2077 (2024).
Antunes, R. S., André da Costa, C., Küderle, A., Yari, I. A. & Eskofier, B. Federated learning for healthcare: systematic review and architecture proposal. ACM Trans. Intell. Syst. Technol. 13(4), 1–23 (2022).
Sheller, Micah J. et al. Federated learning in medicine: facilitating multi-institutional collaborations without sharing patient data. Sci. Rep. 10(1), 12598 (2020).
Prayitno, S. et al. A systematic review of federated learning in the healthcare area: from the perspective of data properties and applications. Appl. Sci. 11(23), 11191 (2021).
Pfitzner, B., Steckhan, N. & Arnrich, B. Federated learning in a medical context: a systematic literature review. ACM Trans. Internet Technol. 21(2), 1–31 (2021).
Zhang, P., White, J., Schmidt, D. C. & Lenz, G. Applying software patterns to address interoperability in blockchain-based healthcare apps (2017). arXiv preprint arXiv:1706.03700.
Gondal, C. H. et al. Automated leukemia screening and sub-types classification using deep learning. Comput. Syst. Sci. Eng. 46, 3 (2023).
Shafique, S. & Tehsin, S. Computer? Aided diagnosis of acute lymphoblastic leukaemia. Comput. Math. Methods Med. 2018(1), 6125289 (2018).
Masoudi, B. VKCS: a pre-trained deep network with attention mechanism to diagnose acute lymphoblastic leukemia. Multimedia Tools Appl. 82(12), 18967–18983 (2023).
Shah, A. et al. Automated diagnosis of leukemia: a comprehensive review. IEEE Access 9, 132097–132124 (2021).
Kaissis, G. A., Makowski, M. R., Rückert, D. & Braren, R. F. Secure, privacy-preserving and federated machine learning in medical imaging. Nat. Mach. Intell. 2(6), 305–312 (2020).
Li, X. et al. Multi-site fMRI analysis using privacy-preserving federated learning and domain adaptation: ABIDE results. Med. Image Anal. 65, 101765 (2020).
Kalapaaking, A. P. et al. Smpc-based federated learning for 6g-enabled internet of medical things. IEEE Netw. 36(4), 182–189 (2022).
Falcetta, A. & Roveri, M. Privacy-preserving deep learning with homomorphic encryption: an introduction. IEEE Comput. Intell. Mag. 17(3), 14–25 (2022).
Wibawa, F., Catak, F. O., Kuzlu, M., Sarp, S., & Cali, U. Homomorphic encryption and federated learning based privacy-preserving cnn training: Covid-19 detection use-case. In Proceedings of the 2022 European Interdisciplinary Cybersecurity Conference 85–90 (2022).
Yi, F., Jeong, O. & Moon, I. Privacy-preserving image classification with deep learning and double random phase encoding. IEEE Access 9, 136126–136134 (2021).
Wang, G., Lu, R. & Guan, Y. L. Achieve privacy-preserving priority classification on patient health data in remote eHealthcare system. IEEE Access 7, 33565–33576 (2019).
Ito, H., Kinoshita, Y., Aprilpyone, M. & Kiya, H. Image to perturbation: an image transformation network for generating visually protected images for privacy-preserving deep neural networks. IEEE Access 9, 64629–64638 (2021).
Saha, C. et al. Lung-AttNet: an attention mechanism based CNN architecture for lung cancer detection with federated learning. IEEE Access (2025).
Lin, J. et al. A lightweight privacy-preserving federated learning framework for heterogeneity-resilient skin cancer diagnosis. IEEE J. Biomed. Health Inf. (2025).
Aria, M. et al. Acute Lymphoblastic Leukemia (ALL) Image Dataset (Kaggle, 2021).
Funding
This article has been produced with the financial support of the European Union under the LERCO project number \(CZ.10.03.01/00/22\_003/0000003\) via the Operational Programme Just Transition. This work was also supported by the Ministry of Education, Youth and Sports of the Czech Republic through the e-INFRA CZ (ID:90254).
Author information
Authors and Affiliations
Contributions
The authors confirm their contribution to the paper as follows: Conceptualization: MZA,NAK, and AJ; Methodology: MZA, NAK, and AJ; Software: PS, LR; Validation: AJ, PS; Formal analysis, MZA, and NAK; Investigation: MZA, PS; Resources: LR; Data Curation: MZA and NAK; Writing—original draft preparation: MZA,NAK, and PS; writing—review and editing: AJ and PS; Visualization: MZA; Supervision: LR; Project administration: AJ and LR; Funding acquisition: LR. All authors reviewed the results and approved the final version of the manuscript.
Corresponding author
Ethics declarations
Competing interests
The authors declare that they have no conflicts of interest to report regarding the present study.
Additional information
Publisher’s note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Open Access This article is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License, which permits any non-commercial use, sharing, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if you modified the licensed material. You do not have permission under this licence to share adapted material derived from this article or parts of it. The images or other third party material in this article are included in the article’s Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by-nc-nd/4.0/.
About this article
Cite this article
Awan, M.Z., Khan, N.A., Strakos, P. et al. Privacy-preserving federated learning with light-weight attention improved CNNs for automated leukemia detection across distributed medical imaging. Sci Rep (2026). https://doi.org/10.1038/s41598-026-40581-9
Received:
Accepted:
Published:
DOI: https://doi.org/10.1038/s41598-026-40581-9