Check Your Facts and Try Again: Improving Large Language Models with External Knowledge and Automated Feedback
Written for the Research library (claude-opus-5-5) from the paper text. Original — arxiv.org/pdf/2302.12813
Summary
Large language models such as ChatGPT write fluent text but often invent facts. They also cannot consult fresh or private information. Earlier knowledge-augmentation methods mostly require fine-tuning, which is costly or impossible for very large or closed models.
The authors propose LLM-Augmenter, a set of plug-and-play modules placed around a frozen, black-box LLM and framed as a Markov Decision Process:
- a Working Memory stores the dialog state;
- a Policy chooses the next action;
- a Knowledge Consolidator retrieves evidence from the Web or task databases, links entities and builds evidence chains;
- a Prompt Engine builds prompts that contain this evidence;
- a Utility module scores candidate answers and writes textual feedback.
The model is re-prompted with that feedback until a response passes verification.
The system was tested on information-seeking dialog (including customer service) and on open-domain Wiki QA that needs multi-hop reasoning. It cuts ChatGPT's hallucinations while keeping answers fluent and informative. The work shows that grounding and automated self-correction can be added without changing model weights. This matters for reliable assistants built on domain-specific or local data.
Key points
- Architecture: Working Memory, Policy, Action Executor (Knowledge Consolidator + Prompt Engine) and Utility. Evidence is retrieved through Bing Search APIs and REST APIs, and entities are linked to Wikipedia descriptions, following Ma et al. (2022).
- Policy: it can be hand-written rules or a trainable T5-based network optimized with REINFORCE. Training proceeds in three stages: rule bootstrapping, LLM-based user simulators, then real users. The main ChatGPT results use a manual policy; RL experiments used T5-Base because ChatGPT access was limited.
- Utility functions: these are either model-based (preference scores for fluency, factuality, etc.) or rule-based. Experiments use KF1 (knowledge F1) as the groundedness measure.
- Results: in customer-service dialog, the reported human evaluation shows +32.3% Usefulness and +12.9% Humanness. On Wiki QA, F1 rises by about 10 points absolute over closed-book ChatGPT. It still trails the fine-tuned state of the art (Ma et al., 2022) that uses top-50 evidence. The limitations section states the current version lacks human evaluation, which conflicts with the reported human-evaluation figures.
- Limitations: ChatGPT is often queried twice per answer, which adds latency. Code and models are available at aka.ms/llm-augmenter.
Cited by
- The Internal State of an LLM Knows When It's Lying
- Benchmarking Large Language Models in Retrieval-Augmented Generation
- Large Legal Fictions: Profiling Legal Hallucinations in Large Language Models
- Chain-of-Verification Reduces Hallucination in Large Language Models
- Gemini: A Family of Highly Capable Multimodal Models
- CRITIC: Large Language Models Can Self-Correct with Tool-Interactive Critiquing
- CRUXEval: A Benchmark for Code Reasoning, Understanding and Execution
- LegalBench: A Collaboratively Built Benchmark for Measuring Legal Reasoning in Large Language Models
- TrustLLM: Trustworthiness in Large Language Models
- LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Code
- Chameleon: Plug-and-Play Compositional Reasoning with Large Language Models
- MathVista: Evaluating Mathematical Reasoning of Foundation Models in Visual Contexts
- LINC: A Neurosymbolic Approach for Logical Reasoning by Combining Language Models with First-Order Logic Provers
- Instruction Tuning with GPT-4
- A Survey of Hallucination in Large Foundation Models
- Fine-tuning Language Models for Factuality
- A Comprehensive Survey of Hallucination Mitigation Techniques in Large Language Models
- MINT: Evaluating LLMs in Multi-turn Interaction with Tools and Language Feedback
- Survey on Factuality in Large Language Models: Knowledge, Retrieval and Domain-Specificity
- The Rise and Potential of Large Language Model Based Agents: A Survey
- Hallucination is Inevitable: An Innate Limitation of Large Language Models
- The Dawn of LMMs: Preliminary Explorations with GPT-4V(ision)
- Large Language Model as Attributed Training Data Generator: A Tale of Diversity and Bias
- How Language Model Hallucinations Can Snowball
- Siren's Song in the AI Ocean: A Survey on Hallucination in Large Language Models
- A Survey of Large Language Models
- ToolQA: A Dataset for LLM Question Answering with External Tools
Cite
@misc{peng2023check,
title = {{Check Your Facts and Try Again: Improving Large Language Models with External Knowledge and Automated Feedback}},
author = {Peng, Baolin and Galley, Michel and He, Pengcheng and Cheng, Hao and Xie, Yujia and Hu, Yu and Huang, Qiuyuan and Liden, Lars and Yu, Zhou and Chen, Weizhu and Gao, Jianfeng},
year = {2023},
doi = {10.48550/arxiv.2302.12813},
eprint = {2302.12813},
archivePrefix = {arXiv},
url = {https://doi.org/10.48550/arxiv.2302.12813}
}References
- Satanjeev Banerjee and Alon Lavie. 2005. ME-TEOR: An automatic metric for mt evaluation with improved correlation with human judgments. In ACL Workshop on Intrinsic and Extrinsic Evaluation Measures for Machine Translation and/or Summarization, pages 65–72. find
- Emily M. Bender, Timnit Gebru, Angelina McMillan-Major, and Shmargaret Shmitchell. 2021. On the dangers of stochastic parrots: Can language models be too big? Proceedings of the 2021 ACM Conference on Fairness, Accountability, and Transparency. find
- Rishi Bommasani, Drew A. Hudson, Ehsan Adeli, Russ Altman, Simran Arora, Sydney von Arx, Michael S. Bernstein, Jeannette Bohg, Antoine Bosselut, Emma Brunskill, Erik Brynjolfsson, S. Buch, Dallas Card, Rodrigo Castellon, Niladri S. Chatterji, Annie S. Chen, Kathleen A. Creel, Jared Davis, Dora Demsz… find our card
- Tom Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah, Jared D Kaplan, Prafulla Dhariwal, Arvind Neelakantan, Pranav Shyam, Girish Sastry, Amanda Askell, Sandhini Agarwal, Ariel Herbert-Voss, Gretchen Krueger, Tom Henighan, Rewon Child, Aditya Ramesh, Daniel Ziegler, Jeffrey Wu, Clemens Winter, Chri… find
- Aakanksha Chowdhery, Sharan Narang, Jacob Devlin, Maarten Bosma, Gaurav Mishra, Adam Roberts, Paul Barham, Hyung Won Chung, Charles Sutton, Sebastian Gehrmann, et al. 2022. Palm: Scaling language modeling with pathways. arXiv preprint arXiv:2204.02311. find
- Emily Dinan, Stephen Roller, Kurt Shuster, Angela Fan, Michael Auli, and Jason Weston. 2018. Wizard of Wikipedia: Knowledge-powered conversational agents. In International Conference on Learning Representations. find
- Mihail Eric, Rahul Goel, Shachi Paul, Adarsh Kumar, Abhishek Sethi, Peter Ku, Anuj Kumar Goyal, Sanchit Agarwal, Shuyang Gao, and Dilek Hakkani-Tur. 2019. Multiwoz 2.1: A consolidated multi-domain dialogue dataset with state corrections and state tracking baselines. arXiv preprint arXiv:1907.01669. find
- Michel Galley, Chris Brockett, Xiang Gao, Jianfeng Gao, and Bill Dolan. 2019. Grounded response generation task at DSTC7. In AAAI Dialog System Technology Challenges Workshop. find
- Jianfeng Gao, Michel Galley, and Lihong Li. 2019. Neural approaches to conversational AI. Foundations and Trends in Information Retrieval, 13(2-3):127–298. find
- Jianfeng Gao, Chenyan Xiong, Paul Bennett, and Nick Craswell. 2022. Neural approaches to conversational information retrieval. arXiv preprint arXiv:2201.05176. — [arXiv:2201.05176](https://arxiv.org/abs/2201.05176) find
- Marjan Ghazvininejad, Chris Brockett, Ming-Wei Chang, Bill Dolan, Jianfeng Gao, Wen-tau Yih, and Michel Galley. 2017. A knowledge-grounded neural conversation model. CoRR, abs/1702.01932. — [arXiv:1702.01932](https://arxiv.org/abs/1702.01932) find
- Amelia Glaese, Nathan McAleese, Maja Trkebacz, John Aslanides, Vlad Firoiu, Timo Ewalds, Maribeth Rauh, Laura Weidinger, Martin Chadwick, Phoebe Thacker, Lucy Campbell-Gillingham, Jonathan Uesato, Po-Sen Huang, Ramona Comanescu, Fan Yang, A. See, Sumanth Dathathri, Rory Greig, Charlie Chen, Doug Fri… find our card
- Kelvin Guu, Kenton Lee, Zora Tung, Panupong Pasupat, and Ming-Wei Chang. 2020. REALM: Retrieval-augmented language model pre-training. ArXiv, abs/2002.08909. find
- Hangfeng He, Hongming Zhang, and Dan Roth. 2022. Rethinking with retrieval: Faithful large language model inference. arXiv preprint arXiv:2301.00303. find
- Vladimir Karpukhin, Barlas Oguz, Sewon Min, Patrick Lewis, Ledell Wu, Sergey Edunov, Danqi Chen, and Wen-tau Yih. 2020. Dense passage retrieval for open-domain question answering. In Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP), pages 6769– 6781, Onl… find our card
- Seokhwan Kim, Mihail Eric, Karthik Gopalakrishnan, Behnam Hedayatnia, Yang Liu, and Dilek Hakkani-Tur. 2020. Beyond domain APIs: Task-oriented conversational modeling with unstructured knowledge access. In Proceedings of the 21th Annual Meeting of the Special Interest Group on Discourse and Dialogue… find
- Seokhwan Kim, Spandana Gella, Di Jin, Alexandros Papangelis, Behnam Hedayatnia, Yang Liu, and Dilek Hakkani-Tür. 2023. DSTC11 track proposal: Task-oriented conversational modeling with subjective knowledge. https://github.com/ alexa/dstc11-track5. ↗
- Angeliki Lazaridou, Elena Gribovskaya, Wojciech Stokowiec, and Nikolai Grigorev. 2022. Internet-augmented language models through few-shot prompting for open-domain question answering. arXiv preprint arXiv:2203.05115. find our card
- Patrick Lewis, Ethan Perez, Aleksandara Piktus, Fabio Petroni, Vladimir Karpukhin, Naman Goyal, Heinrich Kuttler, Mike Lewis, Wen tau Yih, Tim Rocktäschel, Sebastian Riedel, and Douwe Kiela. 2020. Retrieval-augmented generation for knowledge-intensive nlp tasks. ArXiv, abs/2005.11401. find
- Rongzhong Lian, Min Xie, Fan Wang, Jinhua Peng, and Hua Wu. 2019. Learning to select knowledge for response generation in dialog systems. In International Joint Conference on Artificial Intelligence. find
- Chin-Yew Lin. 2004. ROUGE: A package for automatic evaluation of summaries. In ACL workshop, pages 74–81. find
- Kaixin Ma, Hao Cheng, Xiaodong Liu, Eric Nyberg, and Jianfeng Gao. 2022. Open-domain question answering via chain of reasoning over heterogeneous knowledge. arXiv preprint arXiv:2210.12338. find
- Aman Madaan, Niket Tandon, Peter Clark, and Yiming Yang. 2022. Memory-assisted prompt editing to improve GPT-3 after deployment. arXiv preprint arXiv:2201.06009. find
- Reiichiro Nakano, Jacob Hilton, Suchir Balaji, Jeff Wu, Long Ouyang, Christina Kim, Christopher Hesse, Shantanu Jain, Vineet Kosaraju, William Saunders, et al. 2021. WebGPT: Browser-assisted question-answering with human feedback. arXiv preprint arXiv:2112.09332. find our card
- Long Ouyang, Jeff Wu, Xu Jiang, Diogo Almeida, Carroll L. Wainwright, Pamela Mishkin, Chong Zhang, Sandhini Agarwal, Katarina Slama, Alex Ray, John Schulman, Jacob Hilton, Fraser Kelton, Luke E. Miller, Maddie Simens, Amanda Askell, Peter Welinder, Paul Francis Christiano, Jan Leike, and Ryan J. Low… find
- Kishore Papineni, Salim Roukos, Todd Ward, and Wei-Jing Zhu. 2002. BLEU: a method for automatic evaluation of machine translation. In ACL, pages 311–318. find our card
- Baolin Peng, Michel Galley, Pengcheng He, Chris Brockett, Lars Lidén, Elnaz Nouri, Zhou Yu, Bill Dolan, and Jianfeng Gao. 2022. GODEL: Large-scale pre-training for goal-directed dialog. ArXiv, abs/2206.11309. — [arXiv:2206.11309](https://arxiv.org/abs/2206.11309) find
- Maja Popović. 2015. chrF: character n-gram F-score for automatic MT evaluation. In Proceedings of the Tenth Workshop on Statistical Machine Translation, pages 392–395, Lisbon, Portugal. Association for Computational Linguistics. find our card
- Alec Radford, Karthik Narasimhan, Tim Salimans, and Ilya Sutskever. 2018. Improving language understanding by generative pre-training. find
- Timo Schick, Jane Dwivedi-Yu, Roberto Dessı̀, Roberta Raileanu, Maria Lomeli, Luke Zettlemoyer, Nicola Cancedda, and Thomas Scialom. 2023. Toolformer: Language models can teach themselves to use tools. find our card
- Thibault Sellam, Dipanjan Das, and Ankur Parikh. 2020. BLEURT: Learning robust metrics for text generation. In Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics, pages 7881–7892, Online. find
- Weijia Shi, Sewon Min, Michihiro Yasunaga, Minjoon Seo, Rich James, Mike Lewis, Luke Zettlemoyer, and Wen tau Yih. 2023. REPLUG: Retrieval-augmented black-box language models. ArXiv, abs/2301.12652. find our card
- Kurt Shuster, Spencer Poff, Moya Chen, Douwe Kiela, and Jason Weston. 2021. Retrieval augmentation reduces hallucination in conversation. arXiv preprint arXiv:2104.07567. find our card
- Kurt Shuster, Jing Xu, Mojtaba Komeili, Da Ju, Eric Michael Smith, Stephen Roller, Megan Ung, Moya Chen, Kushal Arora, Joshua Lane, Morteza Behrooz, W.K.F. Ngan, Spencer Poff, Naman Goyal, Arthur D. Szlam, Y-Lan Boureau, Melanie Kambadur, and Jason Weston. 2022. BlenderBot 3: a deployed conversation… find our card
- Ben Wang and Aran Komatsuzaki. 2021. GPT-J-6B: A 6 billion parameter autoregressive language model. https://github.com/kingoflolz/ mesh-transformer-jax. ↗
- Laura Weidinger, John F. J. Mellor, Maribeth Rauh, Conor Griffin, Jonathan Uesato, Po-Sen Huang, Myra Cheng, Mia Glaese, Borja Balle, Atoosa Kasirzadeh, Zachary Kenton, Sande Min-nich Brown, William T. Hawkins, Tom Stepleton, Courtney Biles, Abeba Birhane, Julia Haas, Laura Rimell, Lisa Anne Hendric… find our card
- Ronald J. Williams. 1992. Simple statistical gradient-following algorithms for connectionist reinforcement learning. Machine Learning, 8:229–256. find
- Yi-Ting Yeh, Maxine Eskenazi, and Shikib Mehri. 2021. A comprehensive assessment of dialog evaluation metrics. In arXiv. find
- Weizhe Yuan, Graham Neubig, and Pengfei Liu. 2021. BARTScore: Evaluating generated text as text generation. find our card
- Susan Zhang, Stephen Roller, Naman Goyal, Mikel Artetxe, Moya Chen, Shuohui Chen, Christopher Dewan, Mona Diab, Xian Li, Xi Victoria Lin, Todor Mihaylov, Myle Ott, Sam Shleifer, Kurt Shuster, Daniel Simig, Punit Singh Koura, Anjali Sridhar, Tianlu Wang, and Luke Zettlemoyer. 2022. OPT: Open pre-trai… find
- Tianyi Zhang, Varsha Kishore, Felix Wu, Kilian Q. Weinberger, and Yoav Artzi. 2019. BERTScore: Evaluating text generation with BERT. CoRR, abs/1904.09675. find
- Yizhe Zhang, Siqi Sun, Xiang Gao, Yuwei Fang, Chris Brockett, Michel Galley, Jianfeng Gao, and Bill Dolan. 2021. RetGen: A joint framework for retrieval and grounded text generation modeling. In AAAI Conference on Artificial Intelligence. find
- Zexuan Zhong, Tao Lei, and Danqi Chen. 2022. Training language models with memory augmentation. ArXiv, abs/2205.12674. find