Dialogue generation method based on near-end strategy optimization and adversarial learning

Abstract

The invention relates to a dialogue generation method based on near-end strategy optimization and adversarial learning, and belongs to the field of computer natural language processing. The dialogue generation method comprises the steps of firstly pre-training a generation model and a discrimination model of an adversarial generative network; then, adopting a Monte Carlo sampling method to calculate awards corresponding to each word in the generated sentence, wherein the size of an award value represents the quality of word generation; secondly, taking the training process of the adversarial generative network as a reinforcement learning process, and training the adversarial generative network by using a near-end strategy optimization algorithm, so that awards obtained by the discrimination model can guide the generation of the generative model, and dialogues obtained by the generative model can guide the training of the discrimination model; and finally, training the generation modelby using a forced guidance method. According to the dialogue generation method, the training efficiency of the model is improved by controlling self-adaptive multiple iterations of the generation model, and the complexity of the sample is improved through the near-end strategy optimization algorithm, and the dialogue generation quality is further improved, and the dialogue closer to human beings can be generated.

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

[0003] When the confrontation generation network is used in the process of dialogue generation, the discriminant model can only evaluate the quality of the entire sentence, that is, to get the reward of the entire sentence. In order to better use the reward obtained by the discriminant model to train the generative model, it is necessary to get the reward of the middle dialogue , the method of Monte Carlo sampling is a commonly used method to ge

Images