arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1906.07589cs.CV
收藏

平均精度的学习:使用列表损失训练图像检索

Learning with Average Precision: Training Image Retrieval with a Listwise Loss

NAVER LABS Europe(NAVER欧洲实验室)

Jerome Revaud, Jon Almazan, Rafael Sampaio de Rezende, Cesar Roberto de Souza

摘要:

图像检索可以用公式表示为一个排序问题,其目标是通过减少与查询的相似性来对数据库图像进行排序。最近的图像检索深度模型通过利用排名定制的损失函数,表现优于传统方法,但仍然存在重要的理论和实践问题。首先,它们不是直接优化全局排名,而是最小化基本损失的上限,这不一定会导致最佳平均精度(mAP)。其次,这些方法需要大量的工程努力才能发挥作用,例如特殊的预培训和硬负采矿。在本文中,我们建议,而不是直接优化的全球mAP利用最新进展列表损失公式。使用直方图分箱近似,可以区分AP,从而用于端到端学习。与现有的损失相比,该方法在每次迭代时同时考虑数千张图像,并消除了对特设技巧的需要。它还在许多标准检索基准上建立了一个新的技术状态。模型和评估脚本已在https://europe.naverlabs.com/Deep-Image-Retrieval/上提供

英文摘要:

Image retrieval can be formulated as a ranking problem where the goal is to order database images by decreasing similarity to the query. Recent deep models for image retrieval have outperformed traditional methods by leveraging ranking-tailored loss functions, but important theoretical and practical problems remain. First, rather than directly optimizing the global ranking, they minimize an upper-bound on the essential loss, which does not necessarily result in an optimal mean average precision (mAP). Second, these methods require significant engineering efforts to work well, e.g. special pre-training and hard-negative mining. In this paper we propose instead to directly optimize the global mAP by leveraging recent advances in listwise loss formulations. Using a histogram binning approximation, the AP can be differentiated and thus employed to end-to-end learning. Compared to existing losses, the proposed method considers thousands of images simultaneously at each iteration and eliminates the need for ad hoc tricks. It also establishes a new state of the art on many standard retrieval benchmarks. Models and evaluation scripts have been made available at https://europe.naverlabs.com/Deep-Image-Retrieval/

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑