Affiliation:
1. School of Internet of Things Engineering, Wuxi University, Wuxi 214105, China
2. School of Artificial Intelligence and Computer Science, Jiangnan University, Wuxi 214122, China
Abstract
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.
Funder
Wuxi University Research Start-up Fund for Introduced Talents
Reference42 articles.
1. Uncertainty-driven ensembles of multi-scale deep architectures for image classification;Arco;Inf. Fusion,2023
2. Spectral–spatial feature tokenization transformer for hyperspectral image classification;Sun;IEEE Trans. Geosci. Remote Sens.,2022
3. PV-RCNN++: Point-voxel feature set abstraction with local vector representation for 3D object detection;Shi;Int. J. Comput. Vis.,2023
4. Anchor-free oriented proposal generator for object detection;Cheng;IEEE Trans. Geosci. Remote Sens.,2022
5. Lin, A., Chen, B., Xu, J., Zhang, Z., Lu, G., and Zhang, D. (2022). Ds-transunet: Dual swin transformer u-net for medical image segmentation. IEEE Trans. Instrum. Meas., 71.