Reproduce “Learning with 3D rotations, a hitchhiker's guide to SO(3)” on OpenResearch

Turn your favorite coding agents into research agents that review literature, develop hypotheses, and run experiments.

Universal app for Apple silicon and Intel. Requires macOS 11 or later.

About this paper

A. René Geist, Jonas Frey, Mikel Zobro, Anna Levina, Georg Martius

Researchers from the Max Planck Institute for Intelligent Systems and ETH Zurich present a comprehensive guide for selecting 3D rotation representations in neural networks. The work establishes that high-dimensional, continuously mapped representations like R9+SVD consistently outperform low-dimensional alternatives in rotation estimation tasks, providing clear recommendations for various learning scenarios.

Key insights

  • For rotation estimation, low-dimensional representations (e.g., Euler angles, quaternions) fundamentally introduce discontinuities in the target function's mapping from SO(3) to the representation space, causing unstable gradient-based learning.
  • Common 'fixes' like distance-picking metrics or half-space maps do not resolve the fundamental topological discontinuities of the underlying target function itself, offering only superficial improvements.
  • High-dimensional representations, specifically R9+SVD and R6+GSO, circumvent these issues by providing a continuous mapping from SO(3) to a higher-dimensional Euclidean space, enabling more stable and effective gradient-based optimization.

Summary by alphaXiv.

Tip: change arxiv.org to autoarxiv.org in any arXiv abstract link to open the paper here.