LocalGAN: Modeling Local Distributions for Adversarial Response Generation

Zhen Xu, Baoxun Wang, Huan Zhang, Kexin Qiu, Deyuan Zhang, Chengjie Sun

Keywords: neural response generation, adversarial learning, local distribution, energy-based distribution modeling

TL;DR: A study on leveraging the local distribution of query-response pairs to adversarial response generation.

Abstract: This paper presents a new methodology for modeling the local semantic distribution of responses to a given query in the human-conversation corpus, and on this basis, explores a specified adversarial learning mechanism for training Neural Response Generation (NRG) models to build conversational agents. The proposed mechanism aims to address the training instability problem and improve the quality of generated results of Generative Adversarial Nets (GAN) in their utilizations in the response generation scenario. Our investigation begins with the thorough discussions upon the objective function brought by general GAN architectures to NRG models, and the training instability problem is proved to be ascribed to the special local distributions of conversational corpora. Consequently, an energy function is employed to estimate the status of a local area restricted by the query and its responses in the semantic space, and the mathematical approximation of this energy-based distribution is finally found. Building on this foundation, a local distribution oriented objective is proposed and combined with the original objective, working as a hybrid loss for the adversarial training of response generation models, named as LocalGAN. Our experimental results demonstrate that the reasonable local distribution modeling of the query-response corpus is of great importance to adversarial NRG, and our proposed LocalGAN is promising for improving both the training stability and the quality of generated results.

Code: https://github.com/Kramgasse49/local_gan_4generation