Geometric Deviation as an Unsupervised Pre-Generation Reliability Signal: Probing LLM Representations for Answerability

· Source: Paper Index on ACL Anthology · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

A new study introduces geometric deviation as an unsupervised pre-generation reliability signal for large language models, designed to indicate when a query is outside an LLM's knowledge before it generates an output. This method measures the deviation of hidden states from an answerable reference set, requiring no labeled failure data or access to model outputs. Evaluated across Llama 3.1-8B, Qwen 2.5-7B, and Mistral-7B-Instruct with Math, Fact, and Code prompts, the signal proved strong for mathematical prompts, achieving ROC-AUC scores of 0.78-0.84. It outperformed simple refusal baselines and compared favorably to self-consistency. While no reliable signal emerged for factual prompts, code prompts showed large effect sizes with higher variance. Layer-wise analysis revealed the signal originates in early layers and attenuates towards the output, suggesting early establishment of answerability-related geometry.

Key takeaway

For AI Engineers deploying LLMs in critical applications where answerability is paramount, especially in structured domains like mathematics or code, consider integrating geometric deviation. This unsupervised pre-generation signal offers a lightweight method to identify queries outside the model's knowledge, improving reliability without needing labeled failure data. You can enhance system robustness by leveraging this early warning mechanism.

Key insights

LLM representation geometry can signal pre-generation answerability, particularly in structured domains like mathematics.

Principles

Method

Measure hidden state deviation from an answerable reference set, requiring no labeled failure data or model outputs for pre-generation reliability.

In practice

Topics

Best for: Research Scientist, NLP Engineer, AI Scientist, Machine Learning Engineer, AI Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Paper Index on ACL Anthology.