HiT: Building Mapping With Hierarchical Transformers

Mingming Zhang,Qingjie Liu,Yunhong Wang

DOI: https://doi.org/10.1109/tgrs.2024.3355274

IF: 8.2

2024-02-02

IEEE Transactions on Geoscience and Remote Sensing

Abstract:Deep-learning-based methods have been extensively explored for automatic building mapping from high-resolution remote sensing images over recent years. While most building mapping models produce vector polygons of buildings for geographic and mapping systems, dominant methods typically decompose polygonal building extraction in some subproblems, including segmentation, polygonization, and regularization, leading to complex inference procedures, low accuracy, and poor generalization. In this article, we propose a simple and novel building mapping method with hierarchical transformers, called HiT, improving polygonal building mapping quality from high-resolution remote sensing images. HiT builds on a two-stage detection architecture by adding a polygon head parallel to classification and bounding box regression heads. HiT simultaneously outputs building bounding boxes and vector polygons, which is fully end-to-end trainable. The polygon head formulates a building polygon as serialized vertices with the bidirectional characteristic, a simple and elegant polygon representation avoiding the start or end vertex hypothesis. Under this new perspective, the polygon head adopts a transformer encoder–decoder architecture to predict serialized vertices supervised by the designed bidirectional polygon loss. Furthermore, a hierarchical attention mechanism combined with convolution operation is introduced in the encoder of the polygon head, providing more geometric structures of building polygons at vertex and edge levels. Comprehensive experiments on two benchmarks (the CrowdAI and Inria datasets) demonstrate that our method achieves a new state-of-the-art (SOTA) in terms of instance segmentation and polygonal metrics compared with SOTA methods. Moreover, qualitative results verify the superiority and effectiveness of our model under complex scenes.

imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics