GraphUnet: Graph Make Strong Encoders for Remote Sensing Segmentation

Yuntao Shou,Wei Ai,Tao Meng,Fuchen Zhang,Keqin Li
DOI: https://doi.org/10.1109/icpads60453.2023.00364
2023-01-01
Abstract:Remote sensing segmentation are widely applied in environmental protection, and urban change detection, etc. Despite the success of deep learning-based remote sensing segmentation methods (e.g., CNN and Transformer), they are not flexible enough to model irregular objects. To tackle the above problems, this paper treats images as graph structures and introduces a simple vision GNN (GraphUNet) architecture for remote sensing segmentation. Specifically, we construct a graph view and utilize GNN to extract the visual feature. To extract the global contextual location information in the image, we introduce a multi-head attention mechanism for global information extraction. Furthermore, we introduce a feed-forward network for each node to perform feature transformation on node features to encourage information diversity. Extensive experiments on three real datasets demonstrate that our model GraphUNet outperforms SOTA remote sensing image segmentation methods.
What problem does this paper attempt to address?