让图文混合查询在图文交错文档中实现更精准检索
Recurrence-Enhanced Vision-and-Language Transformers for Robust Multimodal Document Retrieval
- 用多层视觉与文本特征,结合递归注意力机制融合跨模态信息
- 在M2KR和M-BEIR上达到当前最优效果,尤其在复杂查询下表现突出
- 适合需要高精度图文混合检索的科研与工业应用
跨模态检索因大规模训练、新型架构设计及在大语言模型中的应用而日益受到关注。本文提出ReT模型,支持由图像与文本组成的混合查询,在图文交错的多模态文档集合中进行检索。ReT从视觉与文本骨干网络的不同层次提取多级表示,并引入一种基于Transformer的递归单元,融合不同层级的文本与视觉特征,借鉴LSTM的门控机制,实现多层次跨模态理解。在M2KR和M-BEIR基准上的大量实验表明,ReT在多种设置下均达到领先性能。代码与训练模型已公开于https://github.com/aimagelab/ReT。
原文摘要 · Abstract (English)
Cross-modal retrieval is gaining increasing efficacy and interest from the research community, thanks to large-scale training, novel architectural and learning designs, and its application in LLMs and multimodal LLMs. In this paper, we move a step forward and design an approach that allows for multimodal queries, composed of both an image and a text, and can search within collections of multimodal documents, where images and text are interleaved. Our model, ReT, employs multi-level representations extracted from different layers of both visual and textual backbones, both at the query and document side. To allow for multi-level and cross-modal understanding and feature extraction, ReT employs a novel Transformer-based recurrent cell that integrates both textual and visual features at different layers, and leverages sigmoidal gates inspired by the classical design of LSTMs. Extensive experiments on M2KR and M-BEIR benchmarks show that ReT achieves state-of-the-art performance across diverse settings. Our source code and trained models are publicly available at https://github.com/aimagelab/ReT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。