Is Long Context All You Need? Leveraging LLM's Extended Context for NL2SQL

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chung, Yeounoh, Kakkar, Gaurav T., Gan, Yu, Milne, Brenton, Ozcan, Fatma
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916789345058816
author Chung, Yeounoh
Kakkar, Gaurav T.
Gan, Yu
Milne, Brenton
Ozcan, Fatma
author_facet Chung, Yeounoh
Kakkar, Gaurav T.
Gan, Yu
Milne, Brenton
Ozcan, Fatma
contents Large Language Models (LLMs) have demonstrated impressive capabilities across a range of natural language processing tasks. In particular, improvements in reasoning abilities and the expansion of context windows have opened new avenues for leveraging these powerful models. NL2SQL is challenging in that the natural language question is inherently ambiguous, while the SQL generation requires a precise understanding of complex data schema and semantics. One approach to this semantic ambiguous problem is to provide more and sufficient contextual information. In this work, we explore the performance and the latency trade-offs of the extended context window (a.k.a., long context) offered by Google's state-of-the-art LLM (\textit{gemini-1.5-pro}). We study the impact of various contextual information, including column example values, question and SQL query pairs, user-provided hints, SQL documentation, and schema. To the best of our knowledge, this is the first work to study how the extended context window and extra contextual information can help NL2SQL generation with respect to both accuracy and latency cost. We show that long context LLMs are robust and do not get lost in the extended contextual information. Additionally, our long-context NL2SQL pipeline based on Google's \textit{gemini-pro-1.5} achieve strong performances on various benchmark datasets without finetuning and expensive self-consistency based techniques.
format Preprint
id arxiv_https___arxiv_org_abs_2501_12372
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Is Long Context All You Need? Leveraging LLM's Extended Context for NL2SQL
Chung, Yeounoh
Kakkar, Gaurav T.
Gan, Yu
Milne, Brenton
Ozcan, Fatma
Databases
Artificial Intelligence
Large Language Models (LLMs) have demonstrated impressive capabilities across a range of natural language processing tasks. In particular, improvements in reasoning abilities and the expansion of context windows have opened new avenues for leveraging these powerful models. NL2SQL is challenging in that the natural language question is inherently ambiguous, while the SQL generation requires a precise understanding of complex data schema and semantics. One approach to this semantic ambiguous problem is to provide more and sufficient contextual information. In this work, we explore the performance and the latency trade-offs of the extended context window (a.k.a., long context) offered by Google's state-of-the-art LLM (\textit{gemini-1.5-pro}). We study the impact of various contextual information, including column example values, question and SQL query pairs, user-provided hints, SQL documentation, and schema. To the best of our knowledge, this is the first work to study how the extended context window and extra contextual information can help NL2SQL generation with respect to both accuracy and latency cost. We show that long context LLMs are robust and do not get lost in the extended contextual information. Additionally, our long-context NL2SQL pipeline based on Google's \textit{gemini-pro-1.5} achieve strong performances on various benchmark datasets without finetuning and expensive self-consistency based techniques.
title Is Long Context All You Need? Leveraging LLM's Extended Context for NL2SQL
topic Databases
Artificial Intelligence
url https://arxiv.org/abs/2501.12372