We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Quaternion Equivariant Capsule Networks for 3D Point Clouds

Abstract: We present a 3D capsule module for processing point clouds that is equivariant to 3D rotations and translations, as well as invariant to permutations of the input points. The operator receives a sparse set of local reference frames, computed from an input point cloud and establishes end-to-end transformation equivariance through a novel dynamic routing procedure on quaternions. Further, we theoretically connect dynamic routing between capsules to the well-known Weiszfeld algorithm, a scheme for solving \emph{iterative re-weighted least squares} (IRLS) problems with provable convergence properties. It is shown that such group dynamic routing can be interpreted as robust IRLS rotation averaging on capsule votes, where information is routed based on the final inlier scores. Based on our operator, we build a capsule network that disentangles geometry from pose, paving the way for more informative descriptors and a structured latent space. Our architecture allows joint object classification and orientation estimation without explicit supervision of rotations. We validate our algorithm empirically on common benchmark datasets.
Comments: Oral Presentation at ECCV 2020. Find our video under: this https URL We release our sources at: this http URL
Subjects: Machine Learning (cs.LG); Computer Vision and Pattern Recognition (cs.CV); Graphics (cs.GR); Robotics (cs.RO); Machine Learning (stat.ML)
Cite as: arXiv:1912.12098 [cs.LG]
  (or arXiv:1912.12098v3 [cs.LG] for this version)

Submission history

From: Tolga Birdal [view email]
[v1] Fri, 27 Dec 2019 13:51:17 GMT (2996kb,D)
[v2] Thu, 16 Apr 2020 19:09:44 GMT (2998kb,D)
[v3] Sun, 23 Aug 2020 13:12:46 GMT (5662kb,D)

Link back to: arXiv, form interface, contact.