一种基于阻变存储器的图神经网络计算方法和装置专利查询专利号|摘要-满商公司网

一种基于阻变存储器的图神经网络计算方法和装置

基本信息

申请号	CN202110856642.1	申请日	-
公开（公告）号	CN114186598A	公开（公告）日	2022-03-15
申请公布号	CN114186598A	申请公布日	2022-03-15
分类号	G06K9/62（2022.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I	分类	计算；推算；计数;
发明人	何银涛;王颖;李华伟;李晓维	申请（专利权）人	中国科学院计算技术研究所
代理机构	北京泛华伟业知识产权代理有限公司	代理人	王勇
地址	100190北京市海淀区中关村科学院南路6号
法律状态	-

摘要

本发明实施例提供了一种基于阻变存储器的图神经网络计算方法和装置，该方法包括：对于图神经网络的任一层，分析该层中将要在阻变存储器阵列中运算的图数据在权重原位计算模式和混合原位计算模式下的处理时延相对大小，选择时延最小的模式作为该层的计算模式；在权重原位计算模式，对所述图神经网络的所述层将图数据的邻接矩阵和图神经网络的权重参数作为原位数据分别映射到相应的阻变存储器阵列中，以将图神经网络的节点特征作为输入数据与相应的原位数据进行运算；在混合原位计算模式，对所述图神经网络的所述层将图数据的邻接矩阵和节点特征作为原位数据分别映射到相应的阻变存储器阵列中，以将权重参数作为输入数据与相应的原位数据进行运算。