Mitigating Hallucination in Abstractive Summarization with Domain-Conditional Mutual Information

Kyubyung Chae, Jaepill Choi, Yohan Jo, Taesup Kim
4/15/2024
cs.CLcs.AI

Official code: qqplot/dcpmi (9 stars, MIT, last commit 2024-08-09).

Abstract

A primary challenge in abstractive summarization is hallucination -- the phenomenon where a model generates plausible text that is absent in the source text. We hypothesize that the domain (or topic) of the source text triggers the model to generate text that is highly probable in the domain, neglecting the details of the source text. To alleviate this model bias, we introduce a decoding strategy based on domain-conditional pointwise mutual information. This strategy adjusts the generation probability of each token by comparing it with the token's marginal probability within the domain of the source text. According to evaluation on the XSUM dataset, our method demonstrates improvement in terms of faithfulness and source relevance. The code is publicly available at \url{https://github.com/qqplot/dcpmi}.

View on arXivView PDF

Code Implementations(1)

qqplot/dcpmiOfficial100%
MIT

[NAACL24] Official Implementation of Mitigating Hallucination in Abstractive Summarization with Domain-Conditional Mutual Information

93C, C++Nov 27, 20232 years agoMIT
hallucinationnaacl2024

Similar Papers

Cite this paper

@article{chae2024mitigating,
  title  = {Mitigating Hallucination in Abstractive Summarization with Domain-Conditional Mutual Information},
  author = {Kyubyung Chae and Jaepill Choi and Yohan Jo and Taesup Kim},
  year   = {2024},
  eprint = {2404.09480},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2404.09480v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2404.09480.svg)](https://codeofpaper.com/paper/2404.09480)

Discussion