arXiv:2412.17573cs.CV2024-12被引 1

提出新型稀疏注意力U-Net,高效分割复杂道路网络。

URoadNet: Dual Sparse Attentive U-Net for Multiscale Road Network Extraction

  • 引入双稀疏注意力机制,捕捉道路局部连通与全局拓扑。
  • 在多个遥感数据集上超越现有方法,精度显著提升。
  • 计算量低,适合大规模道路提取,适合遥感图像分析者。

道路网络分割面临稀疏不规则形状和多变上下文的挑战,传统编码解码结构与简单Transformer常失效。本文提出URoadNet,一种计算高效且强大的道路感知分割框架。该方法通过连接性注意力有效编码细粒度道路连通性,结合整体性注意力学习道路与背景间的全局交互,二者交替互补并联合训练,显著提升性能而无需大幅增加计算开销。在马萨诸塞、DeepGlobe、SpaceNet及大尺度遥感图像等多分辨率数据集上的大量实验表明,该方法优于当前最先进技术。本工作为道路网络提取提供了高精度、可计算的解决方案。

原文摘要 · Abstract (English)

The challenges of road network segmentation demand an algorithm capable of adapting to the sparse and irregular shapes, as well as the diverse context, which often leads traditional encoding-decoding methods and simple Transformer embeddings to failure. We introduce a computationally efficient and powerful framework for elegant road-aware segmentation. Our method, called URoadNet, effectively encodes fine-grained local road connectivity and holistic global topological semantics while decoding multiscale road network information. URoadNet offers a novel alternative to the U-Net architecture by integrating connectivity attention, which can exploit intra-road interactions across multi-level sampling features with reduced computational complexity. This local interaction serves as valuable prior information for learning global interactions between road networks and the background through another integrality attention mechanism. The two forms of sparse attention are arranged alternatively and complementarily, and trained jointly, resulting in performance improvements without significant increases in computational complexity. Extensive experiments on various datasets with different resolutions, including Massachusetts, DeepGlobe, SpaceNet, and Large-Scale remote sensing images, demonstrate that URoadNet outperforms state-of-the-art techniques. Our approach represents a significant advancement in the field of road network extraction, providing a computationally feasible solution that achieves high-quality segmentation results.

道路提取注意力机制遥感图像U-Net

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