arrow
返回

Distributed Optimization of Graph Convolutional Network Using Subgraph Variance

delete2024-08-01
delete5
delete
OA
AI
T
Taige Zhao
宋
宋翔宇 (Xiangyu Song)
M
Man Li
李建新 封面图
李建新 (Jianxin Li) *
罗玮 封面图
罗玮 (Wei Luo)
I
Imran Razzak
DOI:10.1109/TNNLS.2023.3243904delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
In recent years, distributed graph convolutional networks (GCNs) training frameworks have achieved great success in learning the representation of graph-structured data with large sizes. However, existing distributed GCN training frameworks require enormous communication costs since a multitude of dependent graph data need to be transmitted from other processors. To address this issue, we propose a graph augmentation-based distributed GCN framework (GAD). In particular, GAD has two main components: GAD-Partition and GAD-Optimizer. We first propose an augmentation-based graph partition (GAD-Partition) that can divide the input graph into augmented subgraphs to reduce communication by selecting and storing as few significant vertices of other processors as possible. To further speed up distributed GCN training and improve the quality of the training result, we design a subgraph variance-based importance calculation formula and propose a novel weighted global consensus method, collectively referred to as GAD-Optimizer. This optimizer adaptively adjusts the importance of subgraphs to reduce the effect of extra variance introduced by GAD-Partition on distributed GCN training. Extensive exper-iments on four large-scale real-world datasets demonstrate that our framework significantly reduces the communication overhead (asymptotic to 50%), improves the convergence speed (asymptotic to 2x) of distributed GCN training, and obtains a slight gain in accuracy (asymptotic to 0.45%) based on minimal redundancy compared to the state-of-the-art methods.
Keyword:
Communicate reduction
distribution optimize
graph augmentation
graph convolutional network (GCN)
sub-graph variance

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

D
Deakin University
学者数:
2.0W
论文数: 2.1W
被引数: 2.8W
引用论文

引用论文

vPipe: A Virtualized Acceleration System for Achieving Efficient and Scalable Pipeline Parallel DNN Training
err2022-03-01
err22
errOAAI
errZhao, Shixiong; Li, Fanxin; Chen, Xusheng; Guan, Xiuxian; Jiang, Jianyu; Huang, Dong; Qing, Yuhao; Wang, Sen; Wang, Peng; Zhang, Gong; Li, Cheng; Luo, Ping; Cui, Heming
err分享
err收藏
Improved bounds on sorting by length-weighted reversals改进了按长度加权反转排序的界限
err2008-08-01
err0
errOAAI
errMichael A. Bender; Dongdong Ge; Simai He; Haodong Hu; Ron Y. Pinter; Steven Skiena; Firas Swidan
err分享
err收藏
学者 查看更多内容