Improve Decoding Factuality by Token-wise Cross Layer Entropy of Large Language Models

Jialiang Wu, Yi Shen, Sijia Liu, Yi Tang, Sen Song +2 more
2/5/2025
cs.CLcs.AI

No official code found. Best implementation: Arcade-Master/END (5 stars, MIT, last commit 2025-07-05).

Abstract

Despite their impressive capacities, Large language models (LLMs) often struggle with the hallucination issue of generating inaccurate or fabricated content even when they possess correct knowledge. In this paper, we extend the exploration of the correlation between hidden-state prediction changes and output factuality into a deeper, token-wise level. Based on the insights , we propose cross-layer Entropy eNhanced Decoding (END), a decoding method that mitigates hallucinations without requiring extra training. END leverages inner probability changes across layers to individually quantify the factual knowledge required for each candidate token, and adjusts the final predicting distribution to prioritize tokens with higher factuality. Experiments on both hallucination and QA benchmarks demonstrate that END significantly enhances the truthfulness and informativeness of generated content while maintaining robust QA accuracy. Moreover, our work provides a deeper perspective on understanding the correlations between inherent knowledge and output factuality.

View on arXivView PDF

Code Implementations(1)

Arcade-Master/ENDCommunity73%
MIT

Implementation for the paper "Improve Decoding Factuality by Token-wise Cross Layer Entropy of Large Language Models"

50C, C++Jan 25, 20251 years agoMIT

Similar Papers

Cite this paper

@article{wu2025improve,
  title  = {Improve Decoding Factuality by Token-wise Cross Layer Entropy of Large Language Models},
  author = {Jialiang Wu and Yi Shen and Sijia Liu and Yi Tang and Sen Song and Xiaoyi Wang and Longjun Cai},
  year   = {2025},
  eprint = {2502.03199},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2502.03199v1}
}

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/2502.03199.svg)](https://codeofpaper.com/paper/2502.03199)

Discussion