標題: Learning Discriminative CNN Features and Similarity Metrics for Image Retrieval
作者: Sun Pei-Xia
Lin Hui-Ting
Luo Tao
資訊工程學系
Department of Computer Science
關鍵字: image retrieval;image representation;similarity learning
公開日期: 2016
摘要: This paper proposes a CNN-based retrieval framework that uses Siamese network to learn a CNN model for image feature extraction. Model training and testing stages often use the same similarity metric. But this paper adopts a contrastive loss function with different distance metrics to fine-tune a pre-trained CNN model, and applies different distance metrics in testing stage. Through experimenting with different similarity metrics, this paper finally finds that using the L2 distance to specify the contrastive loss function while applying cosine similarity during testing achieves the best performance. Its mean average precision (mAP) achieves 58.2%, and is 1-2% higher than the previous best method [4]. It shows that the similarity metrics for training and testing need not be same. Subsequently, we learn more generalized similarity metrics for model training and testing independently by minimizing a hinge loss function defined over a pair of global image representations. Compared with the existing image retrieval methods that involve hand-crafted features, our proposed framework performs well on four typical retrieval datasets.
URI: http://hdl.handle.net/11536/136371
ISBN: 978-1-5090-2708-8
期刊: 2016 IEEE INTERNATIONAL CONFERENCE ON SIGNAL PROCESSING, COMMUNICATIONS AND COMPUTING (ICSPCC)
顯示於類別:會議論文