Inference and Learning for Generative Capsule Models. 2023

Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
Amazon Development Centre Scotland, Edinburgh EH1 3EG, U.K. alfrena@amazon.com.

Capsule networks (see Hinton et al., 2018) aim to encode knowledge of and reason about the relationship between an object and its parts. In this letter, we specify a generative model for such data and derive a variational algorithm for inferring the transformation of each model object in a scene and the assignments of observed parts to the objects. We derive a learning algorithm for the object models, based on variational expectation maximization (Jordan et al., 1999). We also study an alternative inference algorithm based on the RANSAC method of Fischler and Bolles (1981). We apply these inference methods to data generated from multiple geometric objects like squares and triangles ("constellations") and data from a parts-based model of faces. Recent work by Kosiorek et al. (2019) has used amortized inference via stacked capsule autoencoders to tackle this problem; our results show that we significantly outperform them where we can make comparisons (on the constellations data).

UI MeSH Term Description Entries

Related Publications

Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
January 2020, Frontiers in computational neuroscience,
Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
January 2016, Journal of vision,
Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
November 2020, Neuroscience and biobehavioral reviews,
Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
August 2021, IEEE transactions on neural networks and learning systems,
Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
January 2018, Frontiers in computational neuroscience,
Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
April 2011, Proteins,
Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
February 2022, Neural networks : the official journal of the International Neural Network Society,
Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
February 2021, Journal of molecular modeling,
Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
April 2022, Nature communications,
Alfredo Nazabal, and Nikolaos Tsagkas, and Christopher K I Williams
April 2018, Philosophical transactions of the Royal Society of London. Series B, Biological sciences,
Copied contents to your clipboard!