Domain-aware Double Attention Network for Zero-Shot Sketch-Based Image Retrieval with Similarity Loss

Ming Zhu,Chen Zhao,Nian Wang,Feiyang Gu,Yu Liu,Xin Li
DOI: https://doi.org/10.1007/s00371-023-03012-8
IF: 2.835
2023-01-01
The Visual Computer
Abstract:Zero-shot sketch-based image retrieval (ZS-SBIR) has attracted increasing attention in computer vision, with the aim of searching the natural images that match a given sketch under the setting of zero-shot learning. In this paper, we propose a novel ZS-SBIR method based on the specifically developed domain-aware dual attention (DADA) module and similarity loss, which utilizes prior knowledge more fully than previous work. The DADA module emphasizes the importance of different channels and spaces according to the prior knowledge of whether the input is a sketch or a natural image. To preserve more prior knowledge from the models pre-trained on ImageNet, a similarity loss function is designed to supervise inter-class similarity. Furthermore, the center loss is introduced into our model, which can bridge the representation gap between the sketch domain and the natural image domain. We perform experiments on the TU-Berlin extended dataset and the Sketchy extended dataset, and the corresponding results verify the effectiveness of the proposed method.
What problem does this paper attempt to address?