Deep Metric Learning by Online Soft Mining and Class-Aware Attention

Xinshao Wang, Yang Hua, Elyor Kodirov, Guosheng Hu, Neil Robertson

Research output: Chapter in Book/Report/Conference proceedingConference contribution

3 Citations (Scopus)
71 Downloads (Pure)

Abstract

Deep metric learning aims to learn a deep embedding that can capture the semantic similarity of data points. Given the availability of massive training samples, deep metric learning is known to suffer from slow convergence due to a large fraction of trivial samples. Therefore, most existing methods generally resort to sample mining strategies for selecting nontrivial samples to accelerate convergence and improve performance. In this work, we identify two critical limitations of the sample mining methods, and provide solutions for both of them. First, previous mining methods assign one binary score to each sample, i.e., dropping or keeping it, so they only selects a subset of relevant samples in a mini-batch. Therefore, we propose a novel sample mining method, called Online Soft Mining (OSM), which assigns one continuous score to each sample to make use of all samples in the mini-batch. OSM learns extended manifolds that preserve useful intraclass variances by focusing on more similar positives. Second, the existing methods are easily influenced by outliers as they are generally included in the mined subset. To address this, we introduce Class-Aware Attention (CAA) that assigns little attention to abnormal data samples. Furthermore, by combining OSM and CAA, we propose a novel weighted contrastive loss to learn discriminative embeddings. Extensive experiments on two fine-grained visual categorisation datasets and two video-based person re-identification benchmarks show that our method significantly outperforms the state-of-the-art.
Original languageEnglish
Title of host publicationAAAI-19: The Thirty-Third AAAI Conference on Artificial Intelligence: Proceedings
PublisherAssociation for the Advancement of Artificial Intelligence (AAAI)
Pages5361-5368
Number of pages8
DOIs
Publication statusPublished - 2019

Publication series

NameAAAI Conference on Artificial Intelligence
PublisherAAAI
ISSN (Print)2159-5399

Fingerprint Dive into the research topics of 'Deep Metric Learning by Online Soft Mining and Class-Aware Attention'. Together they form a unique fingerprint.

Cite this