一种基于类脑学习的图像识别方法技术

技术编号:34961530 阅读:41 留言:0更新日期:2022-09-17 12:41
一种基于类脑学习的图像识别方法,对原始图片数据进行预处理得到输入向量;建立脉冲神经网络模型进行学习,模型包括输入层、类脑学习算子和输出层;先对所述输入向量进行脉冲时序编码得到脉冲神经网络的输入数据,再将所述输入数据输入到脉冲神经网络的输入层,然后通过类脑学习算子学习输入向量的特征,最后将脉冲神经网络的输出层最先激发的脉冲神经元作为网络识别的类别,经过多次训练学习后,得到最后的识别网络;将待识别的原始图片数据进行预处理,用识别网络进行图片识别。本发明专利技术在脉冲神经网络中使用类脑学习算子,优化了脉冲神经网络计算能耗高,可解释性差的缺陷,符合对类脑网络的预期。类脑网络的预期。类脑网络的预期。

【技术实现步骤摘要】
一种基于类脑学习的图像识别方法


[0001]本专利技术属于图像识别
,同时涉及类脑智能与机器学习
;特别涉及一种基于类脑学习的图像识别方法。

技术介绍

[0002]在现在发展飞快的社会,模式识别是人工智能的一个重要任务。其中最常用的就是图像识别。现有的图像识别技术一般是通过人工神经网络来实现的,主要是通过反向传播算法搭配人工神经网络来实现。但反向传播算子有一些缺陷:首先,反向传播学习算法需要占用大量的计算资源;其次,这种反向传播的方式并没有在生物大脑中被观察到,其结构、计算方式和学习规则与大脑存在根本差异;最后,通过人工神经网络进行图像识别的过程是没有可解释性的。同时,人工神经网络也有其他缺点,首先是人工神经网络解决图像识别任务结构过于复杂,能耗较高;其次是人工神经元并非仿生神经元,不具有生物可解释性。

技术实现思路

[0003]为了解决现有方法的缺点,本专利技术通过了一种基于类脑学习的图像识别方法,采用类脑学习算子搭配脉冲神经网络来完成图像识别,这种类脑学习算子以强化学习为基础,通过仿真实现大脑内部的奖励学习机制来达到学习和识别的功能。而类脑学习算子也和脉冲神经网络原生的学习算子不同,类脑学习算子在全局信用分配、能耗降低和生物可解释性上都比原生的学习算子更加优越。
[0004]为了实现上述目的,本专利技术采用的技术方案是:
[0005]一种基于类脑学习的图像识别方法,其特征在于,包括如下步骤:
[0006]步骤1:对原始图片数据进行预处理得到输入向量;所述原始图片数据指训练脉冲神经网络所用的训练数据,其包括了图片和标签两个部分;
[0007]步骤1.1:将原始输入图片直接放入三层卷积神经网络中进行卷积处理,得到大小为24
×
24的处理后的图片;
[0008]步骤1.2:将处理后的图片送入三层卷积神经网络的池化层中进行池化处理,得到1
×
576的行向量,将该行向量作为输入向量。
[0009]步骤2:建立脉冲神经网络模型进行学习,所述脉冲神经网络模型包括输入层、类脑学习算子和输出层;先对所述输入向量进行脉冲时序编码得到脉冲神经网络的输入数据,再将所述输入数据输入到脉冲神经网络的输入层,然后通过类脑学习算子学习输入向量的特征,最后将脉冲神经网络的输出层最先激发的脉冲神经元作为网络识别的类别,经过多次训练学习后,得到最后的识别网络;
[0010]步骤2.1:构建两层脉冲神经网络,将步骤1得到的输入向量经过脉冲时序编码后作为脉冲神经网络的输入;
[0011]步骤2.2:利用类脑学习算子学习输入向量的特征,经过多轮训练学习后得到具有
识别能力的脉冲神经网络。
[0012]步骤2.2.1:脉冲神经网络通过类脑学习算子调整权值进行更新,类脑学习算子通过选择动作a
t
来实现权值调整,a
t
∈A={

α
s
,0,α
s
},0<α
s
<1,其中,α
s
表示学习率。权值变化公式为公式1:
[0013][0014]其中,表示突触前一层的第i个神经元和突触后一层第j个神经元相连的突触k的权值;
[0015]步骤2.2.2:类脑学习算子通过奖励指导脉冲神经网络学习图片特征,奖励是根据前两个时间步的训练损失的变化来定义的,两个时间步中损失函数减少得到奖励,反之得到惩罚,定义为公式2:
[0016][0017]其中,R
t
表示奖励,sign(x)函数表示符号函数,即x>0,sign(x)=1,x<0,sign(x)=

1,L
t
‑1、L
t
分别表示t时刻和t

1时刻的训练损失。
[0018]步骤2.2.3:在脉冲神经网络学习完图片特征后,需要验证脉冲神经网络的识别能力,通过输出层最先激活的神经元序号和图片的标签进行对比,若一致则表示网络已经学习到一张图片的特征,需要更多不同图片多次训练达到更好的效果;若不一致则进行下一步骤;
[0019]步骤2.2.4:通过策略加速类脑学习算子进行动作选择,达到更快学习图片特征的目的。类脑学习算子通过Q函数得到给定状态和动作的预期总回报,并选择使总回报最大时的动作,Q函数和策略π的数学表达为公式3和公式4:
[0020]Q(s
t
,a)=R
t+1
+γmax
a

Q(s
t+1
,a)
ꢀꢀ
公式3
[0021][0022]其中,s
t
表示脉冲神经网络当前状态,a表示当前时刻类脑学习算子选择的动作,γ表示折扣率,max
a

Q(s
t+1
,a)表示在t+1时刻Q函数取得最大值时候类脑学习算子选择的动作,π(s
t
)表示在t时刻类脑学习算子的策略,argmax
a∈A
Q(s
t
,a)表示Q(s
t
,a)取得最大值时动作a的取值,random

uniform(a∈A)表示从集合A中随机选择一个元素作为动作a。对于策略π的选择上,使用ε

greedy策略(ε

贪婪策略)扩大类脑学习算子的探索能力。
[0023]步骤2.2.5:重复步骤2.2.1

2.2.4,对网络进行多次训练学习,最终得到具有识别能力的脉冲神经网络。
[0024]步骤2.2.5.1:在通过步骤2.2.1

步骤2.2.4更新一次脉冲神经网络权值之前,计算得到网络的初始误差L
old

[0025]步骤2.2.5.2:重复步骤2.2.1

2.2.4,更新所有网络权值参数,计算得到网络的更新误差L
new

[0026]步骤2.2.5.3:计算L
old
和L
new
的差值,如果L
old
<L
new
说明网络误差在变大,是错误的
变化方向,网络的学习没有收敛;如果L
old
>L
new
,说明网路误差在减小,是正确的变化方向,需要继续学习;直到两次学习后网络误差几乎不变,结束学习,完成训练。
[0027]步骤3:将待识别的原始图片数据进行预处理,用识别网络进行图片识别。
[0028]步骤3.1:将未知类别的图片数据先进行卷积神经网络的预处理,得到输入向量;
[0029]步骤3.2:利用ROC编码方式对输入向量进行脉冲时序编码,得到能作为具有识别能力的脉冲神经网络本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.一种基于类脑学习的图像识别方法,其特征在于,包括如下步骤:步骤1:对原始图片数据进行预处理得到输入向量;所述原始图片数据指训练脉冲神经网络所用的训练数据,其包括了图片和标签两个部分;步骤2:建立脉冲神经网络模型进行学习,所述脉冲神经网络模型包括输入层、类脑学习算子和输出层;先对所述输入向量进行脉冲时序编码得到脉冲神经网络的输入数据,再将所述输入数据输入到脉冲神经网络的输入层,然后通过类脑学习算子学习输入向量的特征,最后将脉冲神经网络的输出层最先激发的脉冲神经元作为网络识别的类别,经过多次训练学习后,得到最后的识别网络;步骤3:将待识别的原始图片数据进行预处理,用识别网络进行图片识别。2.根据权利要求1所述基于类脑学习的图像识别方法,其特征在于,所述步骤1具体包括以下步骤:步骤1.1:将原始输入图片直接放入三层卷积神经网络中进行卷积处理,得到大小为24
×
24的处理后的图片;步骤1.2:将处理后的图片送入三层卷积神经网络的池化层中进行池化处理,得到1
×
576的行向量,将该行向量作为输入向量。3.根据权利要求1所述基于类脑学习的图像识别方法,其特征在于,所述步骤2具体包括以下步骤:步骤2.1:构建两层脉冲神经网络,将步骤1得到的输入向量经过脉冲时序编码后作为脉冲神经网络的输入;步骤2.2:利用类脑学习算子学习输入向量的特征,经过多轮训练学习后得到具有识别能力的脉冲神经网络。4.根据权利要求3基于类脑学习的图像识别方法,其特征在于,所述步骤2.1,两层脉冲神经网络中,前一层为输入层,接受脉冲时序数据输入,后一层为输出层,将最先激发的神经元作为响应类别;所述脉冲时序编码的方法为ROC方法。5.根据权利要求3基于类脑学习的图像识别方法,其特征在于,所述步骤2.2具体包括以下步骤:步骤2.2.1:脉冲神经网络通过类脑学习算子调整权值进行更新,类脑学习算子通过选择动作a
t
来实现权值调整,a
t
∈A={

α
s
,0,α
s
},0<α
s
<1,其中,α
s
表示学习率,权值变化公式为公式1:其中,表示突触前一层的第i个神经元和突触后一层第j个神经元相连的突触k的权值;和分别为变化后和变化前的值;步骤2.2.2:类脑学习算子通过奖励指导脉冲神经网络学习图片特征,奖励是根据前两个时间步的训练损失的变化来定义的,两个时间步中损失函数减少得到奖励,反之得到惩罚,定义为公式2:
其中,R
t
表示奖励,sign(x)函数表示符号函数,即x>0,sign(x)=1,x<0,sign(x)=

1,L
t
‑1、L
t
分别表示t时刻和t

1时刻的训练损失;步骤2.2.3:在脉冲神经网络学习完图片特征后,需要验证脉冲神经网络的识别能力,通过输出层最先激活的神经元序号和图片的标签进行对比,若一致则表示网络已经学习到一张图片的特征,需要更多不同图片多次训练达到更好的效果;若不一致则进行下一步骤;步骤2.2.4:通过策略加速类脑学习算子进行...

【专利技术属性】
技术研发人员:杨旭王淼毛思扬雷云霖杨君涵
申请(专利权)人:北京理工大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1