摘要
大型语言模型(LLMs)在文本到SQL任务中已成为强大的工具,展现出卓越的推理能力。与数学应用题和常识推理等任务不同,SQL解决方案具有相对固定的模式。这有助于探讨LLMs是否能从分类思维中受益,类似于人类通过基于相似例子的归纳推理来获取知识的方式。在本研究中,我们提出,采用查询组分区的方法使LLMs能够专注于学习单一问题类型的思维过程,从而在不同难度级别和问题类别中提升其推理能力。我们的实验表明,配备PTD-SQL的多个高级LLMs在Spider和BIRD数据集上能够超越或匹配先前的最先进(SOTA)方法。有趣的是,不同初始性能的模型在经过针对性训练后,其能力边界均显示出显著提升,这与人类进步有相似之处。代码可在https://github.com/lrlbbzl/PTD-SQL获取。