Diversity Measures: Domain-Independent Proxies for Failure in Language Model Queries
Noel Ngu, Nathaniel Lee, Paulo Shakarian
TL;DR
The paper introduces domain-independent proxies for predicting failure in language model queries by quantifying response diversity. It defines three metrics—entropy $H$, Gini impurity $G$, and centroid distance $CD$—to capture diversity of outputs for a given prompt. Across multiple datasets and temperature settings, these proxies show strong correlation with the likelihood of error and are demonstrated to support few-shot prompting, chain-of-thought reasoning, and error detection. This approach provides a practical, domain-agnostic tool for assessing LM reliability and guiding prompting strategies without relying on task-specific signals.
Abstract
Error prediction in large language models often relies on domain-specific information. In this paper, we present measures for quantification of error in the response of a large language model based on the diversity of responses to a given prompt - hence independent of the underlying application. We describe how three such measures - based on entropy, Gini impurity, and centroid distance - can be employed. We perform a suite of experiments on multiple datasets and temperature settings to demonstrate that these measures strongly correlate with the probability of failure. Additionally, we present empirical results demonstrating how these measures can be applied to few-shot prompting, chain-of-thought reasoning, and error detection.
