New Riemannian MeanFlow method enables faster generative model sampling

Researchers have introduced Riemannian MeanFlow (RMF), a novel method for generative models operating on Riemannian manifolds. Unlike previous approaches that require extensive simulation for sampling, RMF enables one-step generation by defining an average-velocity field through parallel transport. This method is practical in a log-map tangent representation, reducing computational costs and avoiding trajectory simulation. Experiments on various manifolds, including spheres, tori, SO(3), and SE(3), show that RMF achieves competitive sampling quality with improved efficiency and lower costs, and also supports conditional generation. AI

IMPACT Introduces a more efficient method for generative models on complex data structures, potentially speeding up training and sampling.

RANK_REASON Academic paper detailing a new method for generative models. [lever_c_demoted from research: ic=1 ai=1.0]

Read on arXiv cs.LG →

• arXiv
• DagsHub
• Flow Matching for Generative Modeling
• Haoliang Sun
• Hugging Face
• Riemannian MeanFlow
• rotation group SO(3)
• SE(3)-Transformers: 3D Roto-Translation Equivariant Attention Networks

• paper
• model release

AI-generated summary · Google Gemini · from 1 sources. How we write summaries →