Automatic Image Annotation based on Co-Training
Zhixin Li, Lan Lin, Canlong Zhang, Huifang Ma, Weizhong Zhao · 2019
To learn a well-performed image annotation model, a large number of labeled samples are usually required. Although the unlabeled samples are readily available and abundant, it's a difficult task for humans to annotate large amounts of images manually. In this paper, we propose a novel semi-supervised approach based on co-training algorithm for automatic image annotation, which can utilize the labeled data and unlabeled data for the system simultaneously. Firstly, two different classifiers, namely the CNN (convolutional neural network) and the LDA-SVM, are constructed by all the labeled data. These two classifiers are independently represented as different feature views. Then, the most confident data with relevant pseudo-labels are chosen and amalgamated with the whole labeled dataset. After that, the two classifiers are retrained with the new labeled dataset until a stop condition is reached. In each iteration process, the unlabeled samples are labeled by high confidence pseudo-labels that are estimated by an adaptive weighted fusion method. Finally, we conduct experiments on two datasets, namely, IAPR TC-2 and NUS-WIDE, and measure the performance of the model with standard criteria, including precision, recall, F-measure, N+ and mAP. The experimental results show that our approach has superior annotation performance and outperforms many state-of-the-art automatic image annotation approaches.