【技术实现步骤摘要】
基于代理辅助进化算法的卷积神经网络结构优化方法
[0001]本专利技术属于智能优化领域,涉及一种卷积神经网络结构优化方法,具体涉及一种基于代理辅助进化算法的卷积神经网络结构优化方法。
技术介绍
[0002]卷积神经网络(Convolutional Neural Network,CNN)是一种被广泛应用于计算机视觉、自然语言处理、语音识别和语义分析等领域的神经网络模型,它的基本结构由输入层、卷积层、池化层、全连接层及输出层构成。卷积层通过卷积操作提取特征,池化层通过降低特征面的分辨率来获得具有空间不变性的特征,常用的池化方法有最大池化和均值池化,全连接层可以整合具有类别区分性的特征信息。在全连接层中采用神经元随机失活(Dropout)技术,降低网络复杂度,使神经元能够学习到更加鲁棒的特征。然而,卷积神经网络的结构设计是十分重要且困难的,不同的结构参数会对模型的性能产生很大的影响。当基于经验设计或经典的网络结构无法满足任务需求时,网络结构参数的优化研究十分重要。
[0003]例如华南理工大学与佛山市顺德致可智能科技有限公司在其申请的专利文献“一种基于PSO的权值直接确定神经网络结构优化方法”(申请号:CN201811024634.5公开号:CN109344961A)中,公开了一种基于PSO的权值直接确定神经网络结构优化方法,该方法将PSO算法应用于优化网络结构,寻找神经网络最佳神经元数目,所训练的网络具有很强的逼近能力,实用价值高。但是,该方法的不足之处是,对神经元数量这个连续型结构参数进行优化,没有对激活函数类
【技术保护点】
【技术特征摘要】
1.一种基于代理辅助进化算法的卷积神经网络结构优化方法,其特征在于,包括如下步骤:(1)获取训练样本集和验证样本集:获取包括C类别的M幅图像,并对每幅图像的类别进行标注,然后将V幅图像及其对应的标签组成训练图像样本集D
train
,将其余M
‑
V幅图像及其对应的标签组成验证图像样本集D
val
,其中,C≥2,M≥50000,V>M/2;(2)获取卷积神经网络的结构参数:获取包括A个卷积层、B个池化层、E个全连接层的卷积神经网络模型O的结构参数为s,s包括由A个卷积层卷积核的数量、前E
‑
1个全连接层神经元的数量和神经元随机失活率组成的连续型参数x,以及由A个卷积层卷积核的大小、使用的激活函数类型和B个池化层使用的池化函数类型组成的离散型参数z,其中,A≥1,B≥1,E≥2,x包含的参数数量为L=A+2E
‑
2,z包含的参数数量为R=2A+B;(3)初始化参数样本集和验证准确率集:对结构参数s进行N次随机采样,并将随机采样得到的参数样本集S1={s1,s2,...,s
n
,...,s
N
}中的每一个参数样本s
n
作为卷积神经网络模型O的结构参数取值得到卷积神经网络然后将训练图像样本集D
train
作为卷积神经网络的输入进行e次迭代训练,再将验证图像样本集D
val
作为训练好的卷积神经网络的输入对验证图像样本的类别进行预测,并计算预测类别与真实标签一致的样本数量与验证图像样本总数量M
‑
V的比值作为s
n
对应的验证准确率f
n
,最后得到参数样本集S1对应的验证准确率集F1={f1,f2,...,f
n
,...,f
N
},其中N≥100,s
n
表示第n个参数样本,f
n
表示s
n
对应的验证准确率,e≥10;(4)基于代理辅助进化算法对结构参数进行优化:(4a)初始化迭代次数为t,最大迭代次数为T,T≥200,多样性阈值为τ,0<τ<1,种群为P
t
,并令t=1,P1=S1;(4b)计算当前种群P
t
的多样性指标R
t
,并判断R
t
<τ是否成立,若是,执行步骤(4c),否则,执行步骤(4d);(4c)采用全局代理辅助混合进化算法对结构参数s进行全局优化,并将优化后的结构参数作为卷积神经网络模型O的结构参数取值得到卷积神经网络然后将训练图像样本集D
train
作为卷积神经网络的输入进行e次迭代训练,再将验证图像样本集D
val
作为训练好的卷积神经网络的输入对验证图像样本的类别进行预测,最后计算预测类别与真实标签一致的样本数量与验证图像样本总数量M
‑
V的比值作为对应的验证准确率f
1t
,更新参数样本集S'
t
和对应的验证准确率集F
t
',F
t
'=F
t
‑1∪{f
1t
},并执行步骤(4e);(4d)采用局部代理辅助分布估计算法对结构参数s进行局部优化,并将优化后的结构参数作为卷积神经网络模型O的结构参数取值得到卷积神经网络然后将训练图像样本集D
train
作为卷积神经网络的输入进行e次迭代训练,再将验证图像样本集D
val
作为训
练好的卷积神经网络的输入对验证图像样本的类别进行预测,最后计算预测类别与真实标签一致的样本数量与验证图像样本总数量M
‑
V的比值作为对应的验证准确率并更新参数样本集S'
t
和对应的验证准确率集F
t
',',(4e)将从当前种群P
t
中选出的验证准确率最高的个体和从参数样本集S
t
中选出的N
cb
个离散型参数取值为的样本构成样本参数集然后判断N
cb
≥5
×
L是否成立,若是,执行步骤(4f),否则,执行步骤(4g);(4f)采用连续代理辅助差分进化算法对连续型结构参数x进行优化,得到优化后的连续型结构参数组合和得到的作为卷积神经网络模型O的结构参数得到卷积神经网络然后将训练图像样本集D
train
作为卷积神经网络的输入进行e次迭代训练,再将验证图像样本集D
val
作为训练好的卷积神经网络的输入对验证图像样本的类别进行预测,最后计算预测类别与真实标签一致的样本数量与验证图像样本总数量M
‑
V的比值作为对应的验证准确率更新参数样本集S'
t
和对应的验证准确率集F
t
',',(4g)判断t=T是否成立,若是,输出参数样本集S
T
和对应的验证准确率集F
T
,并从S
T
中取F
T
中验证准确率最高的结构参数s
best
={x
best
,z
best
}作为卷积神经网络模型O的结构参数,得到最优卷积神经网络O
sbest
,否则,令t=t+1,S
t
=S'
t
,F
t
=F
t
',并从参数样本集S
t
中选取F
t
中验证准确率高的前N个参数样本作为当前种群P
t
,并执行步骤(4b)。2.根据权利要求1所述的基于代理辅助进化算法的卷积神经网络结构优化方法,其特征在于,步骤(4b)中所述的当前种群P
t
的多样性指标R
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。