Training method and device of neural network model for sample classification

Abstract

The embodiment of the invention provides a training method and device of a neural network model for sample classification. The method comprises the steps of obtaining training samples in a training sample set, wherein the training samples have sample identifiers and sample category labels; inputting the training sample into a feature extraction model to obtain a feature representation vector; inputting the feature representation vector into a discriminator model to obtain an identification mark; determining a first prediction loss according to the identification identifier and the sample identifier, and performing first training on the discriminator model and the feature extraction model by taking minimization of the first prediction loss as a target; inputting the feature representation vector into a classifier model to obtain an identification category; and determining a second prediction loss according to the identification category and the sample category label, determining a thirdprediction loss according to the negative correlation with the first prediction loss and the positive correlation with the second prediction loss, and carrying out second training on the classifier model and the feature extraction model by taking minimization of the third prediction loss as a target. And the generalization of the model can be improved.

Description

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine

Login to view more

AI Technical Summary

Problems solved by technology

However, due to too many parameters, the complexity of the neural network model is too high. In most cases, the neur

Images