English

Sequence-Level Certainty Reduces Hallucination In Knowledge-Grounded Dialogue Generation

Computation and Language 2024-04-16 v3 Artificial Intelligence

Abstract

In this work, we propose sequence-level certainty as a common theme over hallucination in Knowledge Grounded Dialogue Generation (KGDG). We explore the correlation between the level of hallucination in model responses and two types of sequence-level certainty: probabilistic certainty and semantic certainty. Empirical results reveal that higher levels of both types of certainty in model responses are correlated with lower levels of hallucination. We further propose Certainty-based Response Ranking (CRR), a decoding-time hallucination mitigation method that samples several response candidates, ranks them based on sequence-level certainty, and outputs the response with the highest certainty level. Aligning with our definitions of sequence-level certainty, we design 2 types of CRR approaches: Probabilistic CRR (P-CRR) and Semantic CRR (S-CRR). P-CRR ranks individually sampled model responses using the arithmetic mean log-probability of the entire sequence. S-CRR approaches certainty estimation from meaning-space, and ranks model response candidates based on their semantic certainty level as measured by an entailment-based Agreement Score (AS). Through extensive experiments across 3 KGDG datasets, 3 decoding methods, and 4 KGDG models, we validate the effectiveness of CRR for reducing hallucination in KGDG task.

Cite

@article{arxiv.2310.18794,
  title  = {Sequence-Level Certainty Reduces Hallucination In Knowledge-Grounded Dialogue Generation},
  author = {Yixin Wan and Fanyou Wu and Weijie Xu and Srinivasan H. Sengamedu},
  journal= {arXiv preprint arXiv:2310.18794},
  year   = {2024}
}
R2 v1 2026-06-28T13:04:46.511Z