用细粒度描述训练模型,让图文匹配更精准
FLAIR: VLM with Fine-grained Language-informed Image Representations
- 用多样化子描述学习局部图像表征,实现文本相关的图像嵌入
- 在3000万图文对上训练,零样本语义分割优于千亿级数据模型
- 适合需要精准图文检索、局部内容识别的研究与应用
CLIP在大规模图文对上实现了出色的图像与文本对齐,但其全局匹配方式限制了对细节视觉特征的捕捉。为此,我们提出FLAIR(细粒度语言引导图像表征),通过采样多样化的细粒度图像描述,训练视觉语言模型生成不仅包含全局嵌入,还包含文本相关的局部图像表示。模型在局部图像标记上引入文本条件注意力池化,以生成精细化图像表征,显著提升局部内容检索能力。我们在现有多模态检索基准及新提出的细粒度检索任务上均取得当前最优性能。实验表明,仅在3000万图像-文本对上训练的FLAIR,在零样本语义分割等任务中表现超越在数十亿对数据上训练的模型。代码已开源。
原文摘要 · Abstract (English)
CLIP has shown impressive results in aligning images and texts at scale. However, its ability to capture detailed visual features remains limited because CLIP matches images and texts at a global level. To address this issue, we propose FLAIR, Fine-grained Language-informed Image Representations, an approach that utilizes long and detailed image descriptions to learn localized image embeddings. By sampling diverse sub-captions that describe fine-grained details about an image, we train our vision-language model to produce not only global embeddings but also text-specific image representations. Our model introduces text-conditioned attention pooling on top of local image tokens to produce fine-grained image representations that excel at retrieving detailed image content. We achieve state-of-the-art performance on both, existing multimodal retrieval benchmarks, as well as, our newly introduced fine-grained retrieval task which evaluates vision-language models' ability to retrieve partial image content. Furthermore, our experiments demonstrate the effectiveness of FLAIR trained on 30M image-text pairs in capturing fine-grained visual information, including zero-shot semantic segmentation, outperforming models trained on billions of pairs. Code is available at https://github.com/ExplainableML/flair .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。