English

API Is Enough: Conformal Prediction for Large Language Models Without Logit-Access

Computation and Language 2024-04-05 v2 Artificial Intelligence Machine Learning

Abstract

This study aims to address the pervasive challenge of quantifying uncertainty in large language models (LLMs) without logit-access. Conformal Prediction (CP), known for its model-agnostic and distribution-free features, is a desired approach for various LLMs and data distributions. However, existing CP methods for LLMs typically assume access to the logits, which are unavailable for some API-only LLMs. In addition, logits are known to be miscalibrated, potentially leading to degraded CP performance. To tackle these challenges, we introduce a novel CP method that (1) is tailored for API-only LLMs without logit-access; (2) minimizes the size of prediction sets; and (3) ensures a statistical guarantee of the user-defined coverage. The core idea of this approach is to formulate nonconformity measures using both coarse-grained (i.e., sample frequency) and fine-grained uncertainty notions (e.g., semantic similarity). Experimental results on both close-ended and open-ended Question Answering tasks show our approach can mostly outperform the logit-based CP baselines.

Keywords

Cite

@article{arxiv.2403.01216,
  title  = {API Is Enough: Conformal Prediction for Large Language Models Without Logit-Access},
  author = {Jiayuan Su and Jing Luo and Hongwei Wang and Lu Cheng},
  journal= {arXiv preprint arXiv:2403.01216},
  year   = {2024}
}
R2 v1 2026-06-28T15:07:06.485Z