Confidence Estimation for Error Detection in Text-to-SQL Systems

Authors

  • Oleg Somov AIRI, Moscow, Russia MIPT, Dolgoprudny, Russia
  • Elena Tutubalina AIRI, Moscow, Russia Sber AI, Moscow, Russia ISP RAS Research Center for Trusted Artificial Intelligence, Moscow, Russia

DOI:

https://doi.org/10.1609/aaai.v39i23.34699

Abstract

Text-to-SQL enables users to interact with databases through natural language, simplifying the retrieval and synthesis of information. Despite the success of large language models (LLMs) in converting natural language questions into SQL queries, their broader adoption is limited by two main challenges: achieving robust generalization across diverse queries and ensuring interpretative confidence in their predictions. To tackle these issues, our research investigates the integration of selective classifiers into Text-to-SQL systems. We analyse the trade-off between coverage and risk using entropy based confidence estimation with selective classifiers and assess its impact on the overall performance of Text-to-SQL models. Additionally, we explore the models' initial calibration and improve it with calibration techniques for better model alignment between confidence and accuracy. Our experimental results show that encoder-decoder T5 is better calibrated than in-context-learning GPT 4 and decoder-only Llama 3, thus the designated external entropy-based selective classifier has better performance. The study also reveal that, in terms of error detection, selective classifier with a higher probability detects errors associated with irrelevant questions rather than incorrect query generations.

Published

2025-04-11

How to Cite

Somov, O., & Tutubalina, E. (2025). Confidence Estimation for Error Detection in Text-to-SQL Systems. Proceedings of the AAAI Conference on Artificial Intelligence, 39(23), 25137–25145. https://doi.org/10.1609/aaai.v39i23.34699

Issue

Section

AAAI Technical Track on Natural Language Processing II