DIN-SQL: Decomposed In-Context Learning of Text-to-SQL with Self-Correction

91 阅读1分钟

DIN-SQL: Decomposed In-Context Learning of Text-to-SQL with Self-Correction

分解上下文学习在nl2sql领域,带自我纠错

  • 链接:arxiv.org/pdf/2304.11…
  • 摘要:LLM生成sql的问题分解为子问题,划分为简单问题,一般问题、复杂问题等,针对性设计prompt策略;区分主键、外键

image.png

3. 错误分析

image.png 具体类型见论文第三章

4.4 纠错模块:

模型最后添加 Self-correction Module,用于异常错误兜底处理

  1. Generic self-correction prompt(gpt-4) , 把所有输入sql默认为 buggy sql

image.png

  1. Gentle self-correction prompt(codeX) , 给定规则,让模型自主判断

image.png