Federated Learning Framework for Brain Tumor Detection Using MRI Images in Non-IID Data Distributions

基于非独立同分布数据中MRI图像的脑肿瘤检测联邦学习框架

阅读:1

Abstract

Brain tumor detection from medical images, especially magnetic resonance imaging (MRI) scans, is a critical task in early diagnosis and treatment planning. Traditional machine learning approaches often rely on centralized data, raising concerns about data privacy, security, and the difficulty of obtaining large annotated datasets. Federated learning (FL) has emerged as a promising solution for training models across decentralized devices while maintaining data privacy. However, challenges remain in dealing with non-IID (independent and identically distributed) data, which is common in real-world scenarios. In this research, we used a client-server-based federated learning framework for brain tumor detection using MRI images, leveraging VGG19 as the backbone model. To improve clinical relevance and model interpretability, we have included explainability techniques, particularly Grad-CAM. We trained our model across four clients with non-IID data distribution to simulate real-world conditions. For performance evaluation, we used a centralized test dataset, consisting of 20% of the original data, with the test set used collectively for evaluating model performance after completing federated learning rounds. Using a separate test dataset ensures that all models are evaluated on the same data, making comparisons fair. Since the test dataset is not part of the FL training process, it does not violate the privacy-preserving nature of FL. The experimental results demonstrate that the VGG19 model achieves a high test accuracy of 97.18% (FedAVG), 98.24% (FedProx), and 98.45% (Scaffold) than other state-of-the-art models, showcasing the effectiveness of federated learning in handling distributed and non-IID data. Our findings highlight the potential of federated learning to address privacy concerns in medical image analysis while maintaining high performance even in non-IID settings. This approach provides a promising direction for future research in privacy-preserving AI for healthcare applications.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。