Decentralized Diagnosis: Privacy-Preserving Brain Tumor Classification with Federated Learning

Published: 29 Jun 2024, Last Modified: 26 Jul 2024KDD-AIDSH 2024 PosterEveryoneRevisionsBibTeXCC BY 4.0
Keywords: Brain Tumor, Federated learning, Classification, Data privacy, Deep learning, Medical imaging, Machine learning
TL;DR: Using federated learning, this study trained a VGG16-based CNN model for brain tumor identification while preserving data privacy. The model achieved comparable performance to a centralized system within the federated learning framework
Abstract: Brain tumors pose a significant global health challenge, driving ongoing research advancements in early detection methods. Artificial intelligence (AI) and deep learning (DL) techniques have shown great potential in this field, enabling the creation of highly accurate models for brain tumor identification from medical images. However, centralized approaches to these methods often raise critical concerns regarding patient data privacy and security. This paper presents a novel federated learning (FL) framework for brain tumor identification that effectively addresses these privacy concerns. FL enables collaborative model training across multiple institutions without the need for raw data sharing. Each participating institution trains the model locally on their Magnetic Resonance Imaging (MRI) datasets and only transmits model updates to a central server for secure aggregation. This iterative process results in a robust global model trained on a distributed dataset while preserving patient data confidentiality. The proposed FL model is evaluated using a dataset of 3,000 MRI images. Experimental results demonstrate the effectiveness of our approach, achieving a high accuracy rate of 96.88% for brain tumor identification. These findings suggest that FL provides a viable solution for privacy-preserving brain tumor identification, maintaining comparable performance to centralized models while ensuring the security of patient data.
Submission Number: 38
Loading