API Is Enough: Conformal Prediction for Large Language Models Without Logit-Access
Su, Jiayuan, Luo, Jing, Wang, Hongwei, Cheng, Lu
–arXiv.org Artificial Intelligence
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.
arXiv.org Artificial Intelligence
Apr-3-2024
- Country:
- Oceania
- North America
- The Bahamas (0.04)
- Costa Rica (0.04)
- United States
- Tennessee (0.04)
- New York > New York County
- New York City (0.04)
- Illinois
- Cook County > Chicago (0.04)
- Champaign County > Urbana (0.04)
- Canada > Ontario
- Toronto (0.04)
- Europe
- Asia
- Japan (0.04)
- Middle East
- Jordan (0.04)
- UAE > Abu Dhabi Emirate
- Abu Dhabi (0.04)
- Africa
- Middle East > Algeria (0.04)
- Angola (0.04)
- Genre:
- Research Report > New Finding (0.46)
- Industry:
- Transportation > Air (0.68)
- Leisure & Entertainment > Sports
- Hockey (0.46)
- Education > Curriculum
- Subject-Specific Education (0.47)
- Technology: