卷积神经网络需要学习的参数有哪些?

发布时间:2021-01-07  栏目:人工智能, 机器视觉  评论:0 Comments

卷积神经网络中有很多执行不同功能的层结构,同时也有很多参数需要学习。其中池化层(包括空间金字塔池化层)、reshape层是没有需要学习的参数的,下面我将会盘点CNN中的所有参数。

Softmax层

Softmax层是一个多分类的逻辑斯蒂回归,其需要学习的参数为权重和偏置。

卷积层

卷积层是CNN的核心,其参数则是卷积核的权值以及各通道的偏置量。其中权值的梯度需要使用反卷积获得,偏置量的梯度则是对应通道的误差之和。

全连接层

全连接层的参数同样也是权值和偏置量。其中权值的梯度是上一层的输出和当前层误差转置的积,而各通道的偏置量则也是对应通道的误差。

BatchNorm层

BN层有一个scale和一个shift参数,也可以看作是权值和偏置量。计算方法相同,在此不再赘述。

激活层

一般的激活层没有参数,但如果激活函数是PReLU则不同了——它需要学习一个x < 0时的权值,方法也是上一层负的输出与对应的当前层误差的积之和。

 

总体上就是这些参数,总结出来就是各层的权值和偏置量。

 

留下评论

You must be logged in to post a comment.

相册集

pix pix pix pix pix pix

关于自己

杨文龙,微软Principal Engineering Manager, 曾在各家公司担任影像技术资深总监、数据科学团队资深经理、ADAS算法总监、资深深度学习工程师等职位,热爱创新发明,专注于人工智能、深度学习、图像处理、机器学习、算法、自然语言处理及软件等领域,目前发明有国际专利19篇,中国专利28篇。

联系我

个人技术笔记

welonshen@gmail.com

2015 in Shanghai