【技术实现步骤摘要】
一种缓解忆阻加速器非理想因素的方法及装置
[0001]本申请属于电子信息
,特别是涉及一种缓解忆阻加速器非理想因素的方法及装置。
技术介绍
[0002]神经网络算法在许多领域取得了突破性的成功,帮助构建了图像识别、目标检测、语音识别等技术的基础。随着神经网络模型规模的进一步增大,以及随着神经网络模型的部署逐渐下沉至边缘设备,神经网络计算的硬件实现方案同时对高算力和低功耗提出了要求。传统冯诺依曼架构处理器的访存瓶颈对计算速度造成限制,且处理神经网络这样的大数据规模的任务所需的频繁访存也带来较大的功耗。考虑到忆阻器的低功耗、高存储密度特性,以及忆阻器交叉阵列的存内计算特性可以有效缓解访存压力,且忆阻器交叉阵列的电压电流关系与神经网络计算大量使用的向量矩阵乘法运算高度匹配,因此使用忆阻器交叉阵列构建的加速器代替传统处理器来加速神经网络计算成为一个很有竞争力的方案。
[0003]然而忆阻器交叉阵列这样的模拟电路的非理想因素可能会严重影响其存储及计算的可靠性,例如忆阻器的器件偏差(Device Variation),即忆阻器件在制造过程中,由于器件长度和宽度、氧化物厚度和表面粗糙度都存在差异,不同的忆阻器的不同阻态的实际阻值会相应地偏离该阻态的标准阻值一定范围;再例如忆阻器的固定型故障(Stuck At Fault,SAF),即由于单元永久损坏或选择器导线损坏等因素导致忆阻器单元始终处于高阻状态,或者由于短路等缺陷引起忆阻器单元始终处于低阻状态,从而导致忆阻器无法被编程写入需要的阻值;再例如忆阻器阻值漂移(Mem ...
【技术保护点】
【技术特征摘要】
1.一种缓解忆阻加速器非理想因素的方法,其特征在于,所述方法包括:获取已训练好的神经网络中卷积层和/或全连接层的浮点值表示的原始权重、映射各层权重的量化位宽以及忆阻器表示精度;根据所述量化位宽和所述忆阻器表示精度,量化所述原始权重,得到第一理想硬件级权重;硬件级权重表征权重参数量化后的整数值由单个或多个忆阻器单元各自保存的数值和所保存数值的位权;将所述第一理想硬件级权重映射到所述忆阻器单元中,得到第一非理想因素硬件级权重;根据所述第一非理想因素硬件级权重,重建浮点值表示的第一非理想因素权重;对所述第一非理想因素权重进行处理,得到浮点值表示的抗非理想因素权重;根据所述抗非理想因素权重,得到第二理想硬件级权重,将所述第二理想硬件级权重,重映射到所述忆阻器单元中。2.根据权利要求1所述的方法,其特征在于,所述方法还包括:在所述忆阻加速器运行的过程中,每隔预设时间和/或每隔预设计算次数,将所述第二理想硬件级权重,重映射到所述忆阻器单元中。3.根据权利要求1所述的方法,其特征在于,对所述第一非理想因素权重进行处理,得到浮点值表示的抗非理想因素权重,包括:以所述第一非理想因素权重作为所述神经网络权重的初始值,进行重新训练;当重新训练得到的第一训练权重收敛,以所述第一训练权重作为第一抗非理想因素权重。4.根据权利要求1所述的方法,其特征在于,对所述第一非理想因素权重进行处理,得到浮点值表示的抗非理想因素权重,包括:以所述第一非理想因素权重作为所述神经网络权重的初始值,作反向传播训练,得到第二训练权重;计算所述反向传播训练中添加的第一偏移量,所述第一偏移量为所述第二训练权重与所述第一非理想因素权重的差值;以所述原始权重和所述第一偏移量之和,作为第二抗非理想因素权重。5.根据权利要求1所述的方法,其特征在于,对所述第一非理想因素权重进行处理,得到浮点值表示的抗非理想因素权重,包括:以所述第一非理想因素权重作为所述神经网络权重的初始值,进行重新训练;当重新训练得到的第三训练权重收敛,根据所述第三训练权重,得到第三理想硬件级权重;将所述第三理想硬件级权重映射到所述忆阻器单元中,得到第二非理想因素硬件级权重;根据所述第二非理想因素硬件级权重,重建浮点值表示的第二非理想因素权重;以所述第二非理想因素权重作为所述神经网络权重的初始值,作反向传播训练,得到第四训练权重;计算所述反向传播训练中添加的第二偏移量,所述第二偏移量为所述第四训练权重与所述第二非理想因素权重的差值;
以所述第三训练权重和所述第二偏移量之和,作为第三抗非理想因素权重。6.根据权利要求1
‑
5任一所述的方法,其特征在于,对权重进行量化处理包括:获取该层浮点值表示的权重绝对值的最大值max_abs;根据该层的量化位宽,获取用于表示所述权重绝对值的位可表示的最大整数max_int,其中:max_int=2
qbits
‑1‑
1该式中,qbits表示所述量化位宽;获取最小量化单位Δ,其中:将该层中每一个浮点值表示的权重参数x,量化为相应的整数值Q(x),其中:该式中,round(.)是取整函数,用...
【专利技术属性】
技术研发人员:梁峰,卞鼐,李冰,梁辉,张洁,李佩嵘,张国和,
申请(专利权)人:西安交通大学,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。