Optimized CapsNet for Traffic Jam Speed Prediction Using Mobile Sensor Data under Urban Swarming Transportation

. 2019 Nov 29;19(23):5277. doi: 10.3390/s19235277

Algorithm 1 DR (Sabour et al., 2017 [29])

Procedure DR $({\hat{u}}_{j | i}, r, l)$
For caps $i$ in layer $l$ and caps j in layer $(l + 1)$ : $b_{i j} \leftarrow 0$
For $r$ iterations do
For all caps $i$ in layer $l : c_{i} \leftarrow s o f t m a x (b_{i})$ (Equation (5))
For all caps $j$ in layer ( $l + 1) : s_{j} \leftarrow \sum_{i} c_{i j} {\hat{u}}_{j | i}$ (Equation (6))
For all caps $j$ in layer ( $l + 1) : v_{j} \leftarrow S q u a s h (s_{j})$ (Equation (7))
For all caps $i$ in layer $l$ and caps j in layer $(l + 1) : b_{i j} \leftarrow b_{i j} + v_{j} {\hat{u}}_{j | i}$ (Equation (10))
Return $v_{j}$