Dynamic manifold-based sample selection in contrastive learning for remote sensing image retrieval

Qiyang Liu,Yun Ge,Sijia Wang,Ting Wang,Jinlong Xu
DOI: https://doi.org/10.1007/s00371-024-03649-z
IF: 2.835
2024-09-28
The Visual Computer
Abstract:In the field of remote sensing, traditional supervised learning is constrained by challenges such as high annotation costs and inconsistent labels, primarily due to the rich content and vast quantity of remote sensing images. Self-supervised contrastive learning is an effective solution because of its ability to learn supervised signals from unlabeled data. However, contrastive learning considers two views of the same image as positive samples and the rest of the image as negative samples. This way of definition pushes images of the same category further away in the feature space, which is not favorable for image retrieval tasks. Consequently, we propose the dynamic manifold-based sample selection (DMBSS) method for improving remote sensing image retrieval. The method employs manifold similarity to quantify the similarity between sample points and selects samples with high similarity as potential positive samples. Furthermore, to enhance the reliability of potential positive samples, we dynamically adjust the neighbor rate during training process, thereby selecting varying numbers of potential positive samples at distinct training epochs. The method fully explores the nonlinear relationships between features to redefine positive and negative samples. The experimental results show that the DMBSS method outperforms other existing methods significantly on three common benchmark datasets, demonstrating excellent performance in unsupervised remote sensing image retrieval. Code available at https://github.com/QiyangLiu00/DMBSS
computer science, software engineering
What problem does this paper attempt to address?