Discourse-Aware Neural Rewards for Coherent Text Generation

Antoine Bosselut,Asli Celikyilmaz,Xiaodong He,Jianfeng Gao,Po-Sen Huang,Yejin Choi
DOI: https://doi.org/10.48550/arXiv.1805.03766
2018-05-10
Abstract:In this paper, we investigate the use of discourse-aware rewards with reinforcement learning to guide a model to generate long, coherent text. In particular, we propose to learn neural rewards to model cross-sentence ordering as a means to approximate desired discourse structure. Empirical results demonstrate that a generator trained with the learned reward produces more coherent and less repetitive text than models trained with cross-entropy or with reinforcement learning with commonly used scores as rewards.
Computation and Language
What problem does this paper attempt to address?