Evaluating language models as risk scores

André F. Cruz, Moritz Hardt, Celestine Mendler-Dünner
7/19/2024
cs.LGcs.CL

No official code found. Best implementation: socialfoundations/folktexts (30 stars, MIT, last commit 2025-12-14).

Abstract

Current question-answering benchmarks predominantly focus on accuracy in realizable prediction tasks. Conditioned on a question and answer-key, does the most likely token match the ground truth? Such benchmarks necessarily fail to evaluate LLMs' ability to quantify ground-truth outcome uncertainty. In this work, we focus on the use of LLMs as risk scores for unrealizable prediction tasks. We introduce folktexts, a software package to systematically generate risk scores using LLMs, and evaluate them against US Census data products. A flexible API enables the use of different prompting schemes, local or web-hosted models, and diverse census columns that can be used to compose custom prediction tasks. We evaluate 17 recent LLMs across five proposed benchmark tasks. We find that zero-shot risk scores produced by multiple-choice question-answering have high predictive signal but are widely miscalibrated. Base models consistently overestimate outcome uncertainty, while instruction-tuned models underestimate uncertainty and produce over-confident risk scores. In fact, instruction-tuning polarizes answer distribution regardless of true underlying data uncertainty. This reveals a general inability of instruction-tuned LLMs to express data uncertainty using multiple-choice answers. A separate experiment using verbalized chat-style risk queries yields substantially improved calibration across instruction-tuned models. These differences in ability to quantify data uncertainty cannot be revealed in realizable settings, and highlight a blind-spot in the current evaluation ecosystem that folktexts covers.

View on arXivView PDF

Code Implementations(1)

MIT

Evaluate uncertainty, calibration, accuracy, and fairness of LLMs on real-world survey data!

306TeX, PythonDec 12, 202310 months agoMIT
fairnesslarge-language-modelsmachine-learningpythontabular-data+2 more

Similar Papers

Cite this paper

@article{cruz2024evaluating,
  title  = {Evaluating language models as risk scores},
  author = {André F. Cruz and Moritz Hardt and Celestine Mendler-Dünner},
  year   = {2024},
  eprint = {2407.14614},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2407.14614},
  url    = {http://arxiv.org/abs/2407.14614v3},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2407.14614.svg)](https://codeofpaper.com/paper/2407.14614)

Discussion