提出轻量框架CMER,提升遥感图文检索效率与性能
A Resource-Efficient Training Framework for Remote Sensing Text--Image Retrieval
- 用侧支结构的Focus-Adapter抑制背景干扰,降低内存消耗
- 通过场景标签增强缩小搜索空间,负样本回收提升泛化能力
- 在RSITMD上性能领先2%-5%,内存降49%,训练吞吐提高1.4倍
遥感图文检索(RSTIR)旨在根据描述文本从数据库中检索匹配的遥感图像。随着大规模视觉语言预训练模型的发展,RSTIR迎来新机遇,但现有方法在迁移学习中资源效率仍不理想。为此,本文提出计算与内存高效检索框架(CMER)。为减少训练内存开销,设计了焦点适配器(Focus-Adapter)模块,采用侧支结构,其焦点层抑制小目标的背景干扰;为提升数据利用效率,将遥感场景类别作为元数据,设计简洁增强策略,利用土地覆盖先验知识缩小搜索空间。提出负样本回收策略,使负样本池解耦于小批量大小,无需额外编码器即可提升泛化性能。在公开数据集上进行了定量与定性实验,并扩展基准测试包含多种先进方法,结果表明所提CMER具有竞争力。相比最新方法,其在RSITMD上的整体检索性能提升2%–5%。同时,内存消耗降低49%,训练时数据吞吐率达1.4倍。代码与数据集将开源至https://github.com/ZhangWeihang99/CMER。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。