arXiv:2411.15271cs.CVcs.AI2024-11被引 2

用自回归扩散模型高效实现户外点云配准,提升稀疏点云下的匹配鲁棒性。

EADReg: Probabilistic Correspondence Generation with Efficient Autoregressive Diffusion Model for Outdoor Point Cloud Registration

  • 分粗细两阶段:先用双向高斯混合模型剔除离群点,再用自回归扩散生成对应关系。
  • 在KITTI和NuScenes数据集上达到当前最优性能,推理速度媲美卷积方法。
  • 适合处理稀疏、不规则的车载激光雷达点云,对挑战性场景有强鲁棒性。

扩散模型在点云配准(PCR)任务中展现出巨大潜力,尤其在增强对复杂情况的鲁棒性方面。然而,现有基于扩散的PCR方法主要针对实例级场景,在处理户外激光雷达点云时仍面临挑战,因为其具有稀疏性、不规则性和大规模点数的特点,难以建立密集的全局点对点对应关系。为此,我们提出一种名为EADReg的新框架,基于自回归扩散模型实现高效且鲁棒的激光雷达点云配准。EADReg采用从粗到精的配准范式:在粗略阶段,使用双向高斯混合模型(BGMM)剔除离群点,获得净化后的点云对;BGMM在源帧与目标帧的高斯混合模型(GMM)之间建立对应关系,基于过滤后的特征与几何信息实现可靠粗配准。在精细阶段,将基于扩散的PCR视为自回归过程,逐步生成鲁棒的点对应关系,并在高层迭代优化。尽管扩散方法常被诟病推理速度慢,但EADReg的运行时间可与卷积基方法相媲美。在KITTI和NuScenes基准数据集上的大量实验验证了所提方法的领先性能。代码将在发表后公开。

原文摘要 · Abstract (English)

Diffusion models have shown the great potential in the point cloud registration (PCR) task, especially for enhancing the robustness to challenging cases. However, existing diffusion-based PCR methods primarily focus on instance-level scenarios and struggle with outdoor LiDAR points, where the sparsity, irregularity, and huge point scale inherent in LiDAR points pose challenges to establishing dense global point-to-point correspondences. To address this issue, we propose a novel framework named EADReg for efficient and robust registration of LiDAR point clouds based on autoregressive diffusion models. EADReg follows a coarse-to-fine registration paradigm. In the coarse stage, we employ a Bi-directional Gaussian Mixture Model (BGMM) to reject outlier points and obtain purified point cloud pairs. BGMM establishes correspondences between the Gaussian Mixture Models (GMMs) from the source and target frames, enabling reliable coarse registration based on filtered features and geometric information. In the fine stage, we treat diffusion-based PCR as an autoregressive process to generate robust point correspondences, which are then iteratively refined on upper layers. Despite common criticisms of diffusion-based methods regarding inference speed, EADReg achieves runtime comparable to convolutional-based methods. Extensive experiments on the KITTI and NuScenes benchmark datasets highlight the state-of-the-art performance of our proposed method. Codes will be released upon publication.

点云配准扩散模型自回归激光雷达

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