Gemini: A Dual-Task Co-training Model for Partial Label Learning.

Beibei Li,Senlin Shu,Beihong Jin,Tao Xiang,Yiyuan Zheng
DOI: https://doi.org/10.1007/978-981-99-8388-9_27
2024-01-01
Abstract:Partial-Label Learning (PLL) is an important weakly supervised learning task that assumes each training instance is annotated with a set of candidate labels. In recent years, self-training PLL models, which learn label confidence vectors and train the models with them iteratively, have attracted increasing attention and achieve state-of-the-art performance. However, they suffer from error accumulation problem due to that mistakenly disambiguated instances could mislead the model with false positive labels and causes performance degradation. To address this issue, we propose a novel dual-task co-training PLL model named Gemini that trains two structurally identical networks by optimizing two distinct tasks. Specifically, one is trained by optimizing PLL task with the original partially labeled data, and the other is trained by optimizing the noisy label learning (NLL) task with noisy pseudo labels generated according to the disambiguated label confidence. Moreover, we interrupt the error accumulation via a carefully-designed cooperation mechanism between the two networks, which comprises prediction alignment and correctly-disambiguated instance selection. Extensive experiments on benchmark datasets demonstrate the effectiveness and robustness of Gemini on both uniform and instance-dependent partially labeled data. The code is available at https://github.com/libeibeics/Gemini .
What problem does this paper attempt to address?