我要投稿

RB-SQL：利用检索LLM框架处理大型数据库和复杂多表查询的NL2SQL

发布日期：2024-07-19 21:51:09 浏览次数： 3173

NL2SQL的任务是将自然语言问题转换为SQL查询，以便从数据库中获取答案。现有LLM来指导SQL生成的方法在处理大型数据库和复杂多表查询时存在挑战，尤其是在处理冗余信息和提高提示工程效率方面。

(a) 利用大型语言模型（LLM）解决文本到SQL任务的一个示例。(b) DPR模型和提出的RB模型的图表。与DPR模型相比，RB模型将输入从文档扩展到其他数据类型（即表格、列、SQL框架）。

为了解决上述问题，提出了RB-SQL框架，包含三个模块：

Table-Retriever：检索与问题最相关的表格。
Column-Retriever：在检索到的表格中进一步检索相关列。
SQL-Skeleton-Retriever：搜索具有相似SQL骨架的少量示例，并将SQL骨架引入示例组织，以增强上下文学习过程。

RB-SQL框架。表格检索器从数据库中筛选表格，列检索器进一步筛选列。SQL框架检索器用于选择相似的少量示例，并将SQL框架添加到示例组织中。

RB-SQL框架利用密集型段落检索（Dense Passage Retrieval, DPR）模型来检索相关表格、列和示例，以构建有效的提示工程。此外，框架还引入了SQL骨架作为示例组织中的中间步骤，以指导正确的SQL生成过程。

(a) 表格检索器的工作流程。该模块计算问题与表格之间的相似度，并检索与问题高度相关的表格。(b) 表格检索器的框架。使用BERT分别对问题和表格进行编码，并使用基于MaxSim的后期交互来计算相似度得分。

(a) 列检索器的工作流程。该模块检索与问题高度相关的列。(b) 列检索器的框架。

(a) SQL骨架检索器的工作流程。 (b) SQL骨架检索器的框架。

通过在公共数据集BIRD和Spider上的实验，结果表明RB-SQL模型在性能上优于几个竞争基线：GPT-4、DIN-SQL、DAIL-SQL

还进行了消融研究，证明了RB-SQL框架中所有模块对于性能提升都起到了重要作用。

https://arxiv.org/pdf/2407.08273RB-SQL: A Retrieval-based LLM Framework for Text-to-SQLprompt：https://anonymous.4open.science/r/Anonymize-A5E7/prompt_case.txt

相关资讯

2026-06-12

卡内基梅隆最新AI应用成熟度模型：AI转型最后拼的是组织基本功（附8项能力）

2026-06-11

企业AI场景全景拆解——100+落地场景，按业务域一网打尽

2026-06-11

咨询｜FDE 为什么突然火了？到底是咨询顾问、还是AI工程师更适合做FDE呢？

2026-06-10

让「准确率」可裁判：AI 数据分析需要一套可信机制

2026-06-10

DataBuddy 庖丁解牛（系列1）：腾讯刚刚押下的「数据 Agent 全栈」

2026-06-09

三部门发文19个场景，但90%企业AI工作台还在"拼积木"（AI工作台①）

2026-06-05

为什么企业内部AI应用看起来厉害,用起来是垃圾?

2026-06-03

超级个体时代｜腾讯研究院3万字报告

联系获取

160+中大型企业正在使用53AI

立即咨询预约演示

把握AI发展的机遇，共同探索、共同进步

2025-01-22

如何打造基于GenAI的员工服务机器人

2025-01-22

热点资讯

钉钉已经彻底 CLI 化了。

2026-03-19

超级个体时代｜腾讯研究院3万字报告

2026-06-03

TSMC（台积电）：从「制造即代码」到 AI 时代的绝对物理基岩

2026-03-23

AI在企业落地的真实困境：小场景看不上，大项目做不起

2026-03-26

企业 AI 转型为什么需要 FDE 模式

2026-05-13

OpenClaw威胁下，大厂APP会被降维成信息通道么？

2026-03-21

Oracle裁员三万人的警钟：当AI成为新的生产要素，管理者需要思考的三个核心问题

2026-04-09

浏览器自动化：从GUI到OpenCLI

2026-04-14

业界首发：HENGSHI CLI正式发布，开启Agentic BI自动驾驶时代

2026-04-01

从“无所不知”到“无所不能”，企业龙虾将怎样接管企业 IT

2026-04-16

大家都在问

咨询｜FDE 为什么突然火了？到底是咨询顾问、还是AI工程师更适合做FDE呢？

2026-06-11

为什么企业内部AI应用看起来厉害,用起来是垃圾?

2026-06-05

埃森哲押注 FDE：咨询公司的宿命，它真的能破吗？

2026-06-02

FDE越来越火，你认为这会是2026年AI落地之道吗？

2026-05-26

OpenClaw威胁下，大厂APP会被降维成信息通道么？

2026-03-21

为什么越来越多的软件被“用完即弃”？

2026-02-11

老登软件公司的AI路怎么走？

2026-01-21

都知道 AI 会重构组织，但为什么迈不出第一步？以及第一步从哪开始？

2025-12-26

热门标签

内容创作大模型技术个人提效 langchain llamaindex 多模态技术 RAG技术智能客服知识图谱模型微调 RAGFlow coze Dify Fastgpt Bisheng Qanything AI+汽车 AI+金融 AI+工业 AI+培训 AI+SaaS Skill 提示词技巧 AI+电商 AI面试数字员工 ChatBI AI知识库开源大模型智能营销智能硬件 FDE AI+医疗 MaxKB Palantir Glean Openclaw