Confidence-based Weighted Loss for Multi-label Classification with Missing Labels - Archive ouverte HAL Access content directly
Conference Papers Year : 2020

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

Abstract

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
Origin Files produced by the author(s)
Loading...

Dates and versions

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

Identifiers

Cite

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⟩
379 View
1462 Download

Altmetric

Share

Gmail Mastodon Facebook X LinkedIn More