有效

基于神经网络转换的任务推理方法、装置、设备及介质

余肇飞、黄梓涵、黄铁军
北京大学

摘要

本申请公开了一种基于神经网络转换的任务推理方法、装置、设备及介质。包括:获取预训练的人工神经网络,所述人工神经网络中包含多个使用ReLU激活函数的神经元层;选取目标数据集输入所述人工神经网络,分别统计每个神经元在所述目标数据集上的均值和方差;基于所述均值和方差,计算每个神经元的最优脉冲发放阈值;基于所有神经元的最优脉冲发放阈值,将所述人工神经网络转换为脉冲神经网络;基于所述转换后的脉冲神经网络执行任务推理。本申请可以确定神经元的最优脉冲发放阈值,基于确定的最优阈值将人工神经网络转换为脉冲神经网络,显著提升转换后脉冲神经网络的任务推理速度与性能。

暂无引用专利