arXiv:2502.12456cs.CVcs.AI2025-02ICLR被引 17

提出近似最优传输流,高效生成3D点云并提升形状补全性能

Not-So-Optimal Transport Flows for 3D Point Cloud Generation

  • 通过离线预计算近似最优传输对,降低训练复杂度
  • 在ShapeNet上优于现有扩散与流模型,尤其在大点云生成中表现更优
  • 适合需要高效3D点云生成的科研与工业场景

学习3D点云的生成模型是3D生成学习的核心问题之一。点云的关键特性之一是排列不变性,即点的顺序改变不会影响其表示的形状。本文分析了近期提出的等变最优传输流(equivariant OT flows),用于基于点的分子数据生成,并发现这些模型在大规模点云上扩展性差。同时观察到,学习(等变)最优传输流普遍困难,因为拉直流轨迹会使初始阶段的模型变得复杂。为解决这些问题,我们提出“非最优传输流”(not-so-optimal transport flows),通过离线预计算近似最优传输对,实现训练时高效的OT配对构建。训练过程中,还可结合近似最优传输与独立耦合构造混合耦合,使目标流模型更易学习。在广泛的实验中,我们的模型在ShapeNet基准上的无条件生成和形状补全任务中均优于现有的扩散模型和流模型。

原文摘要 · Abstract (English)

Learning generative models of 3D point clouds is one of the fundamental problems in 3D generative learning. One of the key properties of point clouds is their permutation invariance, i.e., changing the order of points in a point cloud does not change the shape they represent. In this paper, we analyze the recently proposed equivariant OT flows that learn permutation invariant generative models for point-based molecular data and we show that these models scale poorly on large point clouds. Also, we observe learning (equivariant) OT flows is generally challenging since straightening flow trajectories makes the learned flow model complex at the beginning of the trajectory. To remedy these, we propose not-so-optimal transport flow models that obtain an approximate OT by an offline OT precomputation, enabling an efficient construction of OT pairs for training. During training, we can additionally construct a hybrid coupling by combining our approximate OT and independent coupling to make the target flow models easier to learn. In an extensive empirical study, we show that our proposed model outperforms prior diffusion- and flow-based approaches on a wide range of unconditional generation and shape completion on the ShapeNet benchmark.

3D生成点云流模型最优传输

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