Equivariant Manifold Flows
Not on the map. Why?
Abstract (arXiv)
Tractably modelling distributions over manifolds has long been an important goal in the natural sciences. Recent work has focused on developing general machine learning models to learn such distributions. However, for many applications these distributions must respect manifold symmetries -- a trait which most previous models disregard. In this paper, we lay the theoretical foundations for learning symmetry-invariant distributions on arbitrary manifolds via equivariant manifold flows. We demonstrate the utility of our approach by using it to learn gauge invariant densities over SU(n) in the context of quantum field theory.
Key facts. Cited by 20 papers in the map (data through 7 Oct 2026). Builds on 25 of its 49 references that are in the map. First posted on arXiv 19 Jul 2021.
Builds on
25 of 49 references.
References · 25
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2013Auto-Encoding Variational Bayescited by 3,980
- 2018Neural Ordinary Differential Equationscited by 3,733
- 2016Density estimation using Real NVPcited by 2,029
- 2017Deep Setscited by 1,394
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,315
- 2016Group Equivariant Convolutional Networkscited by 1,154
- 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,131
- 2019Neural Spline Flowscited by 590
- 2018Spherical CNNscited by 452
- 2018On the Generalization of Equivariance and Convolution in Neural Networks to the Action of Compact Groupscited by 296
13 more
- 2019Gauge Equivariant Convolutional Networks and the Icosahedral CNNcited by 262
- 2020Generalizing Convolutional Neural Networks for Equivariance to Lie Groups on Arbitrary Continuous Datacited by 248
- 2020Equivariant Flows: Exact Likelihood Generative Learning for Symmetric Densitiescited by 248
- 2020Riemannian Continuous Normalizing Flowscited by 143
- 2020Normalizing Flows on Tori and Spherescited by 117
- 2019Mixed-curvature Variational Autoencoderscited by 85
- 2020Sampling using SU(N) gauge equivariant flowscited by 83
- 2020Neural Manifold Ordinary Differential Equationscited by 81
- 2020Targeted free energy estimation via learned mappingscited by 66
- 2020Latent Variable Modelling with Hyperbolic Normalizing Flowscited by 49
- 2019Equivariant Hamiltonian Flowscited by 39
- 2020Neural Ordinary Differential Equations on Manifoldscited by 31
- 2019Backpropagation-Friendly Eigendecompositioncited by 27
Built on by
Citing papers per year: 2021: 2; 2022: 2; 2023: 8; 2024: 3; 2025: 2; 2026: 3.
20 papers.
Sort:
By area:
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generationcited by 147
- 2023Equivariant flow matchingcited by 131
- 2023EDGI: Equivariant Diffusion for Planning with Embodied Agentscited by 35
- 2023Rigid Body Flows for Sampling Molecular Crystal Structurescited by 27
- 2022Symmetry-Based Representations for Artificial and Biological General Intelligencecited by 26
- 2023Advances in machine-learning-based sampling motivated by lattice quantum chromodynamicscited by 23
- 2023Riemannian Residual Neural Networkscited by 21
- 2023Scaling Riemannian Diffusion Modelscited by 18
- 2025Space Group Equivariant Crystal Diffusioncited by 15
- 2021Equivariant Finite Normalizing Flowscited by 11
- 2021Implicit Riemannian Concave Potential Mapscited by 10
- 2024Ensembles provably learn equivariance through data augmentationcited by 8
8 more
- 2022Grassmann Manifold Flows for Stable Shape Generationcited by 3
- 2024Equivariant Manifold Neural ODEs and Differential Invariantscited by 2
- 2024Current Symmetry Group Equivariant Convolution Frameworks for Representation Learningcited by 2
- 2023Conformal Generative Modeling on Triangulated Surfacescited by 1
- 2026EquiDexFlow: Contact-Grounded SE(3)-Equivariant Dexterous Grasp Generative Flowscited by 1
- 2025Normalizing Flows on Quotient Manifolds via Boundary Quotientscited by 0
- 2026Steerable Neural ODEs on Homogeneous Spacescited by 0
- 2026Riemannian Flow Models with Reinforcement Learning for Molecular Crystal Structure Predictioncited by 0
- 2026Riemannian Flow Models with Reinforcement Learning for Molecular Crystal Structure Prediction30 Sep 2026
- 2026EquiDexFlow: Contact-Grounded SE(3)-Equivariant Dexterous Grasp Generative Flows10 Jun 2026
- 2026Steerable Neural ODEs on Homogeneous Spaces11 May 2026
- 2025Normalizing Flows on Quotient Manifolds via Boundary Quotients28 Nov 2025
- 2025Space Group Equivariant Crystal Diffusion16 May 2025
- 2024Ensembles provably learn equivariance through data augmentation2 Oct 2024
- 2024Current Symmetry Group Equivariant Convolution Frameworks for Representation Learning11 Sep 2024
- 2024Equivariant Manifold Neural ODEs and Differential Invariants25 Jan 2024
- 2023Scaling Riemannian Diffusion Models30 Oct 2023
- 2023Riemannian Residual Neural Networks16 Oct 2023
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generation3 Oct 2023
- 2023Advances in machine-learning-based sampling motivated by lattice quantum chromodynamics3 Sep 2023
8 more
- 2023Equivariant flow matching26 Jun 2023
- 2023EDGI: Equivariant Diffusion for Planning with Embodied Agents22 Mar 2023
- 2023Conformal Generative Modeling on Triangulated Surfaces17 Mar 2023
- 2023Rigid Body Flows for Sampling Molecular Crystal Structures26 Jan 2023
- 2022Grassmann Manifold Flows for Stable Shape Generation5 Nov 2022
- 2022Symmetry-Based Representations for Artificial and Biological General Intelligence17 Mar 2022
- 2021Equivariant Finite Normalizing Flows16 Oct 2021
- 2021Implicit Riemannian Concave Potential Maps4 Oct 2021
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generationcited by 147
- 2023Equivariant flow matchingcited by 131
- 2023EDGI: Equivariant Diffusion for Planning with Embodied Agentscited by 35
- 2023Scaling Riemannian Diffusion Modelscited by 18
- 2023Scaling Riemannian Diffusion Models30 Oct 2023
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generation3 Oct 2023
- 2023Equivariant flow matching26 Jun 2023
- 2023EDGI: Equivariant Diffusion for Planning with Embodied Agents22 Mar 2023
- 2023Rigid Body Flows for Sampling Molecular Crystal Structurescited by 27
- 2022Symmetry-Based Representations for Artificial and Biological General Intelligencecited by 26
- 2023Advances in machine-learning-based sampling motivated by lattice quantum chromodynamicscited by 23
- 2023Riemannian Residual Neural Networkscited by 21
- 2025Space Group Equivariant Crystal Diffusioncited by 15
- 2021Equivariant Finite Normalizing Flowscited by 11
- 2021Implicit Riemannian Concave Potential Mapscited by 10
- 2024Ensembles provably learn equivariance through data augmentationcited by 8
- 2022Grassmann Manifold Flows for Stable Shape Generationcited by 3
- 2024Equivariant Manifold Neural ODEs and Differential Invariantscited by 2
- 2024Current Symmetry Group Equivariant Convolution Frameworks for Representation Learningcited by 2
- 2023Conformal Generative Modeling on Triangulated Surfacescited by 1
12 most cited of 16 All 16 (JSON)
- 2026Riemannian Flow Models with Reinforcement Learning for Molecular Crystal Structure Prediction30 Sep 2026
- 2026EquiDexFlow: Contact-Grounded SE(3)-Equivariant Dexterous Grasp Generative Flows10 Jun 2026
- 2026Steerable Neural ODEs on Homogeneous Spaces11 May 2026
- 2025Normalizing Flows on Quotient Manifolds via Boundary Quotients28 Nov 2025
- 2025Space Group Equivariant Crystal Diffusion16 May 2025
- 2024Ensembles provably learn equivariance through data augmentation2 Oct 2024
- 2024Current Symmetry Group Equivariant Convolution Frameworks for Representation Learning11 Sep 2024
- 2024Equivariant Manifold Neural ODEs and Differential Invariants25 Jan 2024
- 2023Riemannian Residual Neural Networks16 Oct 2023
- 2023Advances in machine-learning-based sampling motivated by lattice quantum chromodynamics3 Sep 2023
- 2023Conformal Generative Modeling on Triangulated Surfaces17 Mar 2023
- 2023Rigid Body Flows for Sampling Molecular Crystal Structures26 Jan 2023
12 newest of 16 All 16 (JSON)