College of Public Education, ZheJiang Institute of Economics and Trade HangZhou, ZheJiang, China.
College of Business administration ZheJiang Institute of Economics and Trade HangZhou, ZheJiang, China.
PLoS One. 2024 Jul 11;19(7):e0303462. doi: 10.1371/journal.pone.0303462. eCollection 2024.
Nowadays, federated learning is one of the most prominent choices for making decisions. A significant benefit of federated learning is that, unlike deep learning, it is not necessary to share data samples with the model owner. The weight of the global model in traditional federated learning is created by averaging the weights of all clients or sites. In the proposed work, a novel method has been discussed to generate an optimized base model without hampering its performance, which is based on a genetic algorithm. Chromosome representation, crossover, and mutation-all the intermediate operations of the genetic algorithm have been illustrated with useful examples. After applying the genetic algorithm, there is a significant improvement in inference time and a huge reduction in storage space. Therefore, the model can be easily deployed on resource-constrained devices. For the experimental work, sports data has been used in balanced and unbalanced scenarios with various numbers of clients in a federated learning environment. In addition, we have used four famous deep learning architectures, such as AlexNet, VGG19, ResNet50, and EfficientNetB3, as the base model. We have achieved 92.34% accuracy with 9 clients in the balanced data set by using EfficientNetB3 as the base model using a GA-based approach. Moreover, after applying the genetic algorithm to optimize EfficientNetB3, there is an improvement in inference time and storage space by 20% and 2.35%, respectively.
如今,联邦学习是决策的最突出选择之一。联邦学习的一个显著优势是,与深度学习不同,它不需要与模型所有者共享数据样本。传统联邦学习中全局模型的权重是通过平均所有客户端或站点的权重来创建的。在提出的工作中,讨论了一种新的方法,即在不影响性能的情况下生成优化的基础模型,该方法基于遗传算法。染色体表示、交叉和突变——遗传算法的所有中间操作都用有用的示例进行了说明。应用遗传算法后,推理时间有了显著的提高,存储空间也大大减少。因此,模型可以轻松部署在资源受限的设备上。对于实验工作,在联邦学习环境中使用了具有各种客户端数量的平衡和不平衡场景下的运动数据。此外,我们还使用了四种著名的深度学习架构,如 AlexNet、VGG19、ResNet50 和 EfficientNetB3,作为基础模型。我们使用基于遗传算法的方法,使用 EfficientNetB3 作为基础模型,在平衡数据集上实现了 9 个客户端时达到了 92.34%的准确率。此外,在对 EfficientNetB3 进行遗传算法优化后,推理时间和存储空间分别提高了 20%和 2.35%。