|
Input:
FM, A, PPMI, yL, r, λ(t) and hidden convolution layers (H) |
|
Output: Training model with best features. |
| Step 1: for
t
in range (0, Epoch number) do
|
| Step 2:
using Eq. (16)
|
| Step 3:
using Eq. (22)
|
| Step 4: Compute Loss using Eq. (23)
|
| Step 5: if convergence then break loops |
| Step 6: end if
|
| Step 7: end for
|
| Step 8: Dynamic routing procedure (OCj|i,W′,r). // Where
is the matrix of wights. |
| Step 9:
|
| Step 10: for all input capsules i and all output capsules j
do
|
| Step 11:
squash (OCj|i)) // Where
. |
| Step 12: end for
|
| Step 13: for all input capsules i and all output capsules j
do
|
| Step 14:
|
| Step 15: end for
|
| Step 16: for
r iterations do
|
| Step 17: for all input capsules i
do
|
| Step 18: for all output capsules j
do
|
| Step 19: for all output capsule networks i
do
// Where
. |
| Step 20: for all input capsules i and output capsules j
do
|
| Step 21:
|
| Step 22: end for
|
| Step 23: Return
Oj
|
| Step 24: end for
|