Multilevel Similarity-Aware Deep Metric Learning for Fine-Grained Image Retrieval

Congcong Duan,Yong Feng,Mingliang Zhou,Xiancai Xiong,Yongheng Wang,Baohua Qiang,Weijia Jia
DOI: https://doi.org/10.1109/TII.2022.3227721
IF: 12.3
2023-01-01
IEEE Transactions on Industrial Informatics
Abstract:Fast and accurate image retrieval is an important and challenging task in massive image data scenarios. As the core technology of image retrieval tasks, deep metric learning aims at learning effective embedding representations that possess two properties among data points: positive concentrated and negative separated. In this work, we propose a multilevel similarity-aware method based on deep local descriptors for deep metric learning. We take the rich interclass similarity relationship based on the deep local invariant descriptors from the data into account to optimize sampling strategies for mining informative samples. The method dynamically adjusts the margin between data points to better match the true similarity relationship between classes. Specifically, for images in a batch, we first obtain deep local descriptors and calculate the similarity matrix of the channel, pixel, and spatial levels. Then, depending on the calculated comprehensive similarity matrix, we propose a multilevel similarity-aware loss function through the deviation between pairwise distance and violate margin to make full use of informative samples. The experimental results demonstrate that our proposed method outperforms other state-of-the-art methods in terms of fine-grained image retrieval and clustering tasks.
What problem does this paper attempt to address?