Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Learning to Retrieve: Dual-Level Long-Term Memory for Text-to-SQL Agents

About

Interactive text-to-SQL agents solve database tasks through multi-turn interactions involving schema exploration, query execution, feedback interpretation, and decision revision. Long-term memory helps agents reuse past experiences, but existing retrieval methods remain limited. Static methods rely on fixed similarity heuristics that do not optimize downstream utility, while dynamic methods often learn from sparse final outcomes and retrieve memories at a single decision horizon. This is insufficient when memory usefulness changes across interaction stages, since memories useful for initial planning may differ from those needed for local, state-conditioned execution. We propose MERIT, a dynamic multi-horizon memory retrieval framework. MERIT maintains episode-level memory for global strategic guidance and turn-level memory for local decision support. Both levels use learned retrieval policies optimized with reinforcement learning. To train turn-level retrieval despite limited intermediate supervision, MERIT uses a lightweight Process Reward Model to provide dense proxy rewards for local memory selection. Experiments on BIRD-Interact show that MERIT outperforms no-memory, static-retrieval, and dynamic-retrieval baselines in success rate while reducing average interaction turns. Transfer results on Spider2-Snow further show positive cross-benchmark transfer without benchmark-specific tuning. These results suggest that multi-horizon retrieval improves experience reuse in interactive text-to-SQL agents.

Yibo Wang, Nikki Lijing Kuang, Philip S. Yu, Zhewei Yao, Yuxiong He• 2026

Related benchmarks

TaskDatasetResultRank
Interactive Database InteractionBIRD-Interact Phase 2
Success Rate (WRITE)34.3
9
Interactive Database InteractionBIRD-Interact Phase 1
Write Success Rate53.49
9
Interactive Database InteractionBIRD Interact
Avg Turns (WRITE)7.09
8
Interactive Task SuccessSpider2-Snow
Success Rate69.29
8
ConversationalBIRD Interact
Task Success Rate22.7
4
Showing 5 of 5 rows

Other info

Follow for update