有效

脉冲神经网络权重梯度计算方法及相关设备

周晖晖、李明静、田永鸿、崔小欣、李国齐
鹏城实验室

摘要

本申请提供了一种脉冲神经网络权重梯度计算方法,包括:获取目标脉冲神经网络中每个通道维度的脉冲变化信号以及每个卷积核对应的膜电位梯度变化信息;针对每个脉冲变化信号,分别在多个时钟滑窗内按照时钟周期读取对应的脉冲信号值;在当前时钟周期读取的多个脉冲信号值不均为0的情况下,读取多个膜电位梯度变化信息中的梯度变化值,将多个梯度变化值按行输入至预设的累加器阵列,并将多个脉冲信号值输入至预设的多个选择器,以通过多个选择器按列控制累加器阵列中多个累加器将梯度变化值累加至上一时钟周期的梯度累加结果上,得到当前时钟周期的多个梯度累加结果,进而得到每个卷积核的权重梯度信息,能够有效避免无效计算,提高计算效率。