基于忆阻器阵列的贝叶斯神经网络的变分推理方法和装置
摘要:
一种基于忆阻器阵列的贝叶斯神经网络的变分推理方法和装置。忆阻器阵列包括多个忆阻器,贝叶斯神经网络的经训练得到的权重矩阵被映射至忆阻器阵列中,该方法包括:对于被映射后的忆阻器阵列,获得当前忆阻器阵列的多个忆阻器的电导态;获取对于贝叶斯神经网络的权重矩阵的每个权重的多个权重采样样本,并使用忆阻器阵列进行多次前向传播以得到多个输出结果;基于每个权重的多个权重采样样本和多个输出结果,获取贝叶斯神经网络的损失函数的损失值;对损失函数进行反向传播,得到忆阻器阵列中用于贝叶斯神经网络的权重矩阵的每个权重的忆阻器的电导态的梯度;根据梯度更新忆阻器阵列中用于贝叶斯神经网络的权重矩阵的忆阻器的电导态。
0/0