arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2102.05644cs.CV

Training Vision Transformers for Image Retrieval

Facebook AI(脸书人工智能研究院);ENS(巴黎高等师范学院);Inria(法国国家信息与自动化研究所)

Alaaeldin El-Nouby, Natalia Neverova, Ivan Laptev, Hervé Jégou

英文摘要:

Transformers have shown outstanding results for natural language understanding and, more recently, for image classification. We here extend this work and propose a transformer-based approach for image retrieval: we adopt vision transformers for generating image descriptors and train the resulting model with a metric learning objective, which combines a contrastive loss with a differential entropy regularizer. Our results show consistent and significant improvements of transformers over convolution-based approaches. In particular, our method outperforms the state of the art on several public benchmarks for category-level retrieval, namely Stanford Online Product, In-Shop and CUB-200. Furthermore, our experiments on ROxford and RParis also show that, in comparable settings, transformers are competitive for particular object retrieval, especially in the regime of short vector representations and low-resolution images.

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

↑