跳到正文
arXiv cs.AI· Kaiyang Wan, Forrest Sheng Bao, Amin Ahmad, Yuxia Wang·· 12 小时前AI 评分31

微调目标如何塑造 LLM 幻觉检测中的分层信息

How Fine-Tuning Objectives Shape Layer-Wise Information in LLM Hallucination Detection

AI 导读

在匹配条件下,分类头微调(cls)在二元幻觉检测上持续优于 LLM-as-Judge 的序列到序列(s2s)范式,原因是 cls 学出任务特定的判别方向,使隐藏状态形成类间分离更强的双簇几何。

正文 · 原文

View PDF HTML (experimental)

Abstract:Under matched conditions, classifier-head fine-tuning (\textsc{cls}) paradigm consistently outperforms the LLM-as-Judge sequence-to-sequence (\textsc{s2s}) on binary hallucination detection. We trace this gap to the representations induced by the two objectives. Static analysis shows that \textsc{cls} learns a task-specific discriminative direction, pushing hidden states into a clear two-cluster geometry with stronger class separation and linearly decodable label information. By contrast, \textsc{s2s} relies on the pretrained language-modeling head to make decisions. This minimizes the pressure to reorganize representations, leaving the hidden states much closer to the pretrained backbone. Our analysis also reveals that the \textsc{cls} signal saturates before the final layer, after which upper layers form a plateau of statistically interchangeable and largely redundant representations. However, endpoint redundancy alone does not show whether these layers are dispensable during training. Further training-dynamics analysis reveals 1) a lower pretrained substrate, 2) a narrow upper-middle \textbf{InfoWindow} where task information emerges, and 3) a later-synchronizing upper plateau that mirrors this signal. This mechanism motivates us to remove layers above the InfoWindow. Truncating 31\% of the network leaves accuracy unchanged within seed noise on Qwen3-4B and 8B and cuts latency by $1.4\times$. The same depth fraction, fixed before any analysis of the target models, transfers to Llama-3.2-3B and Mistral-7B-v0.3, and the layer structure behind it reappears there and on QNLI and SST-2. We hope this helps guide efficient task-specific LLM design.
Comments: 37 pages, 6 figures, NeurIPS 2026
Subjects: Artificial Intelligence (cs.AI)
Cite as: arXiv:2610.27711 [cs.AI]
  (or arXiv:2610.27711v1 [cs.AI] for this version)
  https://doi.org/10.48550/arXiv.2610.27711

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Kaiyang Wan [view email]
[v1] Tue, 20 Oct 2026 06:30:27 UTC (269 KB)

来源:arXiv cs.AI · arxiv.org