用图文对齐提升图像搜索准确率,减少误匹配。
Bringing Multimodality to Amazon Visual Search System
- 引入图文对齐损失约束图像匹配,避免依赖局部视觉特征。
- 3塔模型使点击率提升4.95%,4塔模型再增1.13%。
- 适合需要高精度图像搜索的电商系统场景。
图像到图像匹配在计算机视觉领域已得到广泛研究,以往工作主要聚焦于训练深度度量学习模型,以匹配查询图像与图库图像之间的视觉模式。然而,纯图像匹配易因局部视觉特征误匹配导致假阳性问题。为此,本文借鉴视觉-语言预训练最新进展,将额外的图像-文本对齐损失引入深度度量学习,作为图像-图像匹配损失的约束。通过文本(如商品标题)与图像对的对齐,模型能显式学习多模态概念,避免依赖低层视觉特征。我们逐步提出两种变体:三塔模型和四塔模型,后者引入一个简短文本查询输入。大量实验表明,该改进显著提升了图像匹配性能。进一步将其用于多模态搜索,支持图像与重述文本查询联合输入,有效提升搜索质量。离线与在线实验均显示主指标显著改善:三塔模型带来4.95%的相对点击率提升,四塔模型在此基础上再提升1.13%。
原文摘要 · Abstract (English)
Image to image matching has been well studied in the computer vision community. Previous studies mainly focus on training a deep metric learning model matching visual patterns between the query image and gallery images. In this study, we show that pure image-to-image matching suffers from false positives caused by matching to local visual patterns. To alleviate this issue, we propose to leverage recent advances in vision-language pretraining research. Specifically, we introduce additional image-text alignment losses into deep metric learning, which serve as constraints to the image-to-image matching loss. With additional alignments between the text (e.g., product title) and image pairs, the model can learn concepts from both modalities explicitly, which avoids matching low-level visual features. We progressively develop two variants, a 3-tower and a 4-tower model, where the latter takes one more short text query input. Through extensive experiments, we show that this change leads to a substantial improvement to the image to image matching problem. We further leveraged this model for multimodal search, which takes both image and reformulation text queries to improve search quality. Both offline and online experiments show strong improvements on the main metrics. Specifically, we see 4.95% relative improvement on image matching click through rate with the 3-tower model and 1.13% further improvement from the 4-tower model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。