A Fault Diagnostic Scheme Based on Capsule Network for Rolling Bearing under Different Rotational Speeds

. 2020 Mar 26;20(7):1841. doi: 10.3390/s20071841

Algorithm 1 Dynamic routing algorithm

1:
Procedure Routing ( $u_{j | i}, r, l$ )
2:
for all capsule i in layer l and capsule j in layer $(l + 1)$ : $b_{i j} \leftarrow 0$ .
3:
for r iterations do
4:
for all capsule i in layer l: $c_{i} \leftarrow$ $S o f t M a x (b_{i})$
5:
for all capsule j in layer $(l + 1)$ : $s_{j} \leftarrow Σ_{i} c_{i j} u_{i j}$
6:
for all capsule j in layer $(l + 1)$ : $v_{j} \leftarrow$ $s q u a s h (s_{j})$
7:
for all capsule i in layer l and capsule j in layer $(l + 1) : b_{i j} \leftarrow b_{i j} + u_{j | i} \cdot v_{j}$
8:
return $v_{j}$