跳到正文
@omarsar0· @omarsar0 · X·· 14 天前AI 评分44
AI 导读

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