基于依存关系图注意力网络的SQL生成方法.pdfVIP

第58卷第5期浙江大学学报(工学版)Vol.58No.5

2024年5月JournalofZhejiangUniversity(EngineeringScience)May2024

DOI:10.3785/j.issn.1008-973X.2024.05.004

基于依存关系图注意力网络的SQL生成方法

1,2111111

舒晴，刘喜平，谭钊，李希，万常选，刘德喜，廖国琼

(1.江西财经大学信息管理学院，江西南昌330013；2.江西农业大学软件学院，江西南昌330013)

摘要：研究基于自然语言问题的结构化查询语言(SQL)生成问题(Text-to-SQL).提出两阶段框架，旨在解耦模式

链接和SQL生成过程，降低SQL生成的难度.第1阶段通过基于关系图注意力网络的模式链接器识别问题中提

及的数据库表、列和值，利用问题的语法结构和数据库模式项之间的内部关系，指导模型学习问题与数据库的对

齐关系.构建问题图时，针对Text-to-SQL任务的特点，在原始句法依存树的基础上，合并与模式链接无关的关系，

添加并列结构中的从属词与句中其他成分间的依存关系，帮助模型捕获长距离依赖关系.第2阶段进行SQL生

成，将对齐信息注入T5的编码器，对T5进行微调.在Spider、Spider-DK和Spider-Syn数据集上进行实验，实验结

果显示，该方法具有良好的性能，尤其是对中等难度以上的Text-to-SQL问题具有良好的表现.

关键词：Text-to-SQL；自然语言查询；依存句法分析；关系图注意力网络

中图分类号：TP393文献标志码：A文章编号：1008−973X（2024）05−0908−10

SQLgenerationmethodbasedondependencyrelationalgraph

attentionnetwork

1,21111

SHUQing,LIUXiping,TANZhao,LIXi,WANChangxuan,

11

LIUDexi,LIAOGuoqiong

(1.SchoolofInformationTechnology,JiangxiUniversityofFinanceandEconomics,Nanchang330013,China;

2.SchoolofSoftware,JiangxiAgriculturalUniversity,Nanchang330013,China)

Abstract:Theproblemofgeneratingstructuredque