Confidence-based Weighted Loss for Multi-label Classification with Missing Labels - Archive ouverte HAL
Communication Dans Un Congrès Année : 2020

Confidence-based Weighted Loss for Multi-label Classification with Missing Labels

Résumé

The problem of multi-label classification with missing labels (MLML) is a common challenge that is prevalent in several domains, e.g. image annotation and auto-tagging. In multi-label classification, each instance may belong to multiple class labels simultaneously. Due to the nature of the dataset collection and labelling procedure , it is common to have incomplete annotations in the dataset, i.e. not all samples are labelled with all the corresponding labels. However, the incomplete data labelling hinders the training of classification models. MLML has received much attention from the research community. However, in cases where a pre-trained model is fine-tuned on an MLML dataset, there has been no straightforward approach to tackle the missing labels, specifically when there is no information about which are the missing ones. In this paper, we propose a weighted loss function to account for the confidence in each label/sample pair that can easily be incorporated to fine-tune a pre-trained model on an incomplete dataset. Our experiment results show that using the proposed loss function improves the performance of the model as the ratio of missing labels increases.
Fichier principal
Vignette du fichier
ICMR_paper_v5.3.pdf (567.63 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)
Loading...

Dates et versions

hal-02547012 , version 1 (19-04-2020)

Identifiants

Citer

Karim M Ibrahim, Elena Epure, Geoffroy Peeters, Gael Richard. Confidence-based Weighted Loss for Multi-label Classification with Missing Labels. The 2020 International Conference on Multimedia Retrieval (ICMR '20), Jun 2020, Dublin, Ireland. ⟨10.1145/3372278.3390728⟩. ⟨hal-02547012⟩
397 Consultations
1520 Téléchargements

Altmetric

Partager

More