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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Point cloud generation | ShapeNet Car | 1-NNA (CD)60.79 | 41 | |
| 3D Shape Generation | ShapeNet airplane | 1-NNA (CD)72.71 | 30 | |
| Point Cloud Completion | ShapeNet (test) | EMD (Airplane)0.51 | 26 | |
| 3D Shape Generation | ShapeNet chair | 1-NN Accuracy (CD)57.77 | 18 | |
| Shape Generation | ShapeNet airplane | MMD (CD)0.221 | 11 | |
| Shape Generation | ShapeNet Car | MMD (CD)0.918 | 11 | |
| Shape Generation | ShapeNet chair | MMD (CD)2.656 | 11 |