计算机工程与应用 ›› 2023, Vol. 59 ›› Issue (1): 269-277.DOI: 10.3778/j.issn.1002-8331.2204-0165

• 大数据与云计算 • 上一篇    下一篇

非独立同分布数据下的去中心化联邦学习策略

谭荣杰,洪智勇,余文华,曾志强   

  1. 五邑大学 智能制造学部,广东 江门 529020
  • 出版日期:2023-01-01 发布日期:2023-01-01

Decentralized Federated Learning Strategy for Non-Independent and Identically Distributed Data

TAN Rongjie, HONG Zhiyong, YU Wenhua, ZENG Zhiqiang   

  1. Faculty of Intelligent Manufacturing, Wuyi University, Jiangmen, Guangdong 529020, China
  • Online:2023-01-01 Published:2023-01-01

摘要: 区块链具有不可篡改性和去中心化的特点,其与联邦学习的结合成为人工智能领域的热门主题。目前去中心化联邦学习存在训练数据非独立同分布导致的性能下降问题,为了解决这个问题,提出一种模型相似度的计算方法,然后设计一种基于该模型相似度的去中心化联邦学习策略,并使用五个联邦学习任务进行测试,分别是CNN模型训练fashion-mnist数据集、alexnet模型训练cifar10数据集、TextRnn模型训练THUsnews数据集、Resnet18模型训练SVHN数据集和LSTM模型训练sentiment140数据集。实验结果表明,设计的策略在五个任务非独立同分布的数据下进行去中心化联邦学习,准确率分别提升了2.51、5.16、17.58、2.46和5.23个百分点。

关键词: 区块链, 联邦学习, 深度学习, 数据隐私

Abstract: Blockchain has the characteristics of immutability and decentralization, and its combination with federated learning has become a hot topic in the field of artificial intelligence. At present, decentralized federated learning has the problem of performance degradation caused by non-IID distribution of training data. In order to solve this problem, the paper firstly proposes a calculation method of model similarity, and then designs a decentralized federated learning strategy based on the similarity of the model. The learning strategy is tested using five federated learning tasks, using the CNN model to train the fashion-mnist dataset, the alexnet model to train the cifar10 dataset, the TextRnn model to train the THUsnews dataset, the Resnet18 model to train the SVHN dataset and the LSTM model to train the sentiment140 dataset. The experimental results show that the designed strategy performs decentralized federated learning under the non-IID data of the five tasks, and the accuracy rates are increased by 2.51, 5.16, 17.58, 2.46 and 5.23?percentage points respectively.

Key words: blockchain;federated learning, deep learning, data privacy