Deep Regression on Manifolds: A 3D Rotation Case Study
Not on the map. Why?
Abstract (arXiv)
Many machine learning problems involve regressing variables on a non-Euclidean manifold -- e.g. a discrete probability distribution, or the 6D pose of an object. One way to tackle these problems through gradient-based learning is to use a differentiable function that maps arbitrary inputs of a Euclidean space onto the manifold. In this paper, we establish a set of desirable properties for such mapping, and in particular highlight the importance of pre-images connectivity/convexity. We illustrate these properties with a case study regarding 3D rotations. Through theoretical considerations and methodological experiments on a variety of tasks, we review various differentiable mappings on the 3D rotation space, and conjecture about the importance of their local linearity. We show that a mapping based on Procrustes orthonormalization generally performs best among the mappings considered, but that a rotation vector representation might also be suitable when restricted to small angles.
Key facts. Cited by 67 papers in the map (data through 7 Oct 2026). Builds on 24 of its 39 references that are in the map. First posted on arXiv 30 Mar 2021.
Builds on
24 of 39 references.
References · 24
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2015ShapeNet: An Information-Rich 3D Model Repositorycited by 2,880
- 2018On the Continuity of Rotation Representations in Neural Networkscited by 1,186
- 2016Geometric deep learning: going beyond Euclidean datacited by 1,042
- 2017Unsupervised Learning of Depth and Ego-Motion from Videocited by 778
- 2019VIBE: Video Inference for Human Body Pose and Shape Estimationcited by 615
- 2017SSD-6D: Making RGB-based 3D detection and 6D pose estimation great againcited by 243
- 2017BB8: A Scalable, Accurate, Robust to Partial Occlusion Method for Predicting the 3D Poses of Challenging Objects without Using Depthcited by 225
- 2019PVN3D: A Deep Point-wise 3D Keypoints Voting Network for 6DoF Pose Estimationcited by 204
- 2020BOP Challenge 2020 on 6D Object Localizationcited by 192
12 more
- 2018Neural Body Fitting: Unifying Deep Learning and Model-Based Human Pose and Shape Estimationcited by 185
- 2016Image-based localization using LSTMs for structured feature correlationcited by 136
- 2020An Analysis of SVD for Deep Rotation Estimationcited by 85
- 2019TexturePose: Supervising Human Mesh Estimation with Texture Consistencycited by 53
- 2017On Pre-Trained Image Features and Synthetic Images for Deep Learningcited by 51
- 2019Delving Deep Into Hybrid Annotations for 3D Human Recovery in the Wildcited by 38
- 2019Spherical Regression: Learning Viewpoints, Surface Normals and 3D Rotations on n-Spherescited by 37
- 2020Probabilistic orientation estimation with matrix Fisher distributionscited by 36
- 2019On Object Symmetries and 6D Pose Estimation from Imagescited by 33
- 2020A Smooth Representation of Belief over SO(3) for Deep Rotation Learning with Uncertaintycited by 29
- 2016Defining the Pose of any 3D Rigid Object and an Associated Distancecited by 17
- 2013A compact formula for the derivative of a 3-D rotation in exponential coordinatescited by 14
Built on by
Citing papers per year: 2021: 1; 2022: 9; 2023: 15; 2024: 15; 2025: 21; 2026: 6.
67 papers.
Sort:
By area:
- 2022CroCo: Self-Supervised Pre-training for 3D Vision Tasks by Cross-View Completioncited by 158
- 2022DeXtreme: Transfer of Agile In-hand Manipulation from Simulation to Realitycited by 136
- 2024Multi-HMR: Multi-Person Whole-Body Human Mesh Recovery in a Single Shotcited by 98
- 2022PoseGPT: Quantization-based 3D Human Motion Generation and Forecastingcited by 78
- 2025RayZer: A Self-supervised Large View Synthesis Modelcited by 74
- 2024Scene Coordinate Reconstruction: Posing of Image Collections via Incremental Learning of a Relocalizercited by 73
- 2025Pow3R: Empowering Unconstrained 3D Reconstruction with Camera and Scene Priorscited by 69
- 2023CARTO: Category and Joint Agnostic Reconstruction of ARTiculated Objectscited by 64
- 2024Learning Robotic Manipulation Policies from Point Clouds with Conditional Flow Matchingcited by 63
- 2024LoopSplat: Loop Closure by Registering 3D Gaussian Splatscited by 59
- 2022Local-to-Global Registration for Bundle-Adjusting Neural Radiance Fieldscited by 58
- 2023gSDF: Geometry-Driven Signed Distance Functions for 3D Hand-Object Reconstructioncited by 49
38 more
- 2024Learning with 3D rotations, a hitchhiker's guide to SO(3)cited by 39
- 2021Projective Manifold Gradient Layer for Deep Rotation Regressioncited by 31
- 2024Scaling 4D Representationscited by 30
- 2025RenderFormer: Transformer-based Neural Rendering of Triangle Meshes with Global Illuminationcited by 23
- 2023SoftZoo: A Soft Robot Co-design Benchmark For Locomotion In Diverse Environmentscited by 18
- 2023Image to Sphere: Learning Equivariant Features for Efficient Pose Predictioncited by 17
- 2023SHOWMe: Benchmarking Object-agnostic Hand-Object 3D Reconstructioncited by 16
- 2023End-to-End (Instance)-Image Goal Navigation through Correspondence as an Emergent Phenomenoncited by 16
- 2022PoseBERT: A Generic Transformer Module for Temporal 3D Human Modelingcited by 15
- 2024SurFhead: Affine Rig Blending for Geometrically Accurate 2D Gaussian Surfel Head Avatarscited by 15
- 2025From Image to Video: An Empirical Study of Diffusion Representationscited by 14
- 2023Bridging the Gap between Human Motion and Action Semantics via Kinematic Phrasescited by 13
- 2023CenterGrasp: Object-Aware Implicit Representation Learning for Simultaneous Shape Reconstruction and 6-DoF Grasp Estimationcited by 13
- 2025Detection Based Part-level Articulated Object Reconstruction from Single RGBD Imagecited by 13
- 2022Spacecraft Pose Estimation Based on Unsupervised Domain Adaptation and on a 3D-Guided Loss Combinationcited by 12
- 2023MFOS: Model-Free & One-Shot Object Pose Estimationcited by 10
- 2025Flexible MOF Generation with Torsion-Aware Flow Matchingcited by 9
- 2023Learning to Predict 3D Rotational Dynamics from Images of a Rigid Body with Unknown Mass Distributioncited by 7
- 2025Cloth-Splatting: 3D Cloth State Estimation from RGB Supervisioncited by 7
- 2023Equivariant Single View Pose Prediction Via Induced and Restricted Representationscited by 6
- 2025StereoDiff: Stereo-Diffusion Synergy for Video Depth Estimationcited by 6
- 2024AutoURDF: Unsupervised Robot Modeling from Point Cloud Frames Using Cluster Registrationcited by 5
- 2024CondiMen: Conditional Multi-Person Mesh Recoverycited by 5
- 2025Differentiable Simulation of Hard Contacts with Soft Gradients for Learning and Controlcited by 5
- 2022Learning Interpretable Dynamics from Images of a Freely Rotating 3D Rigid Bodycited by 4
- 20243D Equivariant Pose Regression via Direct Wigner-D Harmonics Predictioncited by 4
- 2026Rays as Pixels: Learning A Joint Distribution of Videos and Camera Trajectoriescited by 4
- 2022Image to Icosahedral Projection for SO(3) Object Reasoning from Single-View Imagescited by 3
- 2023SyNDock: N Rigid Protein Docking via Learnable Group Synchronizationcited by 3
- 2025ViTa-Zero: Zero-shot Visuotactile Object 6D Pose Estimationcited by 3
- 2025Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignmentcited by 3
- 2025Forecasting Continuous Non-Conservative Dynamical Systems in SO(3)cited by 3
- 2024Towards Unstructured Unlabeled Optical Mocap: A Video Helps!cited by 2
- 20243D Foundation Models Enable Simultaneous Geometry and Pose Estimation of Grasped Objectscited by 2
- 2024Alignist: CAD-Informed Orientation Distribution Estimation by Fusing Shape and Correspondencescited by 2
- 2025MatrixNet: Learning over symmetry groups using learned group representationscited by 2
- 2025Bi-Manual Joint Camera Calibration and Scene Representationcited by 2
- 2025Continuous-Time SO(3) Forecasting with Savitzky--Golay Neural Controlled Differential Equationscited by 2
50 most cited of 67 All 67 (JSON)
- 2026Predicting build orientation for SLM dental parts: a comparison of rotation representations and direct vector regression14 Sep 2026
- 2026Revisiting Euler-Angle Regression with Kolmogorov-Arnold Networks10 Jul 2026
- 2026G3T Up! Gravity Aligned Coordinate Frames Simplify Pointmap Processing26 May 2026
- 2026RigidFormer: Learning Rigid Dynamics using Transformers9 May 2026
- 2026Rays as Pixels: Learning A Joint Distribution of Videos and Camera Trajectories10 Apr 2026
- 2026Training Without Orthogonalization, Inference With SVD: A Gradient Analysis of Rotation Representations7 Apr 2026
- 2025TAPVid-360: Tracking Any Point in 360 from Narrow Field of View Video26 Nov 2025
- 2025Matcha: Multi-Stage Riemannian Flow Matching for Accurate and Physically Valid Molecular Docking16 Oct 2025
- 2025A Primer on SO(3) Action Representations in Deep Reinforcement Learning13 Oct 2025
- 2025Monocular Visual Odometry without Calibration or Test-time Optimization2 Oct 2025
- 2025Towards Foundation Models for Cryo-ET Subtomogram Analysis29 Sep 2025
- 2025On the Skinning of Gaussian Avatars14 Sep 2025
38 more
- 2025Forecasting Continuous Non-Conservative Dynamical Systems in SO(3)11 Aug 2025
- 2025StereoDiff: Stereo-Diffusion Synergy for Video Depth Estimation25 Jun 2025
- 2025Differentiable Simulation of Hard Contacts with Soft Gradients for Learning and Control17 Jun 2025
- 2025Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignment11 Jun 2025
- 2025Continuous-Time SO(3) Forecasting with Savitzky--Golay Neural Controlled Differential Equations7 Jun 2025
- 2025Bi-Manual Joint Camera Calibration and Scene Representation30 May 2025
- 2025RenderFormer: Transformer-based Neural Rendering of Triangle Meshes with Global Illumination28 May 2025
- 2025Flexible MOF Generation with Torsion-Aware Flow Matching23 May 2025
- 2025RayZer: A Self-supervised Large View Synthesis Model1 May 2025
- 2025ViTa-Zero: Zero-shot Visuotactile Object 6D Pose Estimation17 Apr 2025
- 2025Detection Based Part-level Articulated Object Reconstruction from Single RGBD Image4 Apr 2025
- 2025Pow3R: Empowering Unconstrained 3D Reconstruction with Camera and Scene Priors21 Mar 2025
- 2025From Image to Video: An Empirical Study of Diffusion Representations10 Feb 2025
- 2025MatrixNet: Learning over symmetry groups using learned group representations16 Jan 2025
- 2025Cloth-Splatting: 3D Cloth State Estimation from RGB Supervision3 Jan 2025
- 2024Scaling 4D Representations19 Dec 2024
- 2024CondiMen: Conditional Multi-Person Mesh Recovery17 Dec 2024
- 2024AutoURDF: Unsupervised Robot Modeling from Point Cloud Frames Using Cluster Registration7 Dec 2024
- 2024Multi-Resolution Generative Modeling of Human Motion from Limited Data25 Nov 2024
- 20243D Equivariant Pose Regression via Direct Wigner-D Harmonics Prediction1 Nov 2024
- 2024SurFhead: Affine Rig Blending for Geometrically Accurate 2D Gaussian Surfel Head Avatars15 Oct 2024
- 2024Symmetry-Robust 3D Orientation Estimation2 Oct 2024
- 2024Learning Robotic Manipulation Policies from Point Clouds with Conditional Flow Matching11 Sep 2024
- 2024Alignist: CAD-Informed Orientation Distribution Estimation by Fusing Shape and Correspondences10 Sep 2024
- 2024LoopSplat: Loop Closure by Registering 3D Gaussian Splats19 Aug 2024
- 20243D Foundation Models Enable Simultaneous Geometry and Pose Estimation of Grasped Objects14 Jul 2024
- 2024Towards Unstructured Unlabeled Optical Mocap: A Video Helps!15 May 2024
- 2024Scene Coordinate Reconstruction: Posing of Image Collections via Incremental Learning of a Relocalizer22 Apr 2024
- 2024Learning with 3D rotations, a hitchhiker's guide to SO(3)17 Apr 2024
- 2024Multi-HMR: Multi-Person Whole-Body Human Mesh Recovery in a Single Shot22 Feb 2024
- 2023CenterGrasp: Object-Aware Implicit Representation Learning for Simultaneous Shape Reconstruction and 6-DoF Grasp Estimation13 Dec 2023
- 2023Bridging the Gap between Human Motion and Action Semantics via Kinematic Phrases6 Oct 2023
- 2023MFOS: Model-Free & One-Shot Object Pose Estimation3 Oct 2023
- 2023End-to-End (Instance)-Image Goal Navigation through Correspondence as an Emergent Phenomenon28 Sep 2023
- 2023SHOWMe: Benchmarking Object-agnostic Hand-Object 3D Reconstruction19 Sep 2023
- 2023Hierarchical Attention and Graph Neural Networks: Toward Drift-Free Pose Estimation18 Sep 2023
- 2023Learning to Predict 3D Rotational Dynamics from Images of a Rigid Body with Unknown Mass Distribution24 Aug 2023
- 2023Pseudo-rigid body networks: learning interpretable deformable object dynamics from partial observations16 Jul 2023
50 newest of 67 All 67 (JSON)
- 2024Learning Robotic Manipulation Policies from Point Clouds with Conditional Flow Matchingcited by 63
- 2025From Image to Video: An Empirical Study of Diffusion Representationscited by 14
- 2025StereoDiff: Stereo-Diffusion Synergy for Video Depth Estimationcited by 6
- 2025Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignmentcited by 3
- 2025StereoDiff: Stereo-Diffusion Synergy for Video Depth Estimation25 Jun 2025
- 2025Scalable Non-Equivariant 3D Molecule Generation via Rotational Alignment11 Jun 2025
- 2025From Image to Video: An Empirical Study of Diffusion Representations10 Feb 2025
- 2024Learning Robotic Manipulation Policies from Point Clouds with Conditional Flow Matching11 Sep 2024
- 2022CroCo: Self-Supervised Pre-training for 3D Vision Tasks by Cross-View Completioncited by 158
- 2022DeXtreme: Transfer of Agile In-hand Manipulation from Simulation to Realitycited by 136
- 2024Multi-HMR: Multi-Person Whole-Body Human Mesh Recovery in a Single Shotcited by 98
- 2022PoseGPT: Quantization-based 3D Human Motion Generation and Forecastingcited by 78
- 2025RayZer: A Self-supervised Large View Synthesis Modelcited by 74
- 2024Scene Coordinate Reconstruction: Posing of Image Collections via Incremental Learning of a Relocalizercited by 73
- 2025Pow3R: Empowering Unconstrained 3D Reconstruction with Camera and Scene Priorscited by 69
- 2023CARTO: Category and Joint Agnostic Reconstruction of ARTiculated Objectscited by 64
- 2024LoopSplat: Loop Closure by Registering 3D Gaussian Splatscited by 59
- 2022Local-to-Global Registration for Bundle-Adjusting Neural Radiance Fieldscited by 58
- 2023gSDF: Geometry-Driven Signed Distance Functions for 3D Hand-Object Reconstructioncited by 49
- 2024Learning with 3D rotations, a hitchhiker's guide to SO(3)cited by 39
12 most cited of 63 All 63 (JSON)
- 2026Predicting build orientation for SLM dental parts: a comparison of rotation representations and direct vector regression14 Sep 2026
- 2026Revisiting Euler-Angle Regression with Kolmogorov-Arnold Networks10 Jul 2026
- 2026G3T Up! Gravity Aligned Coordinate Frames Simplify Pointmap Processing26 May 2026
- 2026RigidFormer: Learning Rigid Dynamics using Transformers9 May 2026
- 2026Rays as Pixels: Learning A Joint Distribution of Videos and Camera Trajectories10 Apr 2026
- 2026Training Without Orthogonalization, Inference With SVD: A Gradient Analysis of Rotation Representations7 Apr 2026
- 2025TAPVid-360: Tracking Any Point in 360 from Narrow Field of View Video26 Nov 2025
- 2025Matcha: Multi-Stage Riemannian Flow Matching for Accurate and Physically Valid Molecular Docking16 Oct 2025
- 2025A Primer on SO(3) Action Representations in Deep Reinforcement Learning13 Oct 2025
- 2025Monocular Visual Odometry without Calibration or Test-time Optimization2 Oct 2025
- 2025Towards Foundation Models for Cryo-ET Subtomogram Analysis29 Sep 2025
- 2025On the Skinning of Gaussian Avatars14 Sep 2025
12 newest of 63 All 63 (JSON)