DocETL: Agentic Query Rewriting and Evaluation for Complex Document Processing
Analyzing unstructured data has been a persistent challenge in data processing. Recent proposals offer declarative frameworks for LLM-powered processing of unstructured data, but they typically execute user-specified operations as-is in a single LLM call—focusing on cost rather than accuracy. This is problematic for complex tasks, where even well-prompted LLMs can miss relevant information. For instance, reliably extracting all instances of a specific clause from legal documents often requires decomposing the task, the data, or both. We present DocETL, a system that optimizes complex document processing pipelines, while accounting for LLM shortcomings. DocETL offers a declarative interface for users to deine such pipelines and uses an agent-based approach to automatically optimize them, leveraging novel agent-based rewrites (that we call rewrite directives ), as well as an optimization and evaluation framework. We introduce (i) logical rewriting of pipelines, tailored for LLM-based tasks, (ii) an agent-guided plan evaluation mechanism, and (iii) an optimization algorithm that efficiently finds promising plans, considering the latencies of LLM execution. Across four real-world document processing tasks, DocETL improves accuracy by 21–80% over strong baselines. DocETL is open-source at docetl.org and, as of March 2025, has over 1.7k GitHub stars across diverse domains.
Authors
- Aditya Parameswaran (ORCID: https://orcid.org/0000-0002-4538-4752)
- Shreya Shankar (ORCID: https://orcid.org/0000-0002-0919-9672)
- Eugene Wu (ORCID: https://orcid.org/0000-0003-4254-6688)
- T. Shah
- Tristan Chambers
Institutions
- Berkeley College (US)
- Police Department (LT)
- Columbia University (US)
- University of California, Berkeley (US)
Publication Details
- Journal
- Proceedings of the VLDB Endowment
- Published
- 2025-05-01
- DOI
- https://doi.org/10.14778/3746405.3746426
- Citations
- 6
- Primary Topic
- Data Stream Mining Techniques
- Type
- article
- Field-Weighted Citation Impact
- 14.89