The First Token Knows: Single-Decode Confidence for Hallucination Detection

Mina Gabriel

arXiv:2605.05166 · 2026-05-08 공개 · arXiv · PDF

Abstract

Self-consistency detects hallucinations by generating multiple sampled answers to a question and measuring agreement, but this requires repeated decoding and can be sensitive to lexical variation. Semantic self-consistency improves this by clustering sampled answers by meaning using natural language inference, but it adds both sampling cost and external inference overhead. We show that first-token confidence, phi_first, computed from the normalized entropy of the top-K logits at the first content-bearing answer token of a single greedy decode, matches or modestly exceeds semantic self-consistency on closed-book short-answer factual question answering. Across three 7-8B instruction-tuned models and two benchmarks, phi_first achieves a mean AUROC of 0.820, compared with 0.793 for semantic agreement and 0.791 for standard surface-form self-consistency. A subsumption test shows that phi_first is moderately to strongly correlated with semantic agreement, and combining the two signals yields only a small AUROC improvement over phi_first alone. These results suggest that much of the uncertainty information captured by multi-sample agreement is already available in the model's initial token distribution. We argue that phi_first should be reported as a default low-cost baseline before invoking sampling-based uncertainty estimation.

한국어 요약

📋 한 줄 요약

**[LLM 신뢰성]** 단일 그리디 디코드의 첫 콘텐츠 토큰에서 계산한 정규화 엔트로피 phi_first가 다중 샘플 자기일관성보다 환각 검출에서 동등하거나 우수함을 보인다.

🎯 핵심 기여도

💡 핵심 아이디어

다중 샘플 합의가 포착하는 불확실성 정보의 상당 부분이 모델의 초기 토큰 분포에 이미 들어 있다. 따라서 여러 번 디코딩하지 않고도 첫 토큰 엔트로피만으로 환각을 효과적으로 탐지할 수 있다.

🔬 기술적 접근법

📊 주요 결과

💭 의의 및 한계

**의의**: 추가 샘플링·외부 모델 없이 환각을 탐지할 수 있어 비용·지연 면에서 실용적. **한계**: 폐쇄형 단답 QA에 한정된 평가로, 장문/복합 추론·생성에서의 일반화는 미검증.

🚀 실용적 활용