Dual-Training-Based Semi-Supervised Learning with Few Labels

The continual expansion in the number of images poses a great challenge for the annotation of the data. Therefore, improving the model performance for image classification with limited labeled data has become an important problem to solve. To address the problem, we propose in this paper a simple an...

全面介紹

書目詳細資料
發表在:Applied Sciences
Main Authors: Hao Wu, Jun Sun, Qidong Chen
格式: Article
語言:英语
出版: MDPI AG 2024-06-01
主題:
在線閱讀:https://www.mdpi.com/2076-3417/14/12/4993
實物特徵
總結:The continual expansion in the number of images poses a great challenge for the annotation of the data. Therefore, improving the model performance for image classification with limited labeled data has become an important problem to solve. To address the problem, we propose in this paper a simple and effective dual-training-based semi-supervised learning method for image classification. To enable the model to acquire more valuable information, we propose a dual training approach to enhance model training. Specifically, the model is trained with different augmented data at the same time with soft labels and hard labels, respectively. In addition, we propose a simple and effective weight generation method for generating the weight of samples during training to guide the model training. To further improve the model performance, we employ a projection layer at the end of the network to guide the self-learning of the model by minimizing the distance of features extracted from different layers. Finally, we evaluate the proposed approach on three benchmark image classification datasets. The experimental results demonstrate the effectiveness of our proposed approach.
ISSN:2076-3417