Entropy Guided Extrapolative Decoding to Improve Factuality in Large Language Models

Souvik Das,Lifeng Jin,Linfeng Song,Haitao Mi,Baolin Peng,Dong Yu
2024-04-15
Abstract:Large language models (LLMs) exhibit impressive natural language capabilities but suffer from hallucination -- generating content ungrounded in the realities of training data. Recent work has focused on decoding techniques to improve factuality during inference by leveraging LLMs' hierarchical representation of factual knowledge, manipulating the predicted distributions at inference time. Current state-of-the-art approaches refine decoding by contrasting early-exit distributions from a lower layer with the final layer to exploit information related to factuality within the model forward procedure. However, such methods often assume the final layer is the most reliable and the lower layer selection process depends on it. In this work, we first propose extrapolation of critical token probabilities beyond the last layer for more accurate contrasting. We additionally employ layer-wise entropy-guided lower layer selection, decoupling the selection process from the final layer. Experiments demonstrate strong performance - surpassing state-of-the-art on multiple different datasets by large margins. Analyses show different kinds of prompts respond to different selection strategies.
Computation and Language
What problem does this paper attempt to address?
The paper focuses on the problem of large-scale language models (LLMs) generating content that is inconsistent with the training data (i.e., hallucinations). Existing methods improve factual accuracy by comparing low-level and high-level prediction distributions, but they rely on the maturity of the final layer. The paper proposes a new logit extrapolation method for inference time, which enhances the maturity of the higher layers by extrapolating the probabilities of key tokens and uses entropy-guided hierarchical selection strategy to independently select the comparison layers apart from the final layer. Experiments show that this method significantly improves factual accuracy and reduces hallucination generation on multiple datasets.