arXiv:2510.00983cs.LG2025-10NeurIPS被引 3

首次实现曲面流形上的快速生成,保持几何约束。

Riemannian Consistency Model

  • 基于协变导数与指数映射,构建曲面几何下的生成模型
  • 在球面、环面和SO(3)上实现少步生成,质量优于传统方法
  • 理论证明两种训练方式等价,简化计算流程

一致性模型是一类可在少步内生成图像等欧氏域数据的生成模型。然而,由于曲面几何的复杂性,其在黎曼流形上的应用仍具挑战。本文提出黎曼一致性模型(RCM),首次实现尊重流形内在约束的少步一致性建模。通过协变导数与基于指数映射的参数化,推导出离散与连续时间训练目标的闭式解。理论证明了依赖教师模型的黎曼一致性蒸馏(RCD)与使用条件向量场的黎曼一致性训练(RCT)等价。进一步提出简化训练目标,避免复杂的微分计算。最后从运动学角度解释RCM目标,提供新理论视角。大量实验表明,RCM在平坦环面、球面及三维旋转群SO(3)等非欧流形上,均实现优异的少步生成性能。

原文摘要 · Abstract (English)

Consistency models are a class of generative models that enable few-step generation for diffusion and flow matching models. While consistency models have achieved promising results on Euclidean domains like images, their applications to Riemannian manifolds remain challenging due to the curved geometry. In this work, we propose the Riemannian Consistency Model (RCM), which, for the first time, enables few-step consistency modeling while respecting the intrinsic manifold constraint imposed by the Riemannian geometry. Leveraging the covariant derivative and exponential-map-based parameterization, we derive the closed-form solutions for both discrete- and continuous-time training objectives for RCM. We then demonstrate theoretical equivalence between the two variants of RCM: Riemannian consistency distillation (RCD) that relies on a teacher model to approximate the marginal vector field, and Riemannian consistency training (RCT) that utilizes the conditional vector field for training. We further propose a simplified training objective that eliminates the need for the complicated differential calculation. Finally, we provide a unique kinematics perspective for interpreting the RCM objective, offering new theoretical angles. Through extensive experiments, we manifest the superior generative quality of RCM in few-step generation on various non-Euclidean manifolds, including flat-tori, spheres, and the 3D rotation group SO(3).

生成模型黎曼几何少步生成流形学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。