基本信息:
- 专利标题： 一种基于阻变存储器的图神经网络计算方法和装置
- 申请号：CN202110856642.1 申请日：2021-07-28
- 公开(公告)号：CN114186598A 公开(公告)日：2022-03-15
- 发明人： 何银涛 , 王颖 , 李华伟 , 李晓维
- 申请人： 中国科学院计算技术研究所
- 申请人地址： 北京市海淀区中关村科学院南路6号
- 专利权人： 中国科学院计算技术研究所
- 当前专利权人： 中国科学院计算技术研究所
- 当前专利权人地址： 北京市海淀区中关村科学院南路6号
- 代理机构： 北京泛华伟业知识产权代理有限公司
- 代理人： 王勇
- 主分类号： G06K9/62
- IPC分类号： G06K9/62 ; G06N3/04 ; G06N3/08
摘要：
本发明实施例提供了一种基于阻变存储器的图神经网络计算方法和装置，该方法包括：对于图神经网络的任一层，分析该层中将要在阻变存储器阵列中运算的图数据在权重原位计算模式和混合原位计算模式下的处理时延相对大小，选择时延最小的模式作为该层的计算模式；在权重原位计算模式，对所述图神经网络的所述层将图数据的邻接矩阵和图神经网络的权重参数作为原位数据分别映射到相应的阻变存储器阵列中，以将图神经网络的节点特征作为输入数据与相应的原位数据进行运算；在混合原位计算模式，对所述图神经网络的所述层将图数据的邻接矩阵和节点特征作为原位数据分别映射到相应的阻变存储器阵列中，以将权重参数作为输入数据与相应的原位数据进行运算。
公开/授权文献:
- CN114186598B 一种基于阻变存储器的图神经网络计算方法和装置 公开/授权日：2025-04-25
IPC结构图谱:
| G | 物理 |
| --G06 | 计算；推算；计数 |
| ----G06K | 数据识别；数据表示；记录载体；记录载体的处理 |
| ------G06K9/00 | 用于阅读或识别印刷或书写字符或者用于识别图形，例如，指纹的方法或装置 |
| --------G06K9/62 | .应用电子设备进行识别的方法或装置 |