arrow
返回

QA-SQL: query-augmented SQL generation using few-shot prompting with data augmentation

delete2026-02-11
delete0
PRE
AI
S
Sokheang Chan
V
Vungsovanreach Kong
A
Anand Nayyar
T
Tae-Kyung Kim *
DOI:10.7717/peerj-cs.3525delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
自然语言到结构化查询语言(NL-to-SQL)模型通过将普通语言查询转换为结构化查询语言(SQL)命令,使从结构化数据库中检索数据变得更加容易。然而,当前方法在处理复杂数据库结构、用户问题不明确以及生成高效SQL查询方面常常遇到困难。这些问题使得它们难以在实际大规模数据库中有效应用。为解决这些问题,提出了一种名为查询增强SQL生成(QA-SQL)的新型框架,该框架结合了检索增强生成(RAG)和数据增强等先进技术,以提高自然语言到SQL(NL-to-SQL)生成的准确性和效率。所提出的方法利用模式链接、RAG、动态提示工程以及少样本学习和向量数据库检索,通过查询增强来提升SQL查询生成。为测试和验证所提出的模型,在BIg Bench for LaRge-scale Database Grounded Text-to-SQL Evaluation(BIRD)基准上进行了大量实验,该基准包含1,534个自然语言查询和多种关系数据库,结果显示执行准确率达到46.68%,有效效率得分为87.66%,优于现有方法。该研究提出了一种上下文感知检索系统,显著提高了SQL的准确性,为更可扩展和可靠的NL-to-SQL解决方案铺平了道路。
Keyword:
Text-to-SQL
Retrieval-augmented generation
Data augmentation
Large language models

期刊

PeerJ Computer Science 封面图
PeerJ Computer Science
IF:
2.5
论文数:
3.4K
被引数:
6.9K

机构

D
Duy Tan University
学者数:
495
论文数: 355
被引数: 6.6K
C
chungbuk national university
学者数:
1.7K
论文数: 729
被引数: 0
引用论文

引用论文

err
IF0
err
err0
errOAAI
err
err分享
err收藏
err
IF0
err
err0
errOAAI
err
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
学者 查看更多内容