Dialog State Tracking with Reinforced Data Augmentation

Abstract

Neural dialog state trackers are generally limited due to the lack of quantity and diversity of annotated training data. In this paper, we address this difficulty by proposing a reinforcement learning (RL) based framework for data augmentation that can generate high-quality data to improve the neural state tracker. Specifically, we introduce a novel contextual bandit generator to learn fine-grained augmentation policies that can generate new effective instances by choosing suitable replacements for the specific context. Moreover, by alternately learning between the generator and the state tracker, we can keep refining the generative policies to generate more high-quality training data for neural state tracker. Experimental results on the WoZ and MultiWoZ (restaurant) datasets demonstrate that the proposed framework significantly improves the performance over the state-of-the-art models, especially with limited training data.

arXiv commentAAAI 2020

References in corpus (8)

• A Structured Self-attentive Sentence Embedding
• A Deep Reinforced Model for Abstractive Summarization
• Sequence Level Training with Recurrent Neural Networks
• AutoAugment: Learning Augmentation Policies from Data
• Deep Reinforcement Learning for Dialogue Generation
• Efficient Optimal Learning for Contextual Bandits
• Toward Scalable Neural Dialogue State Tracking Model
• Improving Dialogue State Tracking by Discerning the Relevant Context

Cited by in corpus (2)

• Paraphrase Augmented Task-Oriented Dialog Generation
• CoSDA-ML: Multi-Lingual Code-Switching Data Augmentation for Zero-Shot Cross-Lingual NLP