← 返回资源分享
Neural Text Generation with Unlikelihood Training
论文
论文
发布时间2019-08-12
发表ICLR 2020 1 · arXiv:1908.04319
作者:Emily Dinan,Jason Weston,Kyunghyun Cho,Ilia Kulikov,Stephen Roller,Sean Welleck
详细介绍
Neural text generation is a key tool in natural language applications, but it is well known there are major problems at its core. In particular, standard likelihood training and decoding leads to dull and repetitive outputs. While some post-hoc fixes have been proposed, in particular top-$k$ and nucleus sampling, they do not address the fact that the token-level probabilities predicted by the model are poor. In this paper we show that the likelihood objective itself is at fault, resulting in a model that assigns too much probability to sequences containing repeats and frequent words, unlike those from the human training distribution. We propose a new objective, unlikelihood training, which forces unlikely generations to be assigned lower probability by the model. We show that both token and sequence level unlikelihood training give less repetitive, less dull text while maintaining perplexity, giving superior generations using standard greedy or beam search. According to human evaluations, our approach with standard beam search also outperforms the currently popular decoding methods of nucleus sampling or beam blocking, thus providing a strong alternative to existing techniques.
代码仓库 (6)
fadedcosine/pos-guided-neural-text-generation官方PyTorch
facebookresearch/unlikelihood_training官方PyTorch
c00k1ez/plain-transformersPyTorch
google/t5patchesJAX
ljyflores/loss-libraryPyTorch
griff4692/calibrating-summariesJAX
