Controller-Guided Partial Label Consistency Regularization with Unlabeled Data

Qian-Wei Wang,Bowen Zhao,Mingyan Zhu,Tianxiang Li,Zimo Liu,Shu-Tao Xia
DOI: https://doi.org/10.1609/aaai.v38i14.29484
2024-01-01
Abstract:Partial label learning (PLL) learns from training examples each associatedwith multiple candidate labels, among which only one is valid. In recent years,benefiting from the strong capability of dealing with ambiguous supervision andthe impetus of modern data augmentation methods, consistencyregularization-based PLL methods have achieved a series of successes and becomemainstream. However, as the partial annotation becomes insufficient, theirperformances drop significantly. In this paper, we leverage easily accessibleunlabeled examples to facilitate the partial label consistency regularization.In addition to a partial supervised loss, our method performs acontroller-guided consistency regularization at both the label-level andrepresentation-level with the help of unlabeled data. To minimize thedisadvantages of insufficient capabilities of the initial supervised model, weuse the controller to estimate the confidence of each current prediction toguide the subsequent consistency regularization. Furthermore, we dynamicallyadjust the confidence thresholds so that the number of samples of each classparticipating in consistency regularization remains roughly equal to alleviatethe problem of class-imbalance. Experiments show that our method achievessatisfactory performances in more practical situations, and its modules can beapplied to existing PLL methods to enhance their capabilities.
What problem does this paper attempt to address?