arrow
返回

LR-SQL: A Supervised Fine-Tuning Method for Text2SQL Tasks Under Low-Resource Scenarios

delete2025-09-01
delete0
delete
OA
AI
W
Wuzhenghong Wen
Y
Yongpan Zhang
S
Su Pan *
Y
Yuwei Sun
P
Pengwei Lu
C
Cheng Ding
DOI:10.3390/electronics14173489delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
在文本到SQL(Text2SQL)任务的监督微调(SFT)中,特别是对于包含大量表的数据库,编码模式特征需要过多的token,导致微调过程中GPU资源需求激增。为解决这一问题,我们提出了LR-SQL,一种通用的双模型SFT框架,包含模式链接模型和SQL生成模型。我们框架的核心是模式链接模型,它在一个新颖的下游任务上训练,该任务被称为基于切片的相关表过滤。该任务将数据库动态划分为可调整的表切片,并依次评估每个切片与输入查询的相关性,从而减少每轮迭代的token消耗。然而,切片分割会破坏数据库信息,损害模型理解完整数据库的能力。因此,我们在训练中引入了思维链(Chain of Thought, CoT),使模型能够从离散切片中重建完整的数据库上下文,进而提升推理保真度。最终,SQL生成模型利用模式链接模型的结果生成最终SQL。大量实验表明,我们提出的LR-SQL相比基线SFT方法将总GPU内存使用量降低了40%,模式链接任务的表预测准确率仅下降2%,整体Text2SQL执行准确率下降可忽略的0.6%。
Keyword:
Text2SQL
supervised fine-tuning
schema linking
database slicing
Chain of Thought

期刊

Electronics 封面图
Electronics
IF:
2.6
论文数:
1.1W
被引数:
4.7W

机构

N
nanjing university of posts and telecommunications
学者数:
3.8K
论文数: 1.6K
被引数: 0
引用论文

引用论文

err
IF0
err
err0
errOAAI
err
err分享
err收藏
Demonstration of DB-GPT: Next Generation Data Interaction System Empowered by Large Language ModelsDB-GPT演示:由大型语言模型赋能的下一代数据交互系统
err2024-11-08
err0
errOAAI
errSiqiao Xue; Danrui Qi; Caigao Jiang; Fangyin Cheng; Keting Chen; Zhiping Zhang; Hongyang Zhang; Ganglin Wei; Wang Zhao; Fan Zhou; Hong Yi; Shaodong Liu; Hongjun Yang; Faqiang Chen
err分享
err收藏
err
IF0
err
err0
errOAAI
err
err分享
err收藏
CodeS: Towards Building Open-source Language Models for Text-to-SQLCodeS: 构建用于文本到SQL的开源语言模型
err2024-05-30
err0
errOAAI
errHaoyang Li; Jing Zhang; Hanbing Liu; Ju Fan; Xiaokang Zhang; Jun Zhu; Renjie Wei; Hongyan Pan; Cuiping Li; Hong Chen
err分享
err收藏
学者 查看更多内容