Stable controllable image generation model training method based on W distance

Abstract

The invention provides a stable controllable image generation model training method based on a W distance, and the method comprises the following steps: a, preprocessing image data to acquire sample data of a training set; b, constructing a stable controllable image generation model based on a W distance; c, constructing an overall model according to a loss function of each network; d, alternatelyiterating model training by a gradient descent algorithm to ensure normal convergence of model parameters; and e, after the model parameters converge, respectively separating a generator network G, an encoder network E and a classifier network C as independent service products. The invention is an end-to-end network model, improves a model in the prior art from two aspects of a model structure and a distribution measurement standard, solves the problems of unstable model training gradient and unstable model gradient descending direction, can stably and directionally generate an image sample.Robustness in the model training process and controllability of the generated sample are improved.

Description

technical field [0001] The invention relates to the technical field of image recognition, in particular to a stable and controllable image generation model training method based on W distance. Background technique [0002] In recent years, Generative Adversarial Networks (GAN) and its derived generative models have been one of the core topics in the machine learning and deep learning communities, among which BiGAN is the most important in high-dimensional complex data modeling. One of the deep generative models. The BiGAN model introduces the encoder network E into the GAN model so that the latent variables of similar samples can be gathered together during the encoding process, making the low-dimensional manifold continuous and achieving the effect of implicit regularization, which can improve the model's general ability. [0003] However, in the training process of the BiGAN model, it is very easy for the support set of the real data distribution and the generated sample...

AI Technical Summary

Problems solved by technology

[0028] The purpose of the present invention is to provide a stable and controllable image generation model training method based on W distance to solve the problems of unstable model training gradient and unstable model gradient descent direction in the prior art

Images