Momentum Capsule Networks. (arXiv:2201.11091v2 [cs.CV] UPDATED)

Capsule networks are a class of neural networks that achieved promising
results on many computer vision tasks. However, baseline capsule networks have
failed to reach state-of-the-art results on more complex datasets due to the
high computation and memory requirements. We tackle this problem by proposing a
new network architecture, called Momentum Capsule Network (MoCapsNet).
MoCapsNets are inspired by Momentum ResNets, a type of network that applies
reversible residual building blocks. Reversible networks allow for
recalculating activations of the forward pass …

arxiv cv networks