Abstract
Capsule networks emerged as a promising alternative to convolutional neural networks for learning object-centric representations. The idea is to explicitly model part-whole hierarchies by using groups of neurons called capsules to encode visual entities, then learn the relationships between these entities dynamically from data. However, a major hurdle for capsule network research has been the lack of a reliable point of reference for understanding their foundational ideas and motivations. This survey provides a comprehensive and critical overview of capsule networks which aims to serve as a main point of reference going forward. To that end, we introduce the fundamental concepts and motivations behind capsule networks, such as equivariant inference. We then cover various technical advances in capsule routing algorithms as well as alternative geometric and generative formulations. We provide a detailed explanation of how capsule networks relate to the attention mechanism in Transformers and uncover non-trivial conceptual similarities between them in the context of object-centric representation learning. We also review the extensive applications of capsule networks in computer vision, video and motion, graph representation learning, natural language processing, medical imaging, and many others. To conclude, we provide an in-depth discussion highlighting promising directions for future work.
| Original language | English |
|---|---|
| Article number | 291 |
| Number of pages | 43 |
| Journal | ACM Computing Surveys |
| Volume | 56 |
| Issue number | 11 |
| Early online date | 21 Jun 2024 |
| DOIs | |
| Publication status | Published - 22 Jul 2024 |
Bibliographical note
The authors would like to thank all reviewers, and especially Professor Chris Williams from the School of Informatics of the University of Edinburgh, who provided constructive feedback and ideas on how to improve this work.Keywords
- deep learning
- capsule neural networks
- Deep neural networks
- convolutional neural networks
- Transformers
- routing-by-agreement
- self-attention
- representation learning
- object-centric learning
- generative models
- Computer vision