arXiv:2509.08104cs.CVcs.AI2025-09NeurIPS被引 4

提出可微的自适应匹配损失,提升点云重建精度与效率。

APML: Adaptive Probabilistic Matching Loss for Robust 3D Point Cloud Reconstruction

  • 用温度调节的相似矩阵结合Sinkhorn迭代实现可微一对一匹配
  • 在ShapeNet等数据集上收敛更快,稀疏区域分布更优,性能不降反升
  • 无需调参,适配主流点云模型,适合需要高精度重建的研究者

点云预测任务(如形状补全、生成)的训练依赖于衡量预测点集与真实点集差异的损失函数。常用方法如Chamfer Distance(CD)、HyperCD和InfoCD基于最近邻匹配,常导致多对一对应,造成密集区域点堆积、稀疏区域覆盖不足,且因索引选择引入不可导操作,影响梯度优化。Earth Mover Distance(EMD)虽能实现一对一匹配并更好捕捉结构相似性,但其立方复杂度限制了实际应用。本文提出自适应概率匹配损失(APML),通过温度缩放的成对距离相似矩阵进行Sinkhorn迭代,实现完全可微的一对一匹配。我们分析计算出温度以保证最小分配概率,避免手动调参。APML近似二次时间复杂度,与基于CD的损失相当,且无非可导操作。将其集成到PoinTr、PCN、FoldingNet等先进架构,在ShapeNet基准及基于WiFi CSI的时空变换器(CSI2PC)生成人体点云任务中,均实现更快收敛、更优空间分布,尤其在低密度区域表现显著,定量指标提升或持平,无需额外超参数搜索。代码已开源:https://github.com/apm-loss/apml。

原文摘要 · Abstract (English)

Training deep learning models for point cloud prediction tasks such as shape completion and generation depends critically on loss functions that measure discrepancies between predicted and ground-truth point sets. Commonly used functions such as Chamfer Distance (CD), HyperCD, and InfoCD rely on nearest-neighbor assignments, which often induce many-to-one correspondences, leading to point congestion in dense regions and poor coverage in sparse regions. These losses also involve non-differentiable operations due to index selection, which may affect gradient-based optimization. Earth Mover Distance (EMD) enforces one-to-one correspondences and captures structural similarity more effectively, but its cubic computational complexity limits its practical use. We propose the Adaptive Probabilistic Matching Loss (APML), a fully differentiable approximation of one-to-one matching that leverages Sinkhorn iterations on a temperature-scaled similarity matrix derived from pairwise distances. We analytically compute the temperature to guarantee a minimum assignment probability, eliminating manual tuning. APML achieves near-quadratic runtime, comparable to Chamfer-based losses, and avoids non-differentiable operations. When integrated into state-of-the-art architectures (PoinTr, PCN, FoldingNet) on ShapeNet benchmarks and on a spatiotemporal Transformer (CSI2PC) that generates 3D human point clouds from WiFi CSI measurements, APM loss yields faster convergence, superior spatial distribution, especially in low-density regions, and improved or on-par quantitative performance without additional hyperparameter search. The code is available at: https://github.com/apm-loss/apml.

点云重建损失函数可微匹配深度学习

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