Factuality Enhanced Language Models for Open-Ended Text Generation

By Nayeon Lee, Wei Ping, Peng Xu, M. Patwary, Mohammad Shoeybi, Bryan Catanzaro. 2022

310 citations recorded by the source corpus.

Pretrained language models (LMs) are susceptible to generate text with nonfactual information. In this work, we measure and improve the factual accuracy of large-scale LMs for open-ended text generation. We design the FactualityPrompts test set and metrics to measure the factuality of LM generations. Based on that, we study the factual accuracy of LMs with parameter sizes ranging from 126M to 530B. Interestingly, we find that larger LMs are more factual than smaller ones, although a previous study suggests that larger LMs can be less truthful in terms of misconceptions. In addition, popular sampling algorithms (e.g., top-p) in open-ended text generation can harm the factuality due to the ''uniform randomness'' introduced at every sampling step. We propose the factual-nucleus sampling algorithm that dynamically adapts the randomness to improve the factuality of generation while maintaining quality. Furthermore, we analyze the inefficiencies of the standard training method in learning correct associations between entities from factual text corpus (e.g., Wikipedia). We propose a factuality-enhanced training method that uses TopicPrefix for better awareness of facts and sentence completion as the training objective, which can vastly reduce the factual errors. We release our code and FactualityPrompts benchmark at: https://github.com/nayeon7lee/FactualityPrompt.

Subfields: Artificial Intelligence, Information Systems, Natural Language Processing, Generative Models

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jason Aaron Edward Weston
  2. Luke S. Zettlemoyer
  3. Mohit Bansal
  4. Jianfeng Gao
  5. Hannaneh Hajishirzi
  6. Yejin Choi
  7. Percy Liang
  8. Caiming Xiong

In Information Systems

  1. Hannaneh Hajishirzi
  2. Wen-tau Yih
  3. Sewon Min
  4. Luke S. Zettlemoyer
  5. Daniel Khashabi
  6. Yelong Shen
  7. Weizhu Chen
  8. Nan Duan

In Natural Language Processing

  1. Jason Aaron Edward Weston
  2. Luke S. Zettlemoyer
  3. Mohit Bansal
  4. Hannaneh Hajishirzi
  5. Yejin Choi
  6. Jianfeng Gao
  7. Caiming Xiong
  8. Kurt Shuster

In Generative Models

  1. Arthur Szlam
  2. Josh Tenenbaum
  3. Yilun Du
  4. Antonio Torralba
  5. Shuang Li
  6. Anima Anandkumar
  7. Wei Ping
  8. Jason Aaron Edward Weston

Citations in the corpus