Few-Shot Aerial Image Semantic Segmentation Leveraging Pyramid Correlation Fusion

Wei Ao,Shunyi Zheng,Yan Meng,Zhi Gao
DOI: https://doi.org/10.1109/tgrs.2023.3328339
IF: 8.2
2023-11-10
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Few-shot semantic segmentation (FSS) has gained significant attention due to its ability to segment novel objects using only a limited number of labeled samples, thereby addressing the problem of overfitting caused by a lack of training data. Although this technique is widely studied in the field of computer vision, there are few methods for remote-sensing images. Prevalent FSS methods can achieve remarkable results for natural images, but they are difficult to apply to remote-sensing image processing because existing methods rarely take into consideration the large-scale and resolution differences in remote-sensing images. Consequently, it is hard for them to obtain correct semantic guidance from a few annotated remote-sensing images. To tackle these problems, this article proposes the pyramid correlation fusion network (PCFNet) to promote the ability to mine helpful information by calculating multiscale pixel-wise semantic correspondence. Particularly, the dual-distance correlation (DDC) module is designed to simultaneously compute the cosine similarity and Euclidean distance between query features and support features, producing adequate guidance information to determine the category of each pixel. Moreover, to improve segmentation accuracy for small objects, the scale-aware cross-entropy loss (SACELoss) is introduced to dynamically assign loss weights according to the actual sizes of objects. This enables smaller objects to be assigned larger weight values and thus receive more attention during training. Comprehensive experiments on both the iSAID- and DLRSD- datasets demonstrate that our method outperforms state-of-the-art FSS methods. Our code is available at https://github.com/TinyAway/PCFNet.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?