公开(公告)号：CN116841762A

公开(公告)日：2023-10-03

申请号：CN202310395448.7

申请日：2023-04-14

Applicant: 西南科技大学

Inventor： 王耀彬 ,  张梦洋 ,  唐苹苹 ,  郭明峰 ,  胡丽莎

IPC: G06F9/54 ,  G06T1/40 ,  G06T1/60 ,  G06N3/04 ,  G06N3/08

Abstract: 本发明公开了一种图神经网络训练中的定长式边点结合采样机制，该采样机制包括如下步骤：对图进行节点采样，得到子图；对子图进行边采样，得到新的子图；将得到的子图数据转化为CSR格式；根据GPU共享内存大小，设置所需的共享内存宽度S；将子图数据存入到GPU共享内存之中；进行后续的GNN计算操作。本发明针对目前图神经网络训练中采集机制均无法达到快速并且准确的图特征采集的现状，提出一种图神经网络训练中的定长式边点结合采样机制，通过在节点采样时保留具有连续内存位置的邻居节点避免不规则的内存访问，结合边采样消除冗余的边，以便完成快速并且准确的图特征采集。