论文

MDB-Link:多数据库文本到 SQL 的分层架构链接

MDB-Link: Hierarchical Schema Linking for Multi-Database Text-to-SQL

上下文与知识检索增强

摘要

传统的文本到 SQL 研究和基准测试假设一个已知的目标数据库,忽略了必须在大型异构数据库集合中路由查询的设置。因此,我们研究多数据库设置中的模式链接,其中系统必须首先定位目标数据库,然后构建一个紧凑的、与 SQL 相关的模式以进行生成。我们提出了 MDB-Link,这是一种分层模式链接框架,它从全局索引中检索与问题相关的列,将检索证据聚合到候选数据库中,并使用预算感知 大语言模型 (LLM) 进行数据库重新排名、表选择和列基础。使用 Qwen2.5-14B,MDB-Link 在数据库本地化和列选择方面优于 MMQA、Spider2-Snow 和 BIRD-dev 上的 LinkAlign,同时生成大小接近参考模式的模式子集。 MMQA 上的精确匹配从 16.88 提高到 51.41,Spider2-Snow 上的精确匹配从 2.50 提高到 9.17,BIRD-dev 上的精确匹配从 12.52 提高到 38.01。 MDB-Link 的运行速度也比 LinkAlign 和 AutoLink 更快,这证明了分层架构缩减对于下游 SQL 生成的有效性。