深度模型(三):Capsule
其他
2020-01-25 10:27:13
阅读次数: 0
胶囊网络
胶囊与传统神经元比较:
|
胶囊 |
传统神经元 |
输入类型 |
向量
ui,i∈[1,m],
m表示底层胶囊数量 |
标量
xi,i∈[1,m],m表示底层神经元数量 |
仿射变换 |
u
j∣i=Wijui |
不支持 |
加权求和 |
sj=∑i=1mciju
j∣i |
aj=∑i=1mwixi+b |
激活函数 |
vj=1+∣∣sj∣∣2∣∣sj∣∣2∣∣sj∣∣sj |
hj=sigmoid(aj) |
输出 |
向量
vj |
标量
hj |
训练算法 |
动态路由 |
梯度下降 |
动态路由算法
原论文中对动态路由算法的描述如下:
- 算法的输入为输入网络层
l以及它的所有输出
u
j∣i,以及算法迭代次数
r。
- 第2行,初始化参数
bij=0
- 第3,5,6行,前向计算胶囊输出
vj。
- 第7行,更新参数
bij←bij+u
j∣i⋅vj
- 迭代
r次以后,算法完成
发布了52 篇原创文章 ·
获赞 105 ·
访问量 7万+
转载自blog.csdn.net/gaofeipaopaotang/article/details/97615736