Federated learning with swarm intelligence for efficient and secure medical image analysis

Abstract

Collaborative learning in healthcare faces challenges, including strict regulations and fragmented data. This research introduces a federated learning framework that employs swarm intelligence to augment communication and enhance the analysis of medical images. The method optimizes hyperparameters, selects features, and assigns aggregation weights to federated clients simultaneously by combining Particle Swarm Optimization (PSO) and the Firefly Algorithm (FA) with deep Convolutional Neural Networks (CNNs). The framework was tested on three medical datasets: COVID-19 chest X-rays (5,856 images), monkeypox skin images (569 images), and breast cancer mammograms (320 images). These datasets were shared among four fake healthcare institutions. It strives to strike a balance between privacy, communication costs, and classification accuracy. The results showed that the test was 96.71% accurate in detecting COVID-19, 96.06% accurate in classifying monkeypox, and 97.0% accurate in diagnosing breast cancer. The framework was able to handle noise and attacks from individuals who sought to disrupt it, which reduced communication rounds by 25-30%. A privacy-utility analysis revealed that there were acceptable trade-offs, with accuracy remaining above 94%. This study employs robust privacy measures and statistical validation. It also shows how to use medical AI in smaller healthcare settings without putting patients' privacy at risk.