arXiv cs.CL· Haolin Yang, Jipeng Zhang, Zhiqiang Qian, Jian Xie, Shuaishuai Gong, Yike Guo, Sirui Han·· 9 天前AI 评分34
HarnessSQL:面向真实数据库环境的 SQL 智能体 harness 原生训练框架
HarnessSQL: Harness-Native Training for SQL Agents in Realistic Database Environments
AI 导读
HarnessSQL 是一个 harness 原生后训练框架,在监督微调与强化学习中保留完整多轮交互结构,让模型直接在目标 SQL harness 内训练。
正文 · AI 翻译
摘要:Text-to-SQL 模型通常被训练为将问题直接映射到静态查询,而现实世界中的数据库智能体则通过与实时数据库进行有状态的、多轮交互来运作——检查模式、执行探测查询、诊断错误并修正假设。这造成了一个关键的训练-部署不匹配,因为中介这种交互的执行框架仅在推理时才被引入。为弥合这一差距,我们提出了 HarnessSQL,一个以框架为本的后训练框架,在监督微调和强化学习过程中都保留了完整的交互结构。HarnessSQL 构建了隔离的、可执行的数据库环境,并配以隐藏的执行预言机,直接在目标 SQL 框架内展开教师模型,仅保留经过验证的轨迹用于全序列 SFT,随后进行执行奖励 RL。在 Spider 2.0-SQLite 上,HarnessSQL 大幅提升了紧凑模型的执行准确率,将 Qwen3-8B 从 15.5% 提升至 45.2%,将 Qwen3-14B 从 22.2% 提升至 54.8%,同时有效迁移到 BIRD-Interact 和 LiveSQLBench 等分布外交互式基准。我们的发现表明,直接在执行框架内训练数据库智能体对于掌握复杂的、长时程的数据库工作流至关重要。
| 学科: | 计算与语言(cs.CL) |
| 引用为: | arXiv:2610.12274 [cs.CL] |
| (或 arXiv:2610.12274v1 [cs.CL] 用于此版本) | |
| https://doi.org/10.48550/arXiv.2610.12274 由 DataCite 颁发的 arXiv DOI(待注册) |
提交历史
来自:Haolin Yang [查看邮箱]
[v1]
2026年10月8日星期四 16:36:39 UTC(487 KB)
来源:arXiv cs.CL · arxiv.org