CVJan 23, 2022

Basket-based Softmax

arXiv:2201.09308v13 citations
AI Analysis

This addresses the challenge of using softmax losses in real-world applications where clean, globally labeled datasets are unavailable, though it is incremental as it builds on existing softmax-based methods.

The paper tackles the problem of training softmax-based models on multiple datasets with noisy labels by proposing Basket-based Softmax (BBS), a mining-during-training strategy that dynamically adds negative classes from other datasets, achieving state-of-the-art performance on face recognition and re-identification tasks with both simulated and real-world datasets.

Softmax-based losses have achieved state-of-the-art performances on various tasks such as face recognition and re-identification. However, these methods highly relied on clean datasets with global labels, which limits their usage in many real-world applications. An important reason is that merging and organizing datasets from various temporal and spatial scenarios is usually not realistic, as noisy labels can be introduced and exponential-increasing resources are required. To address this issue, we propose a novel mining-during-training strategy called Basket-based Softmax (BBS) as well as its parallel version to effectively train models on multiple datasets in an end-to-end fashion. Specifically, for each training sample, we simultaneously adopt similarity scores as the clue to mining negative classes from other datasets, and dynamically add them to assist the learning of discriminative features. Experimentally, we demonstrate the efficiency and superiority of the BBS on the tasks of face recognition and re-identification, with both simulated and real-world datasets.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes