公开(公告)号：CN117725971A

公开(公告)日：2024-03-19

申请号：CN202311795213.3

申请日：2023-12-25

Applicant: 电子科技大学

Inventor： 王勇 ,  罗光春 ,  王洁磊 ,  卢亮 ,  廖一傲

IPC: G06N3/063 ,  G06N3/042 ,  G06N3/0464 ,  G06N3/048 ,  G06N3/084 ,  G06N3/0895

Abstract: 本发明公开了一种DPU环境下的图卷积神经网络计算解耦方法及系统，该方法包括如下步骤：在DPU中配置特征变换单元、邻居聚合传播单元和融合单元；将原始GCN网络解耦为特征转换部分和邻居聚合传播部分，并分别部署到特征变换单元和邻居聚合传播单元，进行特征变换并实现传播计算，最终输出最后的聚合信息；将聚合信息导入融合单元，对所聚合的信息特征进行拼接，输出最终计算结果。本发明可在不降低模型效果的前提下，大幅减少GCN的计算时间，该方案可广泛应用于基于GCN的节点分类、目标识别等任务的工业场景。