Application Research for Fusion Model of Pseudolabel and Cross Network

Comput Intell Neurosci. 2022 May 19:2022:9986611. doi: 10.1155/2022/9986611. eCollection 2022.

Abstract

Datasets usually suffer from supervised information missing and weak generalization ability in deep convolution neural network. In this paper, pseudolabel (PL) of Weakly Supervised Learning (WSL) was used to address the problem of supervised information missing, while Cross Network (CN) of Multitask Learning (MTL) was used to solve the problem of weak generalization ability in deep convolution neural network. In PL, the data of supervised information missing was predicted; thus, PL of the corresponding data was generated. In CN, PL data and labeled data were taken as two tasks to train together. Firstly, the labeled data was divided into training dataset and testing dataset, respectively, and image preprocessing was carried out. Secondly, the network was initialized and trained, and the model with high accuracy and good generalization was selected as the optimal model. Then, the optimal model was used to predict the unlabeled data and generate PL. Finally, the steps above were repeated several times to find a better optimal model. In the experiments of the fusion model of PL and CN, Facial Beauty Prediction was regarded as main task and the others as auxiliary tasks. Experimental results show that the model was suitable for multitask training of different tasks in different or similar datasets, and the accuracy of the main task of Facial Beauty Prediction reaches 64.76%, higher than the highest accuracy by conventional methods.

MeSH terms

  • Generalization, Psychological*
  • Neural Networks, Computer*