基于Mamba与自注意力机制的点云上采样方法

    Effective Point Cloud Upsampling Method Based on Mamba and Self-attention Mechanism

    • 摘要: 针对现有点云上采样方法在处理复杂几何结构时难以兼顾全局拓扑结构与局部细节特征,导致生成点云存在分布不均及离群点的问题,本文提出了一种融合选择性状态空间模型(Linear-Time Sequence Modeling with Selective State Spaces,Mamba)与自注意力机制的点云上采样网络(Effective Point Cloud Upsampling Network based on Mamba and Self-attention,PointMamba-UPS)。该方法设计了双分支特征提取架构:一方面,利用基于Mamba块的全局特征提取器,通过特征重排序策略适配点云无序性,在保持线性复杂度的同时建立长距离依赖,捕捉物体的整体形状与语义信息;另一方面,构建基于多头自注意力机制的局部特征提取器,自适应聚合邻域内的几何相似特征,精确刻画边缘与曲率变化。此外,网络采用了对称的密集连接融合结构,实现了全局语义先验与局部几何细节的深度融合。实验结果表明,本文方法在PU1K数据集上相较于PU-Transformer(点云上采样Transformer,Point Cloud Upsampling Transformer),倒角距离、豪斯多夫距离及点到面距离三个指标分别降低5.1%、3.3%和6.9%。同时在推理时间上降低了16.1%。本文方法在多项评价指标上均优于现有主流方法,充分验证了双分支协同架构在点云上采样任务中的有效性与实用性。

       

      Abstract: Point cloud upsampling plays a critical role in enhancing the resolution and quality of 3D data captured by sensors. However, existing methods often struggle to simultaneously preserve global topological structures and recover local geometric details, frequently leading to non-uniform point distributions and outlier artifacts. To overcome these limitations, this paper proposes a novel upsampling network, termed PointMamba-UPS (Effective Point Cloud Upsampling Network based on Mamba and Self-attention), which integrates the Mamba (Linear-Time Sequence Modeling with Selective State Spaces)architecture with self-attention mechanisms. The method employs a dual-branch feature extraction framework. Specifically, a global feature extractor built upon Mamba blocks is designed to capture the overall shape and semantic information of objects. By utilizing a feature reordering strategy to accommodate the disorder inherent in point clouds, this branch establishes long-range dependencies while maintaining linear computational complexity. In parallel, a local feature extractor based on multi-head self-attention mechanism is constructed to adaptively aggregate geometrically similar features within neighborhoods, thereby precisely characterizing edge and curvature variations. Furthermore, a symmetric dense connection fusion structure is adopted to achieve deep integration of global semantic priors and local geometric details. Experimental results demonstrate that, compared with PU-Transformer (Point Cloud Upsampling Transformer), the proposed method reduces the Chamfer Distance, Hausdorff Distance, and Point-to-Surface Distance by 5.1%, 3.3%, and 6.9%, respectively, on the PU1K dataset, while decreasing the inference time by 16.1%. The proposed method outperforms existing mainstream methods across multiple evaluation metrics, fully validating the effectiveness and practicality of the dual-branch collaborative architecture for point cloud upsampling.

       

    /

    返回文章
    返回