Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Mitigating Hallucinations in Large Language Models Via Decoder Layer Skipping

About

Large Language Models (LLMs) have achieved strong performance across diverse natural language tasks, yet their outputs often suffer from hallucinations -- content that is misaligned with factual information. In this work, we conduct a comprehensive layer-wise analysis of the decoding process and reveal that hallucinations tend to originate from deeper decoder layers. To address this issue, we introduce \textbf{DeLask} (\textbf{De}coder \textbf{La}yer \textbf{Sk}ipping), a novel decoding framework that dynamically skips layers prone to producing hallucinations. DeLask leverages the theoretical insight that the forward computation of an $L$-layer Transformer is conditionally equivalent to $L$ steps of gradient descent. We define a \emph{driftance value} by computing the cosine similarity between gradients derived from consecutive decoder steps, identifying problematic layers when the descent direction reverses. Rather than discarding such layers entirely, DeLask partially aggregates their hidden states with preceding layers, thereby preserving consistency while suppressing erroneous signals. Extensive experiments across diverse LLMs and benchmarks demonstrate that DeLask consistently mitigates hallucinations and enhances overall reliability, providing a lightweight and generalizable decoding framework for improving the robustness of large-scale language models.

Hanze Li, Jinhao You, Yichen Guo, Kai Tang, Shuangyang Xie, Xiande Huang• 2026

Related benchmarks

TaskDatasetResultRank
Multiple-Choice ClassificationMMLU
Accuracy65.43
47
Open-ended generationTriviaQA--
37
Text GenerationGSM8K
Accuracy55.42
35
Free-form text generationCoQA
Accuracy68.05
22
Multiple-choice Question AnsweringARC-C
Accuracy54.1
18
Factual HallucinationTruthfulQA
MC1 Score41.74
12
Factual HallucinationFACTOR News
Accuracy68.56
12
Factual HallucinationFACTOR Wiki
Accuracy58.71
12
Showing 8 of 8 rows

Other info

Follow for update