[edit]
Unsupervised Learning of Latent Physical Properties Using Perception-Prediction Networks
Proceedings of the 34th Conference on Uncertainty in Artificial Intelligence, PMLR R16:496-506, 2018.
Abstract
We propose a framework for the completely unsupervised learning of latent object prop- erties from their interactions: the perception- prediction network (PPN). Consisting of a per- ception module that extracts representations of latent object properties and a prediction module that uses those extracted properties to simulate system dynamics, the PPN can be trained in an end-to-end fashion purely from samples of object dynamics. The representations of latent object properties learned by PPNs not only are sufficient to accurately simulate the dynamics of systems comprised of previously unseen ob- jects, but also can be translated directly into human-interpretable properties (e.g. mass, co- efficient of restitution) in an entirely unsuper- vised manner. Crucially, PPNs also generalize to novel scenarios: their gradient-based training can be applied to many dynamical systems and their graph-based structure functions over sys- tems comprised of different numbers of objects. Our results demonstrate the efficacy of graph- based neural architectures in object-centric in- ference and prediction tasks, and our model has the potential to discover relevant object proper- ties in systems that are not yet well understood.