atlas

Research map / Large language models / Language, safety and interpretability

Characterizing Truthfulness in Large Language Model Generations with Local Intrinsic Dimension

ICML 2024 · Large language models › Language, safety and interpretability · approach: Statistical frameworks for LLM evaluation · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes detecting LLM hallucinations by estimating the local intrinsic dimension (LID) of the last-token activations from a middle layer, using a distance-corrected MLE estimator, and reports it beats entropy, verbalized-uncertainty, and probe baselines on four QA tasks with Llama-2.

Problem

Predicting whether LLM generations are truthful (correct) in generative QA, where entropy-based uncertainty is intractable over open output spaces and verbalized or probe-based methods are unreliable.

Approach

Use the local intrinsic dimension of intermediate-layer representations (MLE estimator with a distance-aware polynomial correction, called LID-GeoMLE) as a per-sample truthfulness signal, based on the hypothesis that truthful outputs lie on lower-dimensional manifolds than outputs that mix human and model distributions.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use distance-aware GeoMLE instead of vanilla MLE (ablation)LID truthfulness detection · QA tasks (Llama-2-7B and Llama-2-13B)AUROC: 0.758 -> 0.767 (CoQA, Llama-2-7B); 0.748 -> 0.775 (TriviaQA 0-shot, Llama-2-13B)Helped
use LID-GeoMLE instead of the strongest entropy baselineLlama-2-7B truthfulness detection · TriviaQA 0-shotAUROC: 0.718 -> 0.771Helped
use LID-GeoMLE instead of the strongest entropy baselineLlama-2-13B truthfulness detection · TriviaQA 0-shotAUROC: 0.751 -> 0.775Helped
use CoQA samples as LID reference neighbors instead of HotpotQA samples (ablation)LID truthfulness detection · HotpotQAAUROC: 0.747 -> 0.753Helped
use TriviaQA samples as LID reference neighbors instead of TydiQA samples (ablation)LID truthfulness detection · TriviaQAAUROC: 0.765 -> 0.748No clear effect
increase the number of nearest neighbors (ablation)LID truthfulness detection · QA tasksAUROCHelped
instruction-tune Llama-2-7BLlama-2-7B · TriviaQA and TydiQAaccuracyHelped
continue instruction tuningLlama-2-7B · TriviaQA and TydiQAlocal intrinsic dimensionMixed
mix a human answer prefix with a model-generated continuationTriviaQA answer representations · Controlled answer mixinglocal intrinsic dimensionHelped

Surprising findings

Limitations

Experimental setup

Datasets: TriviaQA CoQA HotpotQA TydiQA-GP Super-NI

Benchmarks: CoQA TydiQA-GP TriviaQA HotpotQA

Metrics: AUROC

Baselines: Predictive Entropy LN-Predictive Entropy Semantic Entropy SAPLMA P(True) SAR LID-MLE KNN TwoNN t-SNE

Builds on: MLE intrinsic dimension estimator (Levina and Bickel) semantic entropy SAPLMA P(True)

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy