Learning global image representation with generalized-mean pooling and smoothed average precision for large-scale CBIR

• Main Authors: Li, Y. (Author), Wang, C. (Author), Yang, B. (Author), Yao, J. (Author)
• Format: Article
• Language: English
• Published: John Wiley and Sons Inc 2023
• Subjects: computer vision; content-based retrieval; image retrieval
• Online Access: View Fulltext in Publisher
• ISBN: 17519659 (ISSN)
• ISSN: 17519659 (ISSN)
• DOI: 10.1049/ipr2.12825

Content-based image retrieval (CBIR) is the problem of searching for items in an image database that are similar to the query image. Most of the existing image retrieval methods are trained based on metric learning loss functions (e.g. contrastive loss or triplet loss), however, which require the use of hard sample mining strategies (HMS) to better train the model. The HMS implies that picking out hard positive or negative samples increases the complexity of model training and requires a large amount of additional training time. To address this issue, lessons from recent work are leveraged on representation learning and a model called GS is proposed that combines the state-of-the-art Generalized-Mean (GeM) pooling and the smoothed average precision (AP). The entire network can be learned end-to-end by approximating the non-differentiable AP function to a differentiable one-without mining hard samples, only image-level annotations. A model named GSA is also presented which achieves excellent retrieval performance jointly trained by two various loss functions. Experimental results validate the effectiveness of the proposed approach and demonstrate the competitive performance on a common standard image retrieval dataset (Revisited Oxford and Paris). © 2023 The Authors. IET Image Processing published by John Wiley & Sons Ltd on behalf of The Institution of Engineering and Technology.