中文

OPTNet: 排序点变换网络用于灾后 3D 语义分割

机器学习 2026-05-19 v1 计算机视觉与模式识别

摘要

Post-disaster damage assessment requires rapid and accurate semantic segmentation of 3D point clouds to identify critical infrastructure such as damaged buildings and roads. Early Point Transformers (e.g., PTv1, PTv2) relied on computationally expensive neighbor searching (k-NN) and Farthest Point Sampling (FPS). To improve efficiency, recent architectures like Point Transformer V3 (PTv3) adopted static serialization methods, such as Hilbert curves or Z-order, to organize unstructured points for window-based attention. However, these fixed orderings are not optimal for capturing the complex geometry of disaster scenes. In this paper, we propose OPTNet (Ordering Point Transformer Network), which introduces a learnable Point Sorter module. OPTNet utilizes a self-supervised ordering loss to dynamically predict an optimal permutation that maximizes the locality of the attention mechanism. We evaluate our method on the 3DAeroRelief dataset, significantly outperforming state-of-the-art baselines.

关键词

引用

@article{arxiv.2605.17197,
  title  = {OPTNet: Ordering Point Transformer Network for Post-disaster 3D Semantic Segmentation},
  author = {Nhut Le and Ehsan Karimi and Maryam Rahnemoonfar},
  journal= {arXiv preprint arXiv:2605.17197},
  year   = {2026}
}

备注

Accepted for International Conference on Pattern Recognition (ICPR) 2026