Picture for Fangzheng Zhao

Fangzheng Zhao

Distribution Aligned Feature Clustering for Zero-Shot Sketch-Based Image Retrieval

Add code
Jan 17, 2023
Yuchen Wu, Kun Song, Fangzheng Zhao, Jiansheng Chen, Huimin Ma

Figure 1 for Distribution Aligned Feature Clustering for Zero-Shot Sketch-Based Image Retrieval
Figure 2 for Distribution Aligned Feature Clustering for Zero-Shot Sketch-Based Image Retrieval
Figure 3 for Distribution Aligned Feature Clustering for Zero-Shot Sketch-Based Image Retrieval
Figure 4 for Distribution Aligned Feature Clustering for Zero-Shot Sketch-Based Image Retrieval

Zero-Shot Sketch-Based Image Retrieval (ZS-SBIR) is a challenging cross-modal retrieval task. In prior arts, the retrieval is conducted by sorting the distance between the query sketch and each image in the gallery. However, the domain gap and the zero-shot setting make neural networks hard to generalize. This paper tackles the challenges from a new perspective: utilizing gallery image features. We propose a Cluster-then-Retrieve (ClusterRetri) method that performs clustering on the gallery images and uses the cluster centroids as proxies for retrieval. Furthermore, a distribution alignment loss is proposed to align the image and sketch features with a common Gaussian distribution, reducing the domain gap. Despite its simplicity, our proposed method outperforms the state-of-the-art methods by a large margin on popular datasets, e.g., up to 31% and 39% relative improvement of mAP@all on the Sketchy and TU-Berlin datasets.

Viaarxiv icon
Github IconAccess Paper or Ask Questions
Twitter IconFacebook IconLinkedin IconWhatsapp IconMessenger Icon