Computer Engineering and Applications ›› 2020, Vol. 56 ›› Issue (5): 43-48.DOI: 10.3778/j.issn.1002-8331.1908-0033

Previous Articles     Next Articles

Research on Convolutional Neural Network Algorithm Combined with Transfer Learning Model

QIU Ningjia, WANG Xiaoxia, WANG Peng, ZHOU Sicheng, WANG Yanchun   

  1. School of Computer Science and Technology, Changchun University of Science and Technology, Changchun 130022, China
  • Online:2020-03-01 Published:2020-03-06



  1. 长春理工大学 计算机科学技术学院,长春 130022


Aiming at the problem of low-precision and high-cost of the traditional convolutional neural network when the training set and test set have different distribution, convolutional neural network algorithm combined with transfer learning model is proposed. Firstly, using the principal component analysis algorithm to conduct unsupervised dimension reduction of the source data, meanwhile, the auto-encoder algorithm is combined to reduce the dimension of the target dataset in order to make the source domain and the target dataset have the similar distribution of features in the low dimension. Secondly, according to the characteristics of the feature extraction of convolutional neural network, JS divergence is used to determine whether convolution pool layer can be migrated and initialized hidden layer is used to complete the model of trCNN. Finally, a small number of the marked target datasets are used to train to finish the building of the classification model. The design experiment verifies that the classification model can accurately perform the classification work with a small amount of annotation data.

Key words: principal component analysis, auto-encoder, convolutional neural network, transfer learning



关键词: 主成分分析, 自编码机, 卷积神经网络, 迁移学习