公开(公告)号：CN117574987A

公开(公告)日：2024-02-20

申请号：CN202311635371.2

申请日：2023-12-01

Applicant: 厦门大学

Inventor： 张一鸣 ,  施杰 ,  吴荣鑫 ,  沈志荣 ,  李乔 ,  高聪明 ,  舒继武

IPC: G06N3/082 ,  G06N3/084 ,  G06N3/063 ,  G06N3/0499

Abstract: 本公开提供了一种加速分布式图神经网络训练的方法、电子设备及可读存储介质。本公开的加速分布式图神经网络训练的方法，包括：将图数据分割成n个子图G1,…,Gi,…,Gn；在第一周期，抽取子图Gi的内部节点及其节点特征，得到压缩子图Gi'，从第二周期开始，对上一周期得到的第一压缩子图的节点特征进行优化，得到本周期的优化压缩子图Gi”；根据本周期的第一压缩子图的节点特征，得到本周期的第一压缩子图的节点之间的边预测；对于子图Gj，根据子图Gi、本周期的第二压缩子图和本周期的第二压缩子图的节点之间的边预测进行训练，得到第一梯度；通过同步所有子图的第一梯度，更新第一训练参数。