spatial_routing_matrix = utl.create_routing_map(child_space=1, k=1, s=1) ?

[ibm]

spatial_routing_matrix = utl.create_routing_map(child_space=1, k=1, s=1) ?

[ashleygritzman]

Hi Jeff,

Thanks for your question.

The spatial_routing_map is used to determine which child capsules vote for which parent capsules. The layer fc_caps() refers to a fully connected capsule layer, which means that all the child capsules feed to the parent capsules. We achieve this by flattening the child capsules along their spatial dimension, so considering an example with a batch size of 1, we would go from (1, 7, 7, 8, ?) to (1, 778, ?) [batch_size, child_space * child_space * child_caps, ?]. In other words, we now have one long vector of capsules, instead of a cube of capsules. So for this reason it is correct to have a spatial_routing_matrix of (1,1), which effectively means that no spatial routing occurs in this layer.

You can contract this to the convolutional capsule layer conv_caps(), where we do have spatial routing of capsules.

Hope this helps :-)

So

[itxiud2015]

Hi Ashley,

Thanks for your clearly explanation :)

I tried 'spatial_routing_matrix = utl.create_routing_map(child_space=5, k=5, s=1)', this result in a 'spatial_routing_matrix' of shape [25, 1]. It function well and the test accuracy is slightly more than 90% after 65 epoch training, which is comparable with that of setting utl.create_routing_map(child_space=1, k=1, s=1) at my workstation.

Best regards
Jeff