Brain Tumor Detection in MRI Based on Federated Learning with YOLOv11
Journal:
arXiv
Published Date:
Mar 6, 2025
Abstract
One of the primary challenges in medical diagnostics is the accurate and
efficient use of magnetic resonance imaging (MRI) for the detection of brain
tumors. But the current machine learning (ML) approaches have two major
limitations, data privacy and high latency. To solve the problem, in this work
we propose a federated learning architecture for a better accurate brain tumor
detection incorporating the YOLOv11 algorithm. In contrast to earlier methods
of centralized learning, our federated learning approach protects the
underlying medical data while supporting cooperative deep learning model
training across multiple institutions. To allow the YOLOv11 model to locate and
identify tumor areas, we adjust it to handle MRI data. To ensure robustness and
generalizability, the model is trained and tested on a wide range of MRI data
collected from several anonymous medical facilities. The results indicate that
our method significantly maintains higher accuracy than conventional
approaches.