PSO-based Client Selection to Improve Medical Image Classification in Federated Learning

Leonardo Gabriel Ferreira Rodrigues; Larissa Ferreira Rodrigues Moreira; Rodrigo Moreira; André Ricardo Backes

doi:10.22456/2175-2745.150621

Authors

Leonardo Gabriel Ferreira Rodrigues Universidade Federal de Uberlândia (UFU)
Larissa Ferreira Rodrigues Moreira Universidade Federal de Viçosa (UFV)
Rodrigo Moreira Universidade Federal de Viçosa (UFV)
André Ricardo Backes Universidade Federal de São Carlos (UFSCar) https://orcid.org/0000-0002-7486-4253

DOI:

https://doi.org/10.22456/2175-2745.150621

Keywords:

federated learning, particle swarm optimization, image classification, client selection, deep learning

Abstract

Federated Learning (FL) offers an effective approach to enable collaborative model training across decentralized medical institutions while preserving patient privacy. However, FL faces challenges related to data heterogeneity and inefficient client participation issues. In this study, we introduce a novel client selection strategy based on Particle Swarm Optimization (PSO) to improve medical image classification in federated settings. Our method dynamically selects informative clients by evaluating their contributions to the global model, which reduces communication overhead and supports faster convergence. We performed experiments using three publicly available medical imaging datasets, Biglycan, ALL-IDB, and LC25000, focusing on binary classification tasks. The results show that the PSO-based approach improves performance over random client selection across all evaluated scenarios, with gains in accuracy, recall, precision, and F1-Score, even when selecting fewer clients. These findings indicate that incorporating PSO into FL frameworks helps address challenges related to data heterogeneity and communication efficiency while maintaining classification performance and privacy preservation.

Downloads

Download data is not yet available.

References

[1] MAHMOOD, T. et al. Recent advancements and future prospects in active deep learning for medical image segmentation and classification. IEEE Access, v. 11, p. 113623–113652, 2023.

[2] BYSANI, H. et al. Detection of pneumonia in chest x-ray using ensemble learners and transfer learning with deep learning models. In: 2023 14th International Conference on Computing Communication and Networking Technologies (ICCCNT). Delhi, India: IEEE, 2023. p. 1–8.

[3] MOREIRA, L. F. R. et al. Maximizing the Power of Cognitive Services with an AI-as-a-Service Architecture for Seamless Delivery. In: 2024 IEEE 13th International Conference on Cloud Networking (CloudNet). Rio de Janeiro, Brazil: IEEE, 2024. p. 1–8.

[4] CHEN, C.; Mat Isa, N. A.; LIU, X. A review of convolutional neural network based methods for medical image classification. Computers in Biology and Medicine, v. 185, p. 109507, 2025. ISSN 0010-4825.

[5] JR., L. A. S. et al. DeepCraftFuse: visual and deeply-learnable features work better together for esophageal cancer detection in patients with Barrett’s esophagus. Neural Computing and Applications, v. 36, n. 18, p. 10445–10459, Jun 2024. ISSN 1433-3058.

[6] GUAN, H. et al. Federated learning for medical image analysis: A survey. Pattern Recognition, v. 151, p. 110424, 2024. ISSN 0031-3203.

[7] BARBOSA, G. V. G. et al. Federated Learning in Breast Cancer Diagnosis. Revista de Informática Teórica e Aplicada, v. 32, n. 1, p. 173–179, Feb. 2025.

[8] RODRIGUES, L. G. F. et al. Medical Image Classification with Privacy: Centralized and Federated Learning Comparison. Revista de Informática Teórica e Aplicada, v. 32, n. 1, p. 180–187, Feb. 2025.

[9] de Souza, A. M. et al. Adaptive client selection with personalization for communication efficient Federated Learning. Ad Hoc Networks, v. 157, p. 103462, 2024. ISSN 1570-8705.

[10] AGARWAL, M. et al. Federated Learning With Dataset Splitting and Weighted Mean Using Particle Swarm Optimization. IEEE Access, v. 12, p. 161968–161981, 2024.

[11] MIAO, Y. et al. Fedsaf: A federated learning framework for enhanced gastric cancer detection and privacy preservation. arXiv preprint arXiv:2503.15870, 2025.

[12] NING, Z. et al. Fedgcs: A generative framework for efficient client selection in federated learning via gradient-based optimization. arXiv preprint arXiv:2405.06312, 2024.

[13] WU, X. et al. A Novel Centralized Federated Deep Fuzzy Neural Network with Multi-objectives Neural Architecture Search for Epistatic Detection. IEEE Transactions on Fuzzy Systems, v. 33, n. 1, p. 94–107, 2025.

[14] OUYANG, C. et al. Federated learning for extreme label noise: Enhanced knowledge distillation and particle swarm optimization. Electronics, MDPI AG, v. 14, n. 2, p. 366, 2025.

[15] SILVA NETO, P. C. d. et al. Breast cancer dataset with biomarker Biglycan. Data in Brief, v. 47, p. 108978, 2023. ISSN 2352-3409.

[16] Labati, R. D.; Piuri, V.; Scotti, F. All-IDB: The acute lymphoblastic leukemia image database for image processing. In: 2011 18th IEEE International Conference on Image Processing. Brussels, Belgium: IEEE, 2011. p. 2045–2048.

[17] BORKOWSKI, A. A. et al. Lung and Colon Cancer Histopathological Image Dataset (LC25000). 2019.

[18] LI, Z. et al. A survey of convolutional neural networks: Analysis, applications, and prospects. IEEE Transactions on Neural Networks and Learning Systems, v. 33, n. 12, p. 6999–7019, 2022.

[19] KOONCE, B. Squeezenet. In: Convolutional Neural Networks with Swift for Tensorflow: Image Recognition and Dataset Categorization. Berkeley, CA: Apress, 2021. p. 73–85. ISBN 978-1-4842-6168-2.

[20] KONEČNÝ, J. et al. Federated Optimization: Distributed Machine Learning for On-Device Intelligence. CoRR, abs/1610.02527, 2016.

[21] MCMAHAN, B. et al. Communication-Efficient Learning of Deep Networks from Decentralized Data. In: SINGH, A.; ZHU, J. (Ed.). Proceedings of the 20th International Conference on Artificial Intelligence and Statistics. Fort Lauderdale, FL, USA: PMLR, 2017. (Proceedings of Machine Learning Research, v. 54), p. 1273–1282.

[22] MOREIRA, R. et al. An intelligent native network slicing security architecture empowered by federated learning. Future Generation Computer Systems, v. 163, p. 107537, 2025. ISSN 0167-739X.

[23] KENNEDY, J.; EBERHART, R. Particle swarm optimization. In: Proceedings of ICNN’95 - International Conference on Neural Networks. Perth, WA, Australia: IEEE, 1995. v. 4, p. 1942–1948 vol.4.

[24] TAN, P.-N. et al. Introduction to Data Mining (2nd Edition). 2nd. ed. USA: Pearson, 2018. ISBN 0133128903.

[25] BALDIN, I. et al. FABRIC: A national-scale programmable experimental network infrastructure. IEEE Internet Computing, IEEE, v. 23, n. 6, p. 38–47, 2019.