计算机与数字工程2024，Vol.52Issue(2)：507-511,547,6.DOI:10.3969/j.issn.1672-9722.2024.02.039
基于生成对抗网络和对比学习的联邦学习方法
Federated Learning Method Based on Generative Adversarial Networks and Contrastive Learning
田耕1
作者信息
- 1. 零八一电子集团有限公司 成都 611700
- 折叠
摘要
Abstract
As a distributed learning mechanism,federated learning is mainly used to protect user data privacy and solve the problem of data islands.Existing research has problems such as the small scale of edge node data and the low accuracy of the model caused by the non-independent and identical distribution of data.Using the method of comparative learning can achieve the perfor-mance closest to the global model when the local model updates the model gradient parameters during the parameter aggregation up-date process of federated learning,which solves the problem of low accuracy of the edge model caused by the non-independent and identical distribution of data.The performance of marginal models is improved.Simulation results show that when the weight of the contrastive loss is 1,the accuracy of the model is increased by 1.05%on average,and when the weight of the contrastive loss is 10,the accuracy of the model is increased by 2.14%on average.Using the method of generative confrontation network to expand the data volume of edge nodes solves the problem of low model accuracy caused by the small amount of data of edge nodes.The simulation ex-periment results show that the accuracy of the model is increased by 0.76%and 1.12%on average after data expansion using the method based on the generative confrontation network.关键词
联邦学习/对比学习/生成对抗网络Key words
federated learning/contrastive learning/generative adversarial networks分类
信息技术与安全科学引用本文复制引用
田耕..基于生成对抗网络和对比学习的联邦学习方法[J].计算机与数字工程,2024,52(2):507-511,547,6.
