Text-to-SQL Calibration: No Need to Ask -- Just Rescale Model Probabilities

Ashwin Ramachandran, Sunita Sarawagi
11/23/2024
cs.DBcs.AIcs.IR

No confident code implementation found for this paper.

Abstract

Calibration is crucial as large language models (LLMs) are increasingly deployed to convert natural language queries into SQL for commercial databases. In this work, we investigate calibration techniques for assigning confidence to generated SQL queries. We show that a straightforward baseline -- deriving confidence from the model's full-sequence probability -- outperforms recent methods that rely on follow-up prompts for self-checking and confidence verbalization. Our comprehensive evaluation, conducted across two widely-used Text-to-SQL benchmarks and multiple LLM architectures, provides valuable insights into the effectiveness of various calibration strategies.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{ramachandran2024texttosql,
  title  = {Text-to-SQL Calibration: No Need to Ask -- Just Rescale Model Probabilities},
  author = {Ashwin Ramachandran and Sunita Sarawagi},
  year   = {2024},
  eprint = {2411.16742},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2411.16742v1}
}

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/2411.16742.svg)](https://codeofpaper.com/paper/2411.16742)

Discussion