arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.13796cs.CLcs.LG

TextGAIL: Generative Adversarial Imitation Learning for Text Generation

University of California, Davis(加利福尼亚大学戴维斯分校);ByteDance(字节跳动)

Qingyang Wu, Lei Li, Zhou Yu

英文摘要:

Generative Adversarial Networks (GANs) for text generation have recently received many criticisms, as they perform worse than their MLE counterparts. We suspect previous text GANs' inferior performance is due to the lack of a reliable guiding signal in their discriminators. To address this problem, we propose a generative adversarial imitation learning framework for text generation that uses large pre-trained language models to provide more reliable reward guidance. Our approach uses contrastive discriminator, and proximal policy optimization (PPO) to stabilize and improve text generation performance. For evaluation, we conduct experiments on a diverse set of unconditional and conditional text generation tasks. Experimental results show that TextGAIL achieves better performance in terms of both quality and diversity than the MLE baseline. We also validate our intuition that TextGAIL's discriminator demonstrates the capability of providing reasonable rewards with an additional task.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑