Denoising Multi-Similarity Formulation: A Self-Paced Curriculum-Driven Approach for Robust Metric Learning

• Published in: Proceedings of the AAAI Conference on Artificial Intelligence
• Main Authors: Zhang, Chenkang, Luo, Lei, Gu, Bin
• Format: Article in Journal/Newspaper
• Language: English
• Published: Association for the Advancement of Artificial Intelligence 2023
• Subjects: ML: Representation Learning; ML: Deep Neural Network Algorithms; DML
• Online Access: https://ojs.aaai.org/index.php/AAAI/article/view/26324; https://doi.org/10.1609/aaai.v37i9.26324

Deep Metric Learning (DML) is a group of techniques that aim to measure the similarity between objects through the neural network. Although the number of DML methods has rapidly increased in recent years, most previous studies cannot effectively handle noisy data, which commonly exists in practical applications and often leads to serious performance deterioration. To overcome this limitation, in this paper, we build a connection between noisy samples and hard samples in the framework of self-paced learning, and propose a Balanced Self-Paced Metric Learning (BSPML) algorithm with a denoising multi-similarity formulation, where noisy samples are treated as extremely hard samples and adaptively excluded from the model training by sample weighting. Especially, due to the pairwise relationship and a new balance regularization term, the sub-problem w.r.t. sample weights is a nonconvex quadratic function. To efficiently solve this nonconvex quadratic problem, we propose a doubly stochastic projection coordinate gradient algorithm. Importantly, we theoretically prove the convergence not only for the doubly stochastic projection coordinate gradient algorithm, but also for our BSPML algorithm. Experimental results on several standard data sets demonstrate that our BSPML algorithm has better generalization ability and robustness than the state-of-the-art robust DML approaches.