A Resource-Efficient Training Framework for Remote Sensing Text--Image Retrieval

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Weihang, Li, Jihao, Li, Shuoke, Niu, Ziqing, Chen, Jialiang, Zhang, Wenkai
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917896704229376
author Zhang, Weihang
Li, Jihao
Li, Shuoke
Niu, Ziqing
Chen, Jialiang
Zhang, Wenkai
author_facet Zhang, Weihang
Li, Jihao
Li, Shuoke
Niu, Ziqing
Chen, Jialiang
Zhang, Wenkai
contents Remote sensing text--image retrieval (RSTIR) aims to retrieve the matched remote sensing (RS) images from the database according to the descriptive text. Recently, the rapid development of large visual-language pre-training models provides new insights for RSTIR. Nevertheless, as the complexity of models grows in RSTIR, the previous studies suffer from suboptimal resource efficiency during transfer learning. To address this issue, we propose a computation and memory-efficient retrieval (CMER) framework for RSTIR. To reduce the training memory consumption, we propose the Focus-Adapter module, which adopts a side branch structure. Its focus layer suppresses the interference of background pixels for small targets. Simultaneously, to enhance data efficacy, we regard the RS scene category as the metadata and design a concise augmentation technique. The scene label augmentation leverages the prior knowledge from land cover categories and shrinks the search space. We propose the negative sample recycling strategy to make the negative sample pool decoupled from the mini-batch size. It improves the generalization performance without introducing additional encoders. We have conducted quantitative and qualitative experiments on public datasets and expanded the benchmark with some advanced approaches, which demonstrates the competitiveness of the proposed CMER. Compared with the recent advanced methods, the overall retrieval performance of CMER is 2%--5% higher on RSITMD. Moreover, our proposed method reduces memory consumption by 49% and has a 1.4x data throughput during training. The code of the CMER and the dataset will be released at https://github.com/ZhangWeihang99/CMER.
format Preprint
id arxiv_https___arxiv_org_abs_2501_10638
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Resource-Efficient Training Framework for Remote Sensing Text--Image Retrieval
Zhang, Weihang
Li, Jihao
Li, Shuoke
Niu, Ziqing
Chen, Jialiang
Zhang, Wenkai
Computer Vision and Pattern Recognition
Information Retrieval
Remote sensing text--image retrieval (RSTIR) aims to retrieve the matched remote sensing (RS) images from the database according to the descriptive text. Recently, the rapid development of large visual-language pre-training models provides new insights for RSTIR. Nevertheless, as the complexity of models grows in RSTIR, the previous studies suffer from suboptimal resource efficiency during transfer learning. To address this issue, we propose a computation and memory-efficient retrieval (CMER) framework for RSTIR. To reduce the training memory consumption, we propose the Focus-Adapter module, which adopts a side branch structure. Its focus layer suppresses the interference of background pixels for small targets. Simultaneously, to enhance data efficacy, we regard the RS scene category as the metadata and design a concise augmentation technique. The scene label augmentation leverages the prior knowledge from land cover categories and shrinks the search space. We propose the negative sample recycling strategy to make the negative sample pool decoupled from the mini-batch size. It improves the generalization performance without introducing additional encoders. We have conducted quantitative and qualitative experiments on public datasets and expanded the benchmark with some advanced approaches, which demonstrates the competitiveness of the proposed CMER. Compared with the recent advanced methods, the overall retrieval performance of CMER is 2%--5% higher on RSITMD. Moreover, our proposed method reduces memory consumption by 49% and has a 1.4x data throughput during training. The code of the CMER and the dataset will be released at https://github.com/ZhangWeihang99/CMER.
title A Resource-Efficient Training Framework for Remote Sensing Text--Image Retrieval
topic Computer Vision and Pattern Recognition
Information Retrieval
url https://arxiv.org/abs/2501.10638