An efficient federated learning method based on enhanced classification-GAN for medical image classification
Wei Liu,Yurong Zheng,Zhihui Xiang,Yingmeng Wang,Zhao Tian,Wei She
DOI: https://doi.org/10.1007/s00530-024-01564-w
IF: 3.9
2024-12-15
Multimedia Systems
Abstract:The scarcity of medical images significantly hampers the advancement of artificial intelligence techniques in the medical field. Medical images face the issue of inferior training accuracy and efficiency in classification tasks due to insufficient labeled data and privacy preserving demands. To address these issues, we propose a new and efficient federated learning (FL) scheme based on blockchain, called FedBG, to generate realistic images to expand the dataset for medical image classification. First, we design an enhanced classification-GAN (EC-GAN) algorithm based on cross-entropy, which expands the training target of the generator from a single improvement of image fidelity to a direction related to the classification tasks. EC-GAN increases the diversity of generated images and reduces the risk of pattern collapse. Second, we improve the generator's loss function, which is guided by the cross-entropy loss term of the classifier model and the gradient information from the discriminator model, to generate medical images with enhanced category features. This design overcomes the limitation of traditional GAN that focuses only on image authenticity judgment, and achieves the dual optimization of synthetic image and classification accuracy. Finally, we propose an improved consensus mechanism based on maximum mean discrepancy and data contribution to ensure the consistency and security of data in the blockchain. The mechanism optimizes the fairness and efficiency of the model training by dynamically evaluating the contribution of each participating node, and guarantees the trust system of FL. Experiments are performed on two datasets and results demonstrate that FedBG reduces the training time by 27–38% and improves the accuracy by 0.9–2% compared to existing methods while ensuring data privacy and generating high-quality images.
computer science, information systems, theory & methods