arrow
返回

A Graph Neural Network Based Decentralized Learning Scheme

delete2022-01-28
delete1
delete
OA
AI
H
Huiguo Gao
M
Mengyuan Lee
G
Guanding Yu *
Z
Zhaolin Zhou
DOI:10.3390/s22031030delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
As an emerging paradigm considering data privacy and transmission efficiency, decentralized learning aims to acquire a global model using the training data distributed over many user devices. It is a challenging problem since link loss, partial device participation, and non-independent and identically distributed (non-iid) data distribution would all deteriorate the performance of decentralized learning algorithms. Existing work may restrict to linear models or show poor performance over non-iid data. Therefore, in this paper, we propose a decentralized learning scheme based on distributed parallel stochastic gradient descent (DPSGD) and graph neural network (GNN) to deal with the above challenges. Specifically, each user device participating in the learning task utilizes local training data to compute local stochastic gradients and updates its own local model. Then, each device utilizes the GNN model and exchanges the model parameters with its neighbors to reach the average of resultant global models. The iteration repeats until the algorithm converges. Extensive simulation results over both iid and non-iid data validate the algorithm's convergence to near optimal results and robustness to both link loss and partial device participation.
Keyword:
decentralized learning
graph neural network
average consensus
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Sensors 封面图
Sensors
IF:
3.5
论文数:
7.2W
被引数:
20.9W

机构

Z
zhejiang university
学者数:
17.7W
论文数: 12.1W
被引数: 152
引用论文

引用论文

Scheduling for Cellular Federated Edge Learning With Importance and Channel Awareness
err2020-11-01
err170
errOAAI
errRen, Jinke; He, Yinghui; Wen, Dingzhu; Yu, Guanding; Huang, Kaibin; Guo, Dongning
err分享
err收藏
err分享
err收藏
err分享
err收藏
学者 查看更多内容