中文

ACT-SQL:用于文本到 SQL 的自动生成思维链的上下文学习

计算与语言 2023-10-27 v1

摘要

近来,大语言模型(LLMs)已被证明在多个领域和任务中具有强大能力。我们研究文本到 SQL 任务中的提示设计问题,并尝试提升 LLMs 在生成 SQL 查询时的推理能力。除平凡的少样本上下文学习设置外,我们以类似于模式链接的方法设计了思维链(CoT)提示。我们提出了一种名为 ACT-SQL 的方法,可自动生成自动思维链示例,因此整个过程无需人工标注。我们的方法节省成本,因为在生成一条 SQL 查询时仅调用一次 LLMs 的 API。此外,我们将上下文学习方法扩展到多轮文本到 SQL 任务。实验结果表明,LLMs 的性能可从我们的 ACT-SQL 方法中获益。我们的方法在现有上下文学习方法中于 Spider 开发集上取得了 SOTA 性能。

关键词

引用

@article{arxiv.2310.17342,
  title  = {ACT-SQL: In-Context Learning for Text-to-SQL with Automatically-Generated Chain-of-Thought},
  author = {Hanchong Zhang and Ruisheng Cao and Lu Chen and Hongshen Xu and Kai Yu},
  journal= {arXiv preprint arXiv:2310.17342},
  year   = {2023}
}