Improved Natural Language Generation via Loss Truncation
Paper · 2020 · Daniel Kang and Tatsunori B. Hashimoto
Summary
Conditional text generators are usually trained with log loss (KL divergence). This objective forces the model to cover every reference in the training set, so a small share of noisy or invalid targets can distort what it learns. Examples include titles containing facts absent from the source. Kang and Hashimoto argue that distinguishability, measured as total variation distance between model and human outputs, is a more robust goal. However, it is hard to optimize directly, and GAN-style attempts have lagged behind plain log-loss models. As a practical substitute, they propose loss truncation: during training, examples with the highest log loss are adaptively dropped, which yields an upper bound on distinguishability while keeping training as cheap as standard log loss. They also add a sequence-level rejection sampling step that keeps only high-probability outputs. On Gigaword summarization, the truncated models beat common decoding baselines on HUSE, a human-judged distinguishability metric. Their summaries also reach factual accuracy comparable to human references. The work suggests that robust training objectives can complement architectural fixes for hallucination.
Key points
- A manual audit of 300 Gigaword titles found that 35% contain facts not supported by the article. These titles average a log loss of 34.3, versus 20.5 for directly entailed titles (over 1.7× higher).
- Among examples with log loss above 40, more than 80% involve some hallucinated content. This shows that high-loss training data is dominated by unfaithful references.
- The authors benchmarked loss truncation, paired with either direct or rejection sampling, against several decoding baselines (beam search, unrestricted sampling from the full distribution, and sampling restricted by top-k or by top-p), using HUSE for distinguishability.
- With rejection sampling, the truncated model surpassed all baselines, including beam search, on human-rated factual accuracy and matched the gold references.
- ROUGE-L (summarization) and BLEU (E2E) were reported as supplementary automatic metrics. Manual inspection showed loss-truncated samples were both varied and accurate, whereas top-k and beam outputs were repetitive and full or top-p sampling introduced invented facts.
Institutions
📚 Shelves 4 Computer Science · 9,086 works Artificial Intelligence · 8,889 works Arts and Humanities · 9,296 works Language and Linguistics · 9,206 works
🧭 Topics 1 Text generation · 145 works
Cite
Cite as Daniel Kang and Tatsunori B. Hashimoto. 2020. Improved Natural Language Generation via Loss Truncation. Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics.
@inproceedings{kang2020improved,
title = {{Improved Natural Language Generation via Loss Truncation}},
author = {Daniel Kang and Tatsunori B. Hashimoto},
booktitle = {Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics},
year = {2020},
doi = {10.18653/v1/2020.acl-main.66},
url = {https://doi.org/10.18653/v1/2020.acl-main.66}
}References
- Anusha Balakrishnan, Jinfeng Rao, Kartikeya Upasani, Michael White, and Rajen Subba. 2019. Constrained decoding for neural nlg from compositional representations in task-oriented dialogue. arXiv preprint arXiv:1906.07220. our card
- Eyal Beigman and Beata Beigman Klebanov. 2009. Learning with annotation noise. In Proceedings of the Joint Conference of the 47th Annual Meeting of the ACL and the 4th International Joint Conference on Natural Language Processing of the AFNLP: Volume 1-Volume 1, pages 280–287. Association for Comput… find
- Andrew Brock, Jeff Donahue, and Karen Simonyan. 2018. Large scale gan training for high fidelity natural image synthesis. arXiv preprint arXiv:1809.11096. — [arXiv:1809.11096](https://arxiv.org/abs/1809.11096) find
- Massimo Caccia, Lucas Caccia, William Fedus, Hugo Larochelle, Joelle Pineau, and Laurent Charlin. 2018. Language gans falling short. arXiv preprint arXiv:1811.02549. — [arXiv:1811.02549](https://arxiv.org/abs/1811.02549) find
- Imre Csiszar and János Körner. 2011. Information theory: coding theorems for discrete memoryless systems. Cambridge University Press. find
- Bo Dai, Sanja Fidler, Raquel Urtasun, and Dahua Lin. 2017. Towards diverse and natural image descriptions via a conditional gan. In Proceedings of the IEEE International Conference on Computer Vision, pages 2970–2979. find
- Bhuwan Dhingra, Manaal Faruqui, Ankur Parikh, Ming-Wei Chang, Dipanjan Das, and William W Cohen. 2019. Handling divergent reference texts when evaluating table-to-text generation. arXiv preprint arXiv:1906.01081. our card
- Ilias Diakonikolas, Gautam Kamath, Daniel M Kane, Jerry Li, Jacob Steinhardt, and Alistair Stewart. 2018. Sever: A robust meta-algorithm for stochastic optimization. arXiv preprint arXiv:1803.02815. — [arXiv:1803.02815](https://arxiv.org/abs/1803.02815) find
- Michael Gamon, Anthony Aue, and Martine Smets. 2005. Sentence-level mt evaluation without reference translations: Beyond language modeling. In Proceedings of EAMT, pages 103–111. find
- Sebastian Gehrmann, Yuntian Deng, and Alexander Rush. 2018. Bottom-up abstractive summarization. In Proceedings of the 2018 Conference on Empirical Methods in Natural Language Processing, pages 4098–4109. our card
- Sebastian Gehrmann, Hendrik Strobelt, and Alexander M Rush. 2019. Gltr: Statistical detection and visualization of generated text. arXiv preprint arXiv:1906.04043. our card
- Ian Goodfellow, Jean Pouget-Abadie, Mehdi Mirza, Bing Xu, David Warde-Farley, Sherjil Ozair, Aaron Courville, and Yoshua Bengio. 2014. Generative adversarial nets. In Advances in neural information processing systems, pages 2672–2680. find
- Tatsunori B Hashimoto, Hugh Zhang, and Percy Liang. 2019. Unifying human and statistical evaluation for natural language generation. North American Chapter of the Association for Computational Linguistics. find
- Ari Holtzman, Jan Buys, Maxwell Forbes, Antoine Bosselut, David Golub, and Yejin Choi. 2018. Learning to write with cooperative discriminators. arXiv preprint arXiv:1805.06087. our card
- Ari Holtzman, Jan Buys, Maxwell Forbes, and Yejin Choi. 2019. The curious case of neural text degeneration. arXiv preprint arXiv:1904.09751. find
- Peter J Huber. 1992. Robust estimation of a location parameter. In Breakthroughs in statistics, pages 492– 518. Springer. find
- Nitish Shirish Keskar, Bryan McCann, Lav R Varshney, Caiming Xiong, and Richard Socher. 2019. Ctrl: A conditional transformer language model for controllable generation. arXiv preprint arXiv:1909.05858. our card
- DL Donoho. 1982. Breakdown properties of multivariate location estimators. The Annals of Statistics. find
- Beata Beigman Klebanov and Eyal Beigman. 2010. Some empirical evidence for annotation noise in a benchmarked dataset. In Human Language Technologies: The 2010 Annual Conference of the North American Chapter of the Association for Computational Linguistics, pages 438–446. Association for Computationa… find
- Ondřej Dušek, Jekaterina Novikova, and Verena Rieser. 2019. Evaluating the state-of-the-art of end-to-end natural language generation: The E2E NLG Challenge. arXiv preprint arXiv:1901.11528. find
- Guillaume Klein, Yoon Kim, Yuntian Deng, Jean Senellart, and Alexander M Rush. 2017. Opennmt: Open-source toolkit for neural machine translation. arXiv preprint arXiv:1701.02810. our card
- Angela Fan, Mike Lewis, and Yann Dauphin. 2018. Hierarchical neural story generation. ACL. find
- Wojciech Kryściński, Nitish Shirish Keskar, Bryan Mc-Cann, Caiming Xiong, and Richard Socher. 2019. Neural text summarization: A critical evaluation. arXiv preprint arXiv:1908.08960. our card
- David L Donoho, Richard C Liu, et al. 1988. The” automatic” robustness of minimum distance functionals. The Annals of Statistics, 16(2):552–586. find
- Martin A Fischler and Robert C Bolles. 1981. Random sample consensus: a paradigm for model fitting with applications to image analysis and automated cartography. Communications of the ACM, 24(6):381– 395. find
- Katherine Lee, Orhan Firat, Ashish Agarwal, Clara Fannjiang, and David Sussillo. 2018. Hallucinations in neural machine translation. Interpretability and Robustness in Audio, Speech, and Language Workshop. of the 2015 Conference on Empirical Methods in Natural Language Processing. find
- Jiwei Li, Will Monroe, Tianlin Shi, Sébastien Jean, Alan Ritter, and Dan Jurafsky. 2017. Adversarial learning for neural dialogue generation. arXiv preprint arXiv:1701.06547. our card
- Abigail See, Peter J Liu, and Christopher D Manning. 2017. Get to the point: Summarization with pointer-generator networks. arXiv preprint arXiv:1704.04368. find
- Chin-Yew Lin and Eduard Hovy. 2003. Automatic evaluation of summaries using n-gram co-occurrence statistics. In Proceedings of the 2003 Human Language Technology Conference of the North American Chapter of the Association for Computational Linguistics, pages 150–157. our card
- Shiqi Shen, Yong Cheng, Zhongjun He, Wei He, Hua Wu, Maosong Sun, and Yang Liu. 2015. Minimum risk training for neural machine translation. arXiv preprint arXiv:1512.02433. our card
- Tsung-Yi Lin, Michael Maire, Serge Belongie, James Hays, Pietro Perona, Deva Ramanan, Piotr Dollár, and C Lawrence Zitnick. 2014. Microsoft coco: Common objects in context. In European conference on computer vision, pages 740–755. Springer. our card
- Ilya Sutskever, Oriol Vinyals, and Quoc V Le. 2014. Sequence to sequence learning with neural networks. Advances in NIPS. find
- Lucas Theis, Aäron van den Oord, and Matthias Bethge. 2016. A note on the evaluation of generative models. ICLR. find
- Ramesh Nallapati, Bowen Zhou, Caglar Gulcehre, Bing Xiang, et al. 2016. Abstractive text summarization using sequence-to-sequence rnns and beyond. arXiv preprint arXiv:1602.06023. find
- Ran Tian, Shashi Narayan, Thibault Sellam, and Ankur P Parikh. 2019. Sticking to the facts: Confident decoding for faithful data-to-text generation. arXiv preprint arXiv:1910.08684. — [arXiv:1910.08684](https://arxiv.org/abs/1910.08684) find
- Jekaterina Novikova, Ondřej Dušek, Amanda Cercas Curry, and Verena Rieser. 2017. Why we need new evaluation metrics for nlg. arXiv preprint arXiv:1707.06875. our card
- Christoph Tillmann and Hermann Ney. 2003. Word reordering and a dynamic programming beam search algorithm for statistical machine translation. Computational linguistics, 29(1):97–133. find
- Franz Josef Och. 2003. Minimum error rate training in statistical machine translation. In Proceedings of the 41st Annual Meeting on Association for Computational Linguistics-Volume 1, pages 160–167. Association for Computational Linguistics. find
- John W Tukey. 1960. A survey of sampling from contaminated distributions. Contributions to probability and statistics, pages 448–485. find
- Kishore Papineni, Salim Roukos, Todd Ward, and Wei-Jing Zhu. 2002. Bleu: a method for automatic evaluation of machine translation. In Proceedings of the 40th annual meeting on association for computational linguistics, pages 311–318. Association for Computational Linguistics. our card
- Yau-Shian Wang and Hung-Yi Lee. 2018. Learning to encode text as human-readable summaries using generative adversarial networks. arXiv preprint arXiv:1810.02851. find
- Sean Welleck, Ilia Kulikov, Stephen Roller, Emily Dinan, Kyunghyun Cho, and Jason Weston. 2019. Neural text generation with unlikelihood training. arXiv preprint arXiv:1908.04319. our card
- Yevgeniy Puzikov and Iryna Gurevych. 2018. E2e nlg challenge: Neural models vs. templates. In Proceedings of the 11th International Conference on Natural Language Generation, pages 463–471. find
- Sam Wiseman, Stuart M Shieber, and Alexander M Rush. 2017. Challenges in data-to-document generation. arXiv preprint arXiv:1707.08052. our card
- Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, and Ilya Sutskever. 2019. Language models are unsupervised multitask learners. OpenAI Blog, 1(8). find
- Rowan Zellers, Ari Holtzman, Hannah Rashkin, Yonatan Bisk, Ali Farhadi, Franziska Roesner, and Yejin Choi. 2019. Defending against neural fake news. arXiv preprint arXiv:1905.12616. our card
- Sai Rajeswar, Sandeep Subramanian, Francis Dutil, Christopher Pal, and Aaron Courville. 2017. Adversarial generation of natural language. arXiv preprint arXiv:1705.10929. find
- Sharon Zhou, Mitchell Gordon, Ranjay Krishna, Austin Narcomey, Li F Fei-Fei, and Michael Bernstein. 2019. Hype: A benchmark for human eye perceptual evaluation of generative models. In Advances in Neural Information Processing Systems, pages 3444–3456. find
- Melissa Roemmele. 2016. Writing stories with help from recurrent neural networks. In Thirtieth AAAI Conference on Artificial Intelligence. find
- Alexander M Rush, SEAS Harvard, Sumit Chopra, and Jason Weston. 2017. A neural attention model for sentence summarization. In ACLWeb. Proceedings Context: Donna Shalala is sporting a mustache to promote public health. Title: Milk on Her Lip Shalala Raises Eyebrows Context: At least two people have t… find
- Context: Southwest China’s Sichuan province has decided to build an inter-city high-tech industrial belt to serve development of Western China. Title: Sichuan to Build High-Tech Industrial Belt (a) Example 1. Context: British investment fund Fidelity has increased its stake in Puma, the German maker… find
Updated
This page in Markdown → https://md.research.elamur.ai/free/en/papers/kang-2020-improved-natural-language-generation-loss/