Trivializations for Gradient-Based Optimization on Manifolds
Not on the map. Why?
Abstract (arXiv)
We introduce a framework to study the transformation of problems with manifold constraints into unconstrained problems through parametrizations in terms of a Euclidean space. We call these parametrizations "trivializations". We prove conditions under which a trivialization is sound in the context of gradient-based optimization and we show how two large families of trivializations have overall favorable properties, but also suffer from a performance issue. We then introduce "dynamic trivializations", which solve this problem, and we show how these form a family of optimization methods that lie between trivializations and Riemannian gradient descent, and combine the benefits of both of them. We then show how to implement these two families of trivializations in practice for different matrix manifolds. To this end, we prove a formula for the gradient of the exponential of matrices, which can be of practical interest on its own. Finally, we show how dynamic trivializations improve the performance of existing methods on standard tasks designed to test long-term memory within neural networks.
Key facts. Cited by 78 papers in the map (data through 7 Oct 2026). First posted on arXiv 20 Sep 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 19; 2022: 8; 2023: 15; 2024: 10; 2025: 16; 2026: 10.
78 papers.
Sort:
By area:
- 2021Orthogonalizing Convolutional Layers with the Cayley Transformcited by 105
- 2021Modeling Irregular Time Series with Continuous Recurrent Unitscited by 101
- 2021RotoGrad: Gradient Homogenization in Multitask Learningcited by 82
- 2021Scaling Scaling Laws with Board Gamescited by 62
- 2021UnICORNN: A recurrent model for learning very long time dependenciescited by 56
- 2021Universal Approximation Theorems for Differentiable Geometric Deep Learningcited by 41
- 2021Long Expressive Memory for Sequence Modelingcited by 41
- 2021Fast and accurate optimization on the orthogonal manifold without retractioncited by 35
- 2023Deep networks for system identification: a Surveycited by 35
- 2021Tractable Density Estimation on Learned Manifolds with Conformal Embedding Flowscited by 30
- 2023RAYEN: Imposition of Hard Convex Constraints on Neural Networkscited by 30
- 2022Contrastive Laplacian Eigenmapscited by 24
38 more
- 2024Lie Algebra Canonicalization: Equivariant Neural Operators under arbitrary Lie Groupscited by 23
- 2021Tractable structured natural gradient descent using local parameterizationscited by 22
- 2025Load Balancing Mixture of Experts with Similarity Preserving Routerscited by 22
- 2023Infeasible Deterministic, Stochastic, and Variance-Reduction Algorithms for Optimization under Orthogonality Constraintscited by 18
- 2023Exploiting Noise as a Resource for Computation and Learning in Spiking Neural Networkscited by 17
- 2023Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraintscited by 15
- 2021Nested Hyperbolic Spaces for Dimensionality Reduction and Hyperbolic NN Designcited by 14
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Framecited by 14
- 2021An Operator Theoretic Approach for Analyzing Sequence Neural Networkscited by 12
- 2021How Does Momentum Benefit Deep Neural Networks Architecture Design? A Few Case Studiescited by 10
- 2023Simplifying Momentum-based Positive-definite Submanifold Optimization with Applications to Deep Learningcited by 10
- 2022O-ViT: Orthogonal Vision Transformercited by 9
- 2024Amortized Control of Continuous State Space Feynman-Kac Model for Irregular Time Seriescited by 9
- 2022Path Development Network with Finite-dimensional Lie Group Representationcited by 8
- 2021Scaling-up Diverse Orthogonal Convolutional Networks with a Paraunitary Frameworkcited by 7
- 2023Bootstrapping Parallel Anchors for Relative Representationscited by 7
- 2023A Study of Neural Collapse Phenomenon: Grassmannian Frame, Symmetry and Generalizationcited by 7
- 2024Learning Hamiltonian neural Koopman operator and simultaneously sustaining and discovering conservation lawcited by 7
- 2022Multi-View Independent Component Analysis with Shared and Individual Sourcescited by 6
- 2023Uncovering Challenges of Solving the Continuous Gromov-Wasserstein Problemcited by 6
- 2021Coordinate descent on the orthogonal group for recurrent neural network trainingcited by 5
- 2022Feedback Gradient Descent: Efficient and Stable Optimization with Orthogonality for DNNscited by 5
- 2022Random orthogonal additive filters: a solution to the vanishing/exploding gradient of deep neural networkscited by 5
- 2022Gated Recurrent Neural Networks with Weighted Time-Delay Feedbackcited by 5
- 2023Unsupervised Interpretable Basis Extraction for Concept-Based Visual Explanationscited by 5
- 2025SPD Matrix Learning for Neuroimaging Analysis: Perspectives, Methods, and Challengescited by 5
- 2025How to Square Tensor Networks and Circuits Without Squaring Themcited by 5
- 2026Neural Networks on Symmetric Spaces of Noncompact Typecited by 5
- 2021Accelerating Training of Batch Normalization: A Manifold Perspectivecited by 4
- 2021Deep Decomposition for Stochastic Normal-Abnormal Transportcited by 4
- 2024Riemannian coordinate descent algorithms on matrix manifoldscited by 4
- 2024Fast and Stable Riemannian Metrics on SPD Manifolds via Cholesky Product Geometrycited by 4
- 2025Position: Adopt Constraints Over Fixed Penalties in Deep Learningcited by 4
- 2025Efficient Robust Conformal Prediction via Lipschitz-Bounded Networkscited by 4
- 2026JPmHC Dynamical Isometry via Orthogonal Hyper-Connectionscited by 4
- 2021A Differential Geometry Perspective on Orthogonal Recurrent Modelscited by 3
- 2022Geometry-Aware Supertagging with Heterogeneous Dynamic Convolutionscited by 3
- 2023Get the Best of Both Worlds: Improving Accuracy and Transferability by Grassmann Class Representationcited by 3
50 most cited of 78 All 78 (JSON)
- 2026Projection Pursuit CPCANet for Domain Generalization24 Jul 2026
- 2026ManifoldFlow: SPD-Relaxed Stiefel Layers with Learnable Singular Spectrum5 Jul 2026
- 2026LieBN: Batch Normalization over Lie Groups13 Jun 2026
- 2026Riemannian Networks over Full-Rank Correlation Matrices18 May 2026
- 2026Unraveling the Mechanism of Drug Binding to SARS-CoV-2 RNA Pseudoknot with Thermodynamics-Driven Machine Learning16 Apr 2026
- 2026SPD Learn: A Geometric Deep Learning Python Library for Neural Decoding Through Trivialization26 Feb 2026
- 2026JPmHC Dynamical Isometry via Orthogonal Hyper-Connections20 Feb 2026
- 2026An Embarrassingly Simple Way to Optimize Orthogonal Matrices at Scale16 Feb 2026
- 2026Inverting Data Transformations via Diffusion Sampling9 Feb 2026
- 2026Neural Networks on Symmetric Spaces of Noncompact Type3 Jan 2026
- 2025Intrinsic-Metric Physics-Informed Neural Networks (IM-PINN) for Reaction-Diffusion Dynamics on Complex Riemannian Manifolds26 Dec 2025
- 2025How to Square Tensor Networks and Circuits Without Squaring Them18 Dec 2025
38 more
- 2025Knowledge Diversion for Efficient Morphology Control and Policy Transfer10 Dec 2025
- 2025PEOAT: Personalization-Guided Evolutionary Question Assembly for One-Shot Adaptive Testing29 Nov 2025
- 2025Randomized HyperSteiner: A Stochastic Delaunay Triangulation Heuristic for the Hyperbolic Steiner Minimal Tree10 Oct 2025
- 2025humancompatible.train: Implementing Optimization Algorithms for Stochastically-Constrained Stochastic Optimization Problems25 Sep 2025
- 2025Neuro-Spectral Architectures for Causal Physics-Informed Networks5 Sep 2025
- 2025rETF-semiSL: Semi-Supervised Learning for Neural Collapse in Temporal Data13 Aug 2025
- 2025Load Balancing Mixture of Experts with Similarity Preserving Routers16 Jun 2025
- 2025Efficient Robust Conformal Prediction via Lipschitz-Bounded Networks5 Jun 2025
- 2025Position: Adopt Constraints Over Fixed Penalties in Deep Learning27 May 2025
- 2025Equivariant Eikonal Neural Networks: Grid-Free, Scalable Travel-Time Prediction on Homogeneous Spaces21 May 2025
- 2025SPD Matrix Learning for Neuroimaging Analysis: Perspectives, Methods, and Challenges26 Apr 2025
- 2025Stochastic Optimal Control for Continuous-Time fMRI Representation Learning7 Feb 2025
- 2025Supervised Quadratic Feature Analysis: Information Geometry Approach for Dimensionality Reduction31 Jan 2025
- 2025Hyperbolic Binary Neural Network7 Jan 2025
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Frame2 Nov 2024
- 2024Amortized Control of Continuous State Space Feynman-Kac Model for Irregular Time Series8 Oct 2024
- 2024Lie Algebra Canonicalization: Equivariant Neural Operators under arbitrary Lie Groups3 Oct 2024
- 2024Robust Multi-view Co-expression Network Inference30 Sep 2024
- 2024Fast and Stable Riemannian Metrics on SPD Manifolds via Cholesky Product Geometry2 Jul 2024
- 2024Noisy Data Visualization using Functional Data Analysis5 Jun 2024
- 2024Riemannian coordinate descent algorithms on matrix manifolds4 Jun 2024
- 2024Learning Hamiltonian neural Koopman operator and simultaneously sustaining and discovering conservation law4 Jun 2024
- 2024Neural Controlled Differential Equations with Quantum Hidden Evolutions30 Apr 2024
- 2024Optimizing Medication Decisions for Patients with Atrial Fibrillation through Path Development Network18 Jan 2024
- 2023Deep Hashing via Householder Quantization7 Nov 2023
- 2023Manifold-Preserving Transformers are Effective for Short-Long Range Encoding22 Oct 2023
- 2023Exact and general decoupled solutions of the LMC Multitask Gaussian Process model18 Oct 2023
- 2023A survey of manifold learning and its applications for multimedia8 Sep 2023
- 2023Get the Best of Both Worlds: Improving Accuracy and Transferability by Grassmann Class Representation3 Aug 2023
- 2023RAYEN: Imposition of Hard Convex Constraints on Neural Networks17 Jul 2023
- 2023Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraints16 Jun 2023
- 2023Exploiting Noise as a Resource for Computation and Learning in Spiking Neural Networks25 May 2023
- 2023A Study of Neural Collapse Phenomenon: Grassmannian Frame, Symmetry and Generalization18 Apr 2023
- 2023Infeasible Deterministic, Stochastic, and Variance-Reduction Algorithms for Optimization under Orthogonality Constraints29 Mar 2023
- 2023Unsupervised Interpretable Basis Extraction for Concept-Based Visual Explanations19 Mar 2023
- 2023Uncovering Challenges of Solving the Continuous Gromov-Wasserstein Problem10 Mar 2023
- 2023Bootstrapping Parallel Anchors for Relative Representations1 Mar 2023
- 2023Simplifying Momentum-based Positive-definite Submanifold Optimization with Applications to Deep Learning20 Feb 2023
50 newest of 78 All 78 (JSON)
- 2026Inverting Data Transformations via Diffusion Samplingcited by 2
- 2026Inverting Data Transformations via Diffusion Sampling9 Feb 2026
- 2021Orthogonalizing Convolutional Layers with the Cayley Transformcited by 105
- 2021Modeling Irregular Time Series with Continuous Recurrent Unitscited by 101
- 2021RotoGrad: Gradient Homogenization in Multitask Learningcited by 82
- 2021Scaling Scaling Laws with Board Gamescited by 62
- 2021UnICORNN: A recurrent model for learning very long time dependenciescited by 56
- 2021Universal Approximation Theorems for Differentiable Geometric Deep Learningcited by 41
- 2021Long Expressive Memory for Sequence Modelingcited by 41
- 2021Fast and accurate optimization on the orthogonal manifold without retractioncited by 35
- 2023Deep networks for system identification: a Surveycited by 35
- 2021Tractable Density Estimation on Learned Manifolds with Conformal Embedding Flowscited by 30
- 2023RAYEN: Imposition of Hard Convex Constraints on Neural Networkscited by 30
- 2022Contrastive Laplacian Eigenmapscited by 24
12 most cited of 76 All 76 (JSON)
- 2026Projection Pursuit CPCANet for Domain Generalization24 Jul 2026
- 2026ManifoldFlow: SPD-Relaxed Stiefel Layers with Learnable Singular Spectrum5 Jul 2026
- 2026LieBN: Batch Normalization over Lie Groups13 Jun 2026
- 2026Riemannian Networks over Full-Rank Correlation Matrices18 May 2026
- 2026Unraveling the Mechanism of Drug Binding to SARS-CoV-2 RNA Pseudoknot with Thermodynamics-Driven Machine Learning16 Apr 2026
- 2026SPD Learn: A Geometric Deep Learning Python Library for Neural Decoding Through Trivialization26 Feb 2026
- 2026JPmHC Dynamical Isometry via Orthogonal Hyper-Connections20 Feb 2026
- 2026An Embarrassingly Simple Way to Optimize Orthogonal Matrices at Scale16 Feb 2026
- 2026Neural Networks on Symmetric Spaces of Noncompact Type3 Jan 2026
- 2025Intrinsic-Metric Physics-Informed Neural Networks (IM-PINN) for Reaction-Diffusion Dynamics on Complex Riemannian Manifolds26 Dec 2025
- 2025How to Square Tensor Networks and Circuits Without Squaring Them18 Dec 2025
- 2025Knowledge Diversion for Efficient Morphology Control and Policy Transfer10 Dec 2025
12 newest of 76 All 76 (JSON)