Baikal: Structured Search for Deep Research over Data Lakes
Dhruv Agarwal, Rishitha Guttapalle Mohan, Aarti Kumari, Ashi Sinha, Athulya Anil, Kavitha Srinivas, Horst Samulowitz, Andrew McCallum
cs.AI, cs.CL, cs.LG
Submitted: 2026-07-30
Code: https://github.com/dhdhagar/baikal
License: http://creativecommons.org/licenses/by/4.0/
The gist: Deep research over data lakes requires an LLM agent to investigate evidence across thousands of heterogeneous tables and passages to synthesize a report.
Terminology
Abstract
Deep research over data lakes requires an LLM agent to investigate evidence across thousands of heterogeneous tables and passages to synthesize a report. Existing methods perform iterative retrieval and generation, letting accumulated context determine what to investigate next, which can overexploit locally promising evidence and fail to cover distinct semantic regions under a fixed budget. To address this, we cast deep research over data lakes as a budgeted search problem and present Baikal - a framework that clusters heterogeneous evidence into semantic regions, then searches over them adaptively to balance exploration and exploitation. Within each selected region, Baikal generates and investigates region-grounded subquestions, using finding quality as rewards to update region-level value estimates and guide search under policies ranging from random and LLM-guided selection to Bayesian epsilon-greedy and UCB. We evaluate Baikal on 15 queries each over HybridQA and TAT-QA data lakes containing 10,993 and 2,757 tables, respectively, together with 227K Wikipedia passages and 13K financial report passages. We assess research quality with a new rubric covering groundedness, relevance, diversity, and utility, and use GPT-5-mini to score Baikal and strong baselines, including DeepSearcher and an OpenCode research agent with retrieval and clustering variants. Across both data lakes, Baikal performs strongly under several region-selection policies; its best configuration improves report scores over the strongest baselines by 28% on HybridQA and 36% on TAT-QA. Our analyses attribute these gains to organizing and exploring semantic evidence regions, which improves groundedness and diversity and yields more useful findings under the same subquestion budget. These results demonstrate the value of structured semantic exploration for systematic research and discovery over heterogeneous data lakes.
Sources
- Evidence-Informed LLM Beliefs for Continual Scientific Discovery
- Bring Your Own KG: Self-Supervised Program Synthesis for Zero-Shot KGQA
- Emergent autonomous scientific research capabilities of large language models
- Bridging Business Intent and Data: A Benchmark for Automatic Relational Data Product Generation
- Accumulating Context Changes the Beliefs of Language Models
- Accelerating scientific discovery with Co-Scientist
- BERTopic: Neural topic modeling with a class-based TF-IDF procedure
- CodeScientist: End-to-End Semi-Automated Scientific Discovery with Code-based Experimentation
- Characterizing Deep Research: A Benchmark and Formal Definition
- AIDE: AI-Driven Exploration in the Space of Code
- Tree Search for Language Model Agents
- WebThinker: Empowering Large Reasoning Models with Deep Research Capability
- The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery
- UMAP: Uniform Manifold Approximation and Projection for Dimension Reduction
- Kosmos: An AI Scientist for Autonomous Discovery
- AlphaEvolve: A coding agent for scientific and algorithmic discovery
- MiGrATe: Mixed-Policy GRPO for Adaptation at Test-Time
- Language agents achieve superhuman synthesis of scientific knowledge
- SQL-PaLM: Improved Large Language Model Adaptation for Text-to-SQL (extended)
- The AI Scientist-v2: Workshop-Level Automated Scientific Discovery via Agentic Tree Search
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems