Privacy-Preserving Multi-Class Brain Tumor Classification Using Federated Learning And Deep Capsule Networks

Abstract

Currently, the Artificial Intelligence (AI) have been widely used in medical imaging for accurate disease classification. However, the currently existing technologies offer centralized learning environments that make them prone to security risks. These security risks may hinder the accuracy of disease classification, leading to a reduced survival rate. To address these issues, a novel multi- class brain tumor classification strategy was proposed in this work using federated learning. The proposed approach integrates the features of Federated Learning with the Deep CapsNet approach (FL-DCN), offering both accuracy and privacy. The proposed framework includes modules: data collection and preprocessing, feature extraction, and classification. In the data collection and preprocessing module, the brain Magnetic Resonance Imaging (MRI) images are collected and preprocessed to ensure consistency and quality. Also, in this stage the collected medical images were effectively pre-processed and enhance the generalizability. In the feature extraction module, a U-Net architecture was employed for extracting the most significant attributes present in the images. Here, dual outcome structure was enabled to implemented the segmentation and feature extraction process simultaneously. The outcomes of this stage is high level feature and tumor masks for CapsNet. In the classification module, a federated learning environment was created and the extracted features are distributed across multiple local nodes. Here, each node has trained three elements such as primary capsules, digit capsules and dynamic routing. Moreover, primary capsules contains 32 capsules and 8D activation vectors; similarly digit capsules consists 4 capsules and 16D vectors for used to match the tumor classes. Consequently, the local node uses Deep CapsNet to learn the spatial and hierarchical patterns within the images. Finally, a Federated Averaging (FedAvg) was employed to combine the model parameters of the local models into a global model, which produces the classification outcomes. The presented methodology is implemented in the Python tool and validated across the public database. The implementation results are determined and validated across the existing technologies in terms of accuracy, precision, recall, f-measure, error rate, and computational efficiency. While comparing the conventional model this study demonstrates that the FL framework can attain privacy protected brain tumor classification from brain MRI image dataset without compromising much accuracy as well as precision parameters.