Invention Publication
- Patent Title: 基于忆阻器阵列的贝叶斯神经网络的变分推理方法和装置
-
Application No.: CN202210497666.7Application Date: 2022-05-09
-
Publication No.: CN114819128APublication Date: 2022-07-29
- Inventor: 吴华强 , 林钰登 , 高滨 , 唐建石 , 张清天 , 钱鹤
- Applicant: 清华大学
- Applicant Address: 北京市海淀区清华大学
- Assignee: 清华大学
- Current Assignee: 清华大学
- Current Assignee Address: 北京市海淀区清华大学
- Agency: 北京市柳沈律师事务所
- Agent 彭久云
- Main IPC: G06N3/063
- IPC: G06N3/063 ; G06N3/04 ; G06N3/08 ; G06N5/04 ; G06N7/00

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