Service interruption on Monday 11 July from 12:30 to 13:00: all the sites of the CCSD (HAL, EpiSciences, SciencesConf, AureHAL) will be inaccessible (network hardware connection).
Skip to Main content Skip to Navigation
Conference papers

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.
Document type :
Conference papers
Complete list of metadata

Cited literature [42 references]  Display  Hide  Download
Contributor : Karim M. Ibrahim Connect in order to contact the contributor
Submitted on : Sunday, April 19, 2020 - 2:33:45 PM
Last modification on : Friday, April 29, 2022 - 10:12:50 AM


Files produced by the author(s)




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⟩



Record views


Files downloads