arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.04718cs.CVcs.LG

Canonical Capsules: Self-Supervised Capsules in Canonical Pose

University of British Columbia(不列颠哥伦比亚大学);University of Victoria(维多利亚大学);University of Toronto(多伦多大学);Google Research(谷歌研究院)

Weiwei Sun, Andrea Tagliasacchi, Boyang Deng, Sara Sabour, Soroosh Yazdani, Geoffrey Hinton, Kwang Moo Yi

英文摘要:

We propose a self-supervised capsule architecture for 3D point clouds. We compute capsule decompositions of objects through permutation-equivariant attention, and self-supervise the process by training with pairs of randomly rotated objects. Our key idea is to aggregate the attention masks into semantic keypoints, and use these to supervise a decomposition that satisfies the capsule invariance/equivariance properties. This not only enables the training of a semantically consistent decomposition, but also allows us to learn a canonicalization operation that enables object-centric reasoning. To train our neural network we require neither classification labels nor manually-aligned training datasets. Yet, by learning an object-centric representation in a self-supervised manner, our method outperforms the state-of-the-art on 3D point cloud reconstruction, canonicalization, and unsupervised classification.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑