• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2022, Vol. 44 ›› Issue (01): 110-117.

Previous Articles     Next Articles

A semi supervised image annotation method based on LDA and convolutional neural network

WANG Bao-cheng1,LIU Li-jun1,HUANG Qing-song1,2   

  1. (1.Faculty of Information Engineering and Automation,Kunming University of Science and Technology,Kunming 650500;

    2.Yunnan Key Laboratory of Computer Technology Application,

    Kunming University of Science and Technology,Kunming 650500,China)

  • Received:2020-06-29 Revised:2020-09-20 Accepted:2022-01-25 Online:2022-01-25 Published:2022-01-13

Abstract: With the continuous emergence of intelligent devices, the number of pictures increases rapidly. However, many images are not fully utilized because they are not labeled. In order to solve this problem, a semi supervised image annotation method based on LDA and convolutional neural network is proposed. Firstly, all text information in the image training set is put into LDA to generate text tagging words. Secondly, the convolutional neural network is used to obtain the high-level visual features of the image, and the convolutional neural network is optimized by adding attention mechanism and modifying loss function. Thirdly, the label words generated by LDA are combined with the high-level visual features of the obtained image, and the semi supervised learning is used to complete the model training. Finally, the correlation between the tagging words and the prediction results using the final model are combined to complete the final tagging of the image. Comparative experiments on the IAPR TC-12 image data set show that the proposed labeling method is more accurate.

Key words: LDA, convolutional neural network, attention mechanism, semi supervised learning