arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2605.08057cs.CLcs.AI
收藏

CA-SQL:基于探索与计算预算分配的文本到SQL推理方法

CA-SQL: Complexity-Aware Inference Time Reasoning for Text-to-SQL via Exploration and Compute Budget Allocation

Brandeis University(布雷纳大学)

James Petullo, Nianwen Xue

AI总结:

CA-SQL通过动态扩展探索广度和自定义提示种子方法,提升文本到SQL任务在Bird-Bench基准中的表现,达到51.72%的挑战性问题得分。

摘要:

尽管最近的推理时间学习进步提升了LLM在文本到SQL任务中的推理能力,但当前解决方案在Bird-Bench(BIRD)基准中最具挑战性的任务上仍表现不佳。这归因于探索解决方案空间不足,无法发现有前途的候选查询以进一步细化为正确输出。为解决这一挑战,我们引入了CA-SQL,一种新的文本到SQL流水线,利用任务估计难度动态调整探索广度以生成解决方案候选。此外,我们使用基于进化搜索原理的自定义提示种子方法,进一步激发基础LLM的探索行为,并采用新的投票方法在搜索结束时选择最佳候选解决方案。实验表明,我们的解决方案仅使用GPT-4o-mini就实现了BIRD开发集“挑战性”层级问题的51.72%得分,优于其他上下文学习方法,即使那些利用更大模型的方法。总体而言,我们的方法在BIRD开发数据集上达到了竞争性的61.06%执行准确率和68.77%软F1得分。

英文摘要:

While recent advancements in inference-time learning have improved LLM reasoning on Text-to-SQL tasks, current solutions still struggle to perform well on the most challenging tasks in the Bird-Bench (BIRD) benchmark. This is due to inadequate solution space exploration, which is necessary to uncover promising candidate queries that can be further refined to produce the correct output. To address this challenge, we introduce CA-SQL, a novel Text-to-SQL pipeline that utilizes the estimated difficulty of a task to dynamically scale the breadth of the exploration for generating solution candidates. In addition, we use a custom prompt seeding method, based on principles of evolutionary search, to further elicit exploratory behavior from the base LLM and a novel voting method to select the best candidate solution at the end of the search. Experiments demonstrate that our solution achieves a state-of-the-art score of 51.72% on the "challenging" tier of BIRD development set problems, using only GPT-4o-mini, out-performing other in-context learning approaches, even those that leverage larger models. Overall, our method attains a competitive 61.06% execution accuracy and 68.77% Soft F1 score on the BIRD development dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑