Controllable Dialogue Generation with Disentangled Multi-grained Style Specification and Attribute Consistency Reward
arXiv:2109.06717
Abstract
Controllable text generation is an appealing but challenging task, which allows users to specify particular attributes of the generated outputs. In this paper, we propose a controllable dialogue generation model to steer response generation under multi-attribute constraints. Specifically, we define and categorize the commonly used control attributes into global and local ones, which possess different granularities of effects on response generation. Then, we significantly extend the conventional seq2seq framework by introducing a novel two-stage decoder, which first uses a multi-grained style specification layer to impose the stylistic constraints and determine word-level control states of responses based on the attributes, and then employs a response generation layer to generate final responses maintaining both semantic relevancy to the contexts and fidelity to the attributes. Furthermore, we train our model with an attribute consistency reward to promote response control with explicit supervision signals. Extensive experiments and in-depth analyses on two datasets indicate that our model can significantly outperform competitive baselines in terms of response quality, content diversity and controllability.
arXiv commentAccepted as a regular paper in IEEE/ACM TASLP
References in corpus (15)
- Adam: A Method for Stochastic Optimization
- Sequence to Sequence Learning with Neural Networks
- Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation
- A Neural Conversational Model
- CTRL: A Conditional Transformer Language Model for Controllable Generation
- Deep Reinforcement Learning for Dialogue Generation
- A Diversity-Promoting Objective Function for Neural Conversation Models
- Neural Responding Machine for Short-Text Conversation
- Personalizing Dialogue Agents: I have a dog, do you have pets too?
- Generating Informative and Diverse Conversational Responses via Adversarial Information Maximization
- Paraphrase Generation with Latent Bag of Words
- Towards Explainable and Controllable Open Domain Dialogue Generation with Dialogue Acts
- An Attentional Neural Conversation Model with Improved Specificity
- Learning to Select Knowledge for Response Generation in Dialog Systems
- Policy-Driven Neural Response Generation for Knowledge-Grounded Dialogue Systems