Beyond Confidence: The Rhythms of Reasoning in Generative Models

TL;DR

This paper introduces the Token Constraint Bound ($elta_{ ext{TCB}}$), a new metric to measure the internal stability of Large Language Models, revealing prediction sensitivities overlooked by traditional metrics like perplexity.

Contribution

We propose $elta_{ ext{TCB}}$, a novel stability metric linked to embedding geometry, to assess and improve LLM prediction robustness against input perturbations.

Findings

$elta_{ ext{TCB}}$ correlates with prompt engineering effectiveness.

It uncovers prediction instabilities missed by perplexity.

$elta_{ ext{TCB}}$ provides insights into LLM internal state resilience.

Abstract

Large Language Models (LLMs) exhibit impressive capabilities yet suffer from sensitivity to slight input context variations, hampering reliability. Conventional metrics like accuracy and perplexity fail to assess local prediction robustness, as normalized output probabilities can obscure the underlying resilience of an LLM's internal state to perturbations. We introduce the Token Constraint Bound ($\delta_{\mathrm{TCB}}$), a novel metric that quantifies the maximum internal state perturbation an LLM can withstand before its dominant next-token prediction significantly changes. Intrinsically linked to output embedding space geometry, $\delta_{\mathrm{TCB}}$ provides insights into the stability of the model's internal predictive commitment. Our experiments show $\delta_{\mathrm{TCB}}$ correlates with effective prompt engineering and uncovers critical prediction instabilities missed by perplexity during in-context learning and text generation. $\delta_{\mathrm{TCB}}$ offers a principled, complementary approach to analyze and potentially improve the contextual stability of LLM predictions.