Improving Uncertainty Estimation through Semantically Diverse Language Generation
Lukas Aichberger Kajetan Schweighofer Mykyta Ielanskyi Sepp Hochreiter
Large language models (LLMs) can suffer from hallucinations when generating text. These hallucinations impede various applications in society and industry by making LLMs untrustworthy. Current LLMs generate text in an autoregressive fashion by predicting and appending text tokens. When an LLM is uncertain about the semantic meaning of the next tokens to generate, it is likely to start hallucinating. Thus, it has been suggested that predictive uncertainty is one of the main causes of hallucinations. We introduce Semantically Diverse Language Generation (SDLG) to quantify predictive uncertainty in LLMs. SDLG steers the LLM to generate semantically diverse yet likely alternatives for an initially generated text. This approach provides a precise measure of aleatoric semantic uncertainty, detecting whether the initial text is likely to be hallucinated. Experiments on question-answering tasks demonstrate that SDLG consistently outperforms existing methods while being the most computationally efficient, setting a new standard for uncertainty estimation in LLMs.
llm, nlg, uncertainty estimation, uncertainty measures, semantic uncertainty, aleatoric uncertainty, semantic entropy, mc estimation, importance sampling
We introduce SDLG, an efficient method to accurately estimate aleatoric semantic uncertainty in LLMs.