Prince Larbi Ampofo

DBRAG框架增强了LLM跨多表进行数据库查询的能力

研究人员开发了DBRAG，一个新颖的检索增强生成框架，旨在改进大型语言模型（LLM）处理跨多个数据库表进行复杂查询的能力。DBRAG首先使用离线索引识别相关表，然后用特定于查询的行来精炼表的摘要，并采用LLM进行候选重排序。最后，一个程序辅助推理器选择必要的表并执行操作，同时保持简洁的提示上下文。在Spider、GeoQuery和ATIS数据集上的实验表明，在表检索和多表问答方面均取得了性能提升。