Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

OT-MeanFlow3D: Bridging Optimal Transport and Meanflow for Efficient 3D Point Cloud Generation

About

Flow-matching models have recently emerged as a powerful framework for continuous generative modeling, including 3D point cloud synthesis. However, their deployment is limited by the need for multiple sequential sampling steps at inference time. MeanFlow enables single-step generation and significantly accelerates inference, but often struggles to approximate the trajectories of the original multi-step flow, leading to degraded sample quality. In this work, we propose an Optimal Transport-enhanced MeanFlow framework (OT-MF3D) for efficient and accurate 3D point cloud generation and completion. By incorporating optimal transport-based sampling, our method better preserves the geometric and distributional structure of the underlying multi-step flow while retaining single-step inference. Experiments on ShapeNet show improved generation and completion quality compared to recent baselines, while reducing training and inference costs relative to conventional diffusion and flow-based models.

Elaheh Akbari, Shansita Sharma, Ping He, Ahmadreza Moradipari, Kyungtae Han, Hamed Pirsiavash, Yikun Bai, Soheil Kolouri• 2025

Related benchmarks

TaskDatasetResultRank
Point cloud generationShapeNet Car
1-NNA (CD)60.79
41
3D Shape GenerationShapeNet airplane
1-NNA (CD)72.71
30
Point Cloud CompletionShapeNet (test)
EMD (Airplane)0.51
26
3D Shape GenerationShapeNet chair
1-NN Accuracy (CD)57.77
18
Shape GenerationShapeNet airplane
MMD (CD)0.221
11
Shape GenerationShapeNet Car
MMD (CD)0.918
11
Shape GenerationShapeNet chair
MMD (CD)2.656
11
Showing 7 of 7 rows

Other info

Follow for update