国防科技大学学报2024，Vol.46Issue(2)：197-204,8.DOI:10.11887/j.cn.202402020
辅助任务增强的中文跨域NL2SQL算法
Chinese cross-domain NL2SQL algorithm enhanced by auxiliary task
摘要
Abstract
NL2SQL(natural language to structured query language)task aims to translate natural language queries into SQL(structured query language)executable by the database.A Chinese cross-domain NL2SQL algorithm enhanced by auxiliary tasks was proposed.Core idea was to perform multi-task training and improve the accuracy of the model by adding auxiliary tasks in the decoder and combining the prototype model.Auxiliary task was designed by modeling the database schema into a graph,predicting the dependency relations between the natural language queries and the nodes in the database schema graph,and explicitly modeling the dependency relations between the natural language query and the database schema.Through the improvement of auxiliary tasks,the model can better identify which tables/columns in the database schema are more effective for predicting the target SQL for specific natural language queries.Experimental results on the Chinese NL2SQL dataset DuSQL show that the algorithm after adding auxiliary tasks has achieved better results than the prototype model,and can better handle cross-domain NL2SQL task.关键词
人工智能/深度学习/自然语言处理/语义解析Key words
artificial intelligence/deep learning/natural language processing/semantic parsing分类
信息技术与安全科学引用本文复制引用
胡亚红,刘亚冬,朱正东,刘鹏杰..辅助任务增强的中文跨域NL2SQL算法[J].国防科技大学学报,2024,46(2):197-204,8.基金项目
国家重点研发计划资助项目(2018YFB0204003,2018YFB0204004) （2018YFB0204003,2018YFB0204004）
