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