Grassmann Manifold Flows for Stable Shape Generation
Ryoma Yataka, Kazuki Hirashima, Masashi Shiraishi
Abstract
Recently, studies on machine learning have focused on methods that use symmetry implicit in a specific manifold as an inductive bias. Grassmann manifolds provide the ability to handle fundamental shapes represented as shape spaces, enabling stable shape analysis. In this paper, we present a novel approach in which we establish the theoretical foundations for learning distributions on the Grassmann manifold via continuous normalization flows, with the explicit goal of generating stable shapes. Our approach facilitates more robust generation by effectively eliminating the influence of extraneous transformations, such as rotations and inversions, through learning and generating within a Grassmann manifold designed to accommodate the essential shape information of the object. The experimental results indicated that the proposed method could generate high-quality samples by capturing the data structure. Furthermore, the proposed method significantly outperformed state-of-the-art methods in terms of the log-likelihood or evidence lower bound. The results obtained are expected to stimulate further research in this field, leading to advances for stable shape generation and analysis.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- Stiefel Flow Matching for Moment-Constrained Structure ElucidationAustin Henry Cheng, Alston Lo, Kin Long Kelvin Lee, Santiago Miret et al.ICLR 2025 · 2 citations
- Registration beyond Points: General Affine Subspace Alignment via Geodesic Distance on Grassmann ManifoldJaeho Shin, Hyeonjae Gil, Junwoo Jang, Maani Ghaffari et al.ICCV 2025
Builds on12
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Denoising Diffusion Implicit ModelsJiaming Song, Chenlin Meng, Stefano ErmonICLR 2021 · 11,743 citations
- Generalizing Convolutional Neural Networks for Equivariance to Lie Groups on Arbitrary Continuous DataMarc Finzi, Samuel Stanton, Pavel Izmailov, Andrew Gordon WilsonICML 2020 · 372 citations
- Equivariant Flows: Exact Likelihood Generative Learning for Symmetric DensitiesJonas Köhler, Leon Klein, Frank NoéICML 2020 · 330 citations
- Riemannian Score-Based Generative ModellingValentin De Bortoli, Emile Mathieu, Michael J. Hutchinson, James Thornton et al.NeurIPS 2022 · 306 citations
Related papers
- Tractable Density Estimation on Learned Manifolds with Conformal Embedding FlowsBrendan Leigh Ross, Jesse C. CresswellNeurIPS 2021 · 39 citations
- Delving into Discrete Normalizing Flows on SO(3) Manifold for Probabilistic Rotation ModelingYulin Liu, Haoran Liu, Yingda Yin, Yang Wang et al.CVPR 2023
- Flows for simultaneous manifold learning and density estimationJohann Brehmer, Kyle CranmerNeurIPS 2020 · 187 citations
- SoftFlow: Probabilistic Framework for Normalizing Flow on ManifoldsHyeongju Kim, Hyeonseung Lee, Woo Hyun Kang, Joun Yeop Lee et al.NeurIPS 2020 · 149 citations
- Discovering Symmetry Groups with Flow MatchingYuxuan Chen, Jung Yeon Park, Floor Eijkelboom, Jianke Yang et al.ICML 2026 · 3 citations
