English

A Little Confidence Goes a Long Way

Machine Learning 2024-08-22 v1 Artificial Intelligence Computation and Language Information Theory Neural and Evolutionary Computing math.IT

Abstract

We introduce a group of related methods for binary classification tasks using probes of the hidden state activations in large language models (LLMs). Performance is on par with the largest and most advanced LLMs currently available, but requiring orders of magnitude fewer computational resources and not requiring labeled data. This approach involves translating class labels into a semantically rich description, spontaneous symmetry breaking of multilayer perceptron probes for unsupervised learning and inference, training probes to generate confidence scores (prior probabilities) from hidden state activations subject to known constraints via entropy maximization, and selecting the most confident probe model from an ensemble for prediction. These techniques are evaluated on four datasets using five base LLMs.

Keywords

Cite

@article{arxiv.2408.11239,
  title  = {A Little Confidence Goes a Long Way},
  author = {John Scoville and Shang Gao and Devanshu Agrawal and Javed Qadrud-Din},
  journal= {arXiv preprint arXiv:2408.11239},
  year   = {2024}
}

Comments

13 pages, 2 figures

R2 v1 2026-06-28T18:18:50.204Z