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

    公开(公告)号:CN114186598B

    公开(公告)日:2025-04-25

    申请号:CN202110856642.1

    申请日:2021-07-28

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

    一种基于阻变存储器的神经网络映射方法、加速器

    公开(公告)号:CN112070204A

    公开(公告)日:2020-12-11

    申请号:CN202010722047.4

    申请日:2020-07-24

    Abstract: 本发明提供一种基于阻变存储器的神经网络权重映射方法,其中,所述阻变存储器以存储单元的高阻态和低阻态存储二值数据,所述映射方法包括:S1、获得神经网络以二值形式表示的二进制权重阵列;S2、获得二进制权重值阵列的每一列的第一值和第二值的数量;S3、将二进制权重值阵列的每一列权重值映射存储到所述阻变存储器的每一列存储单元中,其中,对于第一值的数量大于第二值的数量的列,第一值映射为高阻态,第二值映射为低阻态;否则第二值映射为高阻态,第一值映射为低阻态。本发明在硬件上只是改变了原有的权重存储映射方式,有效的降低存储阵列中低电阻状态数量,大幅降低电流及其在阻变存储器计算阵列和模拟‑数字转换装置的功耗。

    一种基于阻变存储器的神经网络映射方法、加速器

    公开(公告)号:CN112070204B

    公开(公告)日:2023-01-20

    申请号:CN202010722047.4

    申请日:2020-07-24

    Abstract: 本发明提供一种基于阻变存储器的神经网络权重映射方法,其中,所述阻变存储器以存储单元的高阻态和低阻态存储二值数据,所述映射方法包括:S1、获得神经网络以二值形式表示的二进制权重阵列;S2、获得二进制权重值阵列的每一列的第一值和第二值的数量;S3、将二进制权重值阵列的每一列权重值映射存储到所述阻变存储器的每一列存储单元中,其中,对于第一值的数量大于第二值的数量的列,第一值映射为高阻态,第二值映射为低阻态;否则第二值映射为高阻态,第一值映射为低阻态。本发明在硬件上只是改变了原有的权重存储映射方式,有效的降低存储阵列中低电阻状态数量,大幅降低电流及其在阻变存储器计算阵列和模拟‑数字转换装置的功耗。

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

    公开(公告)号:CN114186598A

    公开(公告)日:2022-03-15

    申请号:CN202110856642.1

    申请日:2021-07-28

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

Patent Agency Ranking