发明公开
- 专利标题: 基于忆阻器阵列的贝叶斯神经网络的变分推理方法和装置
-
申请号: CN202210497666.7申请日: 2022-05-09
-
公开(公告)号: CN114819128A公开(公告)日: 2022-07-29
- 发明人: 吴华强 , 林钰登 , 高滨 , 唐建石 , 张清天 , 钱鹤
- 申请人: 清华大学
- 申请人地址: 北京市海淀区清华大学
- 专利权人: 清华大学
- 当前专利权人: 清华大学
- 当前专利权人地址: 北京市海淀区清华大学
- 代理机构: 北京市柳沈律师事务所
- 代理商 彭久云
- 主分类号: G06N3/063
- IPC分类号: G06N3/063 ; G06N3/04 ; G06N3/08 ; G06N5/04 ; G06N7/00
摘要:
一种基于忆阻器阵列的贝叶斯神经网络的变分推理方法和装置。忆阻器阵列包括多个忆阻器,贝叶斯神经网络的经训练得到的权重矩阵被映射至忆阻器阵列中,该方法包括:对于被映射后的忆阻器阵列,获得当前忆阻器阵列的多个忆阻器的电导态;获取对于贝叶斯神经网络的权重矩阵的每个权重的多个权重采样样本,并使用忆阻器阵列进行多次前向传播以得到多个输出结果;基于每个权重的多个权重采样样本和多个输出结果,获取贝叶斯神经网络的损失函数的损失值;对损失函数进行反向传播,得到忆阻器阵列中用于贝叶斯神经网络的权重矩阵的每个权重的忆阻器的电导态的梯度;根据梯度更新忆阻器阵列中用于贝叶斯神经网络的权重矩阵的忆阻器的电导态。