Enhancing Alzheimer’s disease classification through split federated learning and GANs for imbalanced datasets

Abstract

In the rapidly evolving healthcare sector, using advanced technologies to improve medical classification systems has become crucial for enhancing patient care, diagnosis, and treatment planning. There are two main challenges faced in this domain (i) imbalanced distribution of medical data, leading to biased model performance and (ii) the need to preserve patient privacy and comply with data protection regulations. The primary goal of this project is to develop a medical classification model for Alzheimer’s disease detection that can effectively learn from decentralized and imbalanced datasets without compromising on data privacy. The proposed system aims to address these challenges by employing an approach that combines split federated learning (SFL) with conditional generative adversarial networks (cGANs) to enhance medical classification models. SFL enables efficient set of distributed agents that collaboratively train learning models without sharing their data, thus improving data privacy and the integration of conditional GANs aims to improve the model’s ability to generalize across imbalanced classes by generating realistic synthetic samples for minority classes. The proposed system provided an accuracy of approximately 83.54 percentage for the Alzheimer’s disease classification dataset.