Federated Learning Combined Ensemble Aggregation for Brain Tumor Classification in Magnetic Resonance Imaging

In recent years, the use of deep learning techniques in medical imaging has shown promising results, particularly in the classification of brain tumors from Magnetic Resonance Imaging (MRI) scans. This article proposes an innovative approach that combines federated learning (FL) with convolutional neural networks (CNNs) and ensemble aggregation to enhance the accuracy of MRI brain tumor image classification. The proposed method utilizes CNN architectures that are fine-tuned on local datasets at different client sites. The results from these models are then aggregated using ensemble aggregation techniques at a central server and utilize a meta-learner to determine optimal weights for client models based on their validation performance, and aggregate model parameters using weighted averaging. Next, the improved model weights are sent back to the clients for further training. This approach not only preserves data privacy but also enhances model robustness. Experimental results demonstrate that the proposed method outperforms traditional centralized training methods, achieving higher accuracy and better generalization on unseen data.

کلیدواژه ها:

Brain tumor ، Deep Learning ، Ensemble ، Federated Learning ، Magnetic resonance imaging ، Medical Image