Fig. 3

Computation between the PrimaryCaps and ProteinCaps layers.There are 2976 16D capsules (each u_i is an 16D vector) in PrimaryCaps. Each \(\hat {u_{j|i}}\) is produced by multiplying u_i by a weight matrix W_i,j (16 ×16). Capsule V_j (16D vector and j∈ [1, 8]) in ProteinCaps is produced by a weighted sum over all \(\hat {u_{j|i}}\) and the squashing nonlinear activation function. The parameter c_i,j is determined by the iterative dynamic routing process