Google 等提出 DualSQL,用多智能体 RL 将 Text-to-SQL 拆成两个共享同一模型权重的智能体,分别负责定位表和列、生成 SQL,推理中可通过三种工具查询数据库。
Great paper from Google and colleagues.
Trains Text-to-SQL agents using multi-agent RL.
(bookmark it)
This work proposes DualSQL, which splits Text-to-SQL into two agents, one that links the question to the right tables and columns and one that writes the SQL.
Both agents run on the same model weights, so a single multi-agent RL run trains both roles together. The agents can query the database through three tools while they reason.
Multi-agent RL tends to collapse during training, so the authors add guardrails on rollouts and a new reward, robust execution match, that judges SQL correctness more accurately.
Trained on only 3,755 examples, DualSQL-4B reaches 68.0% execution accuracy on BIRD dev, matching earlier 7B models. DualSQL-8B reaches 71.1%, ahead of previous single-model systems with 32B parameters.
Paper: https://t.co/C5E07c5Uzz
Chat with Paper: https://t.co/HIzPmIiOVq
来源:@omarsar0 · x.com