Improving the Faithfulness of LLM-based Abstractive Summarization with Span-level Unlikelihood Training
Summary
A novel fine-tuning approach enhances the faithfulness of LLM-based abstractive summarization by reducing hallucinations – unfaithful words, phrases, or concepts. Researchers Sicong Huang, Qianqi Yan, Shengze Wang, and Ian Lane developed a method that first annotates span-level hallucinations in LLM-generated summaries using GPT-4 for automatic labeling. Subsequently, the Large Language Model is fine-tuned using both summaries devoid of hallucinations and the identified hallucinated text spans. This technique, termed unlikelihood training, significantly improved summary faithfulness. Experimental results demonstrate a reduction in summaries with hallucinations from 31% to 13% (a 58% decrease) on the CNN summarization dataset, and from 33% to 20% (a 39% decrease) on the SAMSum dataset. The study also introduces a new dataset for distinguishing faithful and unfaithful content and compares unlikelihood training against gradient ascent and task vector negation.
Key takeaway
For NLP Engineers developing abstractive summarization systems, this research offers a direct path to mitigate hallucinations. You should consider implementing span-level unlikelihood training, potentially using large models like GPT-4 for automated annotation of unfaithful text. This approach can substantially reduce hallucination rates, improving summary trustworthiness and user confidence in your LLM-generated content. Explore the provided dataset for distinguishing faithful and unfaithful content to refine your models.
Key insights
Unlikelihood training with span-level hallucination annotations significantly improves LLM abstractive summarization faithfulness.
Principles
- LLM unfaithfulness manifests as span-level hallucinations.
- Fine-tuning can directly target hallucinated text spans.
- Automatic labeling can create training data for faithfulness.
Method
Annotate span-level hallucinations using GPT-4, then fine-tune LLMs with unlikelihood training on both hallucination-free summaries and identified unfaithful spans to reduce their occurrence.
In practice
- Use GPT-4 for automated hallucination annotation.
- Apply unlikelihood training for summarization fine-tuning.
- Develop datasets distinguishing faithful/unfaithful content.
Topics
- LLM Summarization
- Abstractive Summarization
- Hallucination Mitigation
- Unlikelihood Training
- Span-level Annotation
- GPT-4
Best for: Research Scientist, AI Scientist, NLP Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Paper Index on ACL Anthology.