Towards Improved Proxy-Based Deep Metric Learning via Data-Augmented Domain Adaptation

• Published in: Proceedings of the AAAI Conference on Artificial Intelligence
• Main Authors: Ren, Li, Chen, Chen, Wang, Liqiang, Hua, Kien
• Format: Article in Journal/Newspaper
• Language: English
• Published: Association for the Advancement of Artificial Intelligence 2024
• Subjects: ML: Representation Learning; CV: Image and Video Retrieval; DML
• Online Access: https://ojs.aaai.org/index.php/AAAI/article/view/29400; https://doi.org/10.1609/aaai.v38i13.29400

Deep Metric Learning (DML) plays an important role in modern computer vision research, where we learn a distance metric for a set of image representations. Recent DML techniques utilize the proxy to interact with the corresponding image samples in the embedding space. However, existing proxy-based DML methods focus on learning individual proxy-to-sample distance, while the overall distribution of samples and proxies lacks attention. In this paper, we present a novel proxy-based DML framework that focuses on aligning the sample and proxy distributions to improve the efficiency of proxy-based DML losses. Specifically, we propose the Data-Augmented Domain Adaptation (DADA) method to adapt the domain gap between the group of samples and proxies. To the best of our knowledge, we are the first to leverage domain adaptation to boost the performance of proxy-based DML. We show that our method can be easily plugged into existing proxy-based DML losses. Our experiments on benchmarks, including the popular CUB-200-2011, CARS196, Stanford Online Products, and In-Shop Clothes Retrieval, show that our learning algorithm significantly improves the existing proxy losses and achieves superior results compared to the existing methods. The code and Appendix are available at: https://github.com/Noahsark/DADA