On the Universal Truthfulness Hyperplane Inside LLMs

Junteng Liu,Shiqi Chen,Yu Cheng,Junxian He
2024-07-11
Abstract:While large language models (LLMs) have demonstrated remarkable abilities across various fields, hallucination remains a significant challenge. Recent studies have explored hallucinations through the lens of internal representations, proposing mechanisms to decipher LLMs' adherence to facts. However, these approaches often fail to generalize to out-of-distribution data, leading to concerns about whether internal representation patterns reflect fundamental factual awareness, or only overfit spurious correlations on the specific datasets. In this work, we investigate whether a universal truthfulness hyperplane that distinguishes the model's factually correct and incorrect outputs exists within the model. To this end, we scale up the number of training datasets and conduct an extensive evaluation -- we train the truthfulness hyperplane on a diverse collection of over 40 datasets and examine its cross-task, cross-domain, and in-domain generalization. Our results indicate that increasing the diversity of the training datasets significantly enhances the performance in all scenarios, while the volume of data samples plays a less critical role. This finding supports the optimistic hypothesis that a universal truthfulness hyperplane may indeed exist within the model, offering promising directions for future research.
Computation and Language
What problem does this paper attempt to address?
### Problems Addressed by the Paper This paper primarily investigates the issue of "hallucination" commonly found in large language models (LLMs) across various tasks. Although existing research explores the accuracy of LLMs through internal representations and proposes mechanisms to decode LLMs' adherence to facts, these methods often fail to generalize well to out-of-distribution (OOD) data. Therefore, the core questions of the paper are: 1. **Do internal representation features truly capture the model's internal hallucinations, or do they merely overfit to the patterns of specific datasets?** 2. **Is there a universal truthfulness hyperplane that can classify factual correctness across diverse tasks?** To answer these questions, the researchers increased the diversity of the training dataset, constructing a comprehensive collection that includes more than 40 different datasets, and evaluated the performance of the truthfulness hyperplane across different tasks and domains. Experimental results indicate that increasing the diversity of the training dataset significantly improves performance in all scenarios, supporting the hypothesis of a universal truthfulness hyperplane and providing new directions for future research.