Partial Label Learning Based on Fully Connected Deep Neural Network

Authors: Houjie Li, Lei Wu, Jianjun He, Ruirui Zheng, Yu Zhou, Shuang Qiao

Abstract: The ambiguity of training samples in the partial label learning framework makes it difficult for us to develop learning algorithms and most of the existing algorithms are proposed based on the traditional shallow machine learn- ing models, such as decision tree, support vector machine, and Gaussian process model. Deep neu- ral networks have demonstrated excellent perfor- mance in many application fields, but currently it is rarely used for partial label learning frame- work. This study proposes a new partial label learning algorithm based on a fully connected deep neural network, in which the relationship between the candidate labels and the ground- truth label of each training sample is established by defining three new loss functions, and a regu- larization term is added to prevent overfitting. The experimental results on the controlled U- CI datasets and real-world partial label datasets reveal that the proposed algorithm can achieve higher classification accuracy than the state-of- the-art partial label learning algorithms.

Pages: 287-297

DOI: 10.46300/9106.2022.16.35

International Journal of Circuits, Systems and Signal Processing, E-ISSN: 1998-4464, Volume 16, 2022, Art. #35