Improving the Faithfulness of LLM-based Abstractive Summarization with Span-level Unlikelihood Training

· Source: Paper Index on ACL Anthology · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Natural Language Processing · Depth: Expert, medium

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

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

Topics

Best for: Research Scientist, AI Scientist, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Paper Index on ACL Anthology.