arXiv:2501.10789cs.CV2025-01被引 13

基于贡献度采样,提升点云简化在各类任务中的表现

CS-Net:Contribution-based Sampling Network for Point Cloud Simplification

  • 将采样建模为带贡献评分的Top-k操作,实现端到端可训练
  • 在ModelNet40和PU147上优于现有方法,分类与重建任务均达最优
  • 适合需要高精度点云简化且避免重复采样的应用场景

点云采样在降低视觉任务计算成本和存储需求方面至关重要。传统方法如最远点采样缺乏任务特定信息,难以保证特定应用下的最优性能。基于学习的方法虽能针对下游任务采样,但无法确保所选点最具相关性,且可能产生重复点,需后期补全处理。为此,本文提出贡献度采样网络(CS-Net),将采样定义为Top-k操作,并通过熵正则化最优传输问题实现可微近似,支持端到端训练。网络包含特征嵌入模块、级联注意力模块和贡献评分模块。特征嵌入模块设计空间池化层,在减少参数的同时保留关键特征;级联注意力模块融合三个跳跃连接的偏移注意力层输出,突出重要特征并抑制次要特征;贡献评分模块为每点生成贡献分,指导采样优先选择关键点。在ModelNet40和PU147数据集上的实验表明,CS-Net在两类语义任务(分类与配准)和两类重建任务(压缩与表面重建)中均达到当前最佳性能。

原文摘要 · Abstract (English)

Point cloud sampling plays a crucial role in reducing computation costs and storage requirements for various vision tasks. Traditional sampling methods, such as farthest point sampling, lack task-specific information and, as a result, cannot guarantee optimal performance in specific applications. Learning-based methods train a network to sample the point cloud for the targeted downstream task. However, they do not guarantee that the sampled points are the most relevant ones. Moreover, they may result in duplicate sampled points, which requires completion of the sampled point cloud through post-processing techniques. To address these limitations, we propose a contribution-based sampling network (CS-Net), where the sampling operation is formulated as a Top-k operation. To ensure that the network can be trained in an end-to-end way using gradient descent algorithms, we use a differentiable approximation to the Top-k operation via entropy regularization of an optimal transport problem. Our network consists of a feature embedding module, a cascade attention module, and a contribution scoring module. The feature embedding module includes a specifically designed spatial pooling layer to reduce parameters while preserving important features. The cascade attention module combines the outputs of three skip connected offset attention layers to emphasize the attractive features and suppress less important ones. The contribution scoring module generates a contribution score for each point and guides the sampling process to prioritize the most important ones. Experiments on the ModelNet40 and PU147 showed that CS-Net achieved state-of-the-art performance in two semantic-based downstream tasks (classification and registration) and two reconstruction-based tasks (compression and surface reconstruction).

点云简化贡献采样深度学习几何处理

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