Abacus: A Cost-Based Optimizer for Semantic Operator Systems
This addresses the challenge of optimizing AI-driven document processing systems for developers, offering significant improvements in performance and efficiency, though it is incremental as it builds on existing semantic operator frameworks.
The paper tackles the problem of optimizing semantic operator systems for LLM-powered data processing by introducing Abacus, a cost-based optimizer that searches for the best implementations, resulting in systems with 18.7%-39.2% better quality and up to 23.6x lower cost and 4.2x lower latency compared to alternatives.
LLMs enable an exciting new class of data processing applications over large collections of unstructured documents. Several new programming frameworks have enabled developers to build these applications by composing them out of semantic operators: a declarative set of AI-powered data transformations with natural language specifications. These include LLM-powered maps, filters, joins, etc. used for document processing tasks such as information extraction, summarization, and more. While systems of semantic operators have achieved strong performance on benchmarks, they can be difficult to optimize. An optimizer for this setting must determine how to physically implement each semantic operator in a way that optimizes the system globally. Existing optimizers are limited in the number of optimizations they can apply, and most (if not all) cannot optimize system quality, cost, or latency subject to constraint(s) on the other dimensions. In this paper we present Abacus, an extensible, cost-based optimizer which searches for the best implementation of a semantic operator system given a (possibly constrained) optimization objective. Abacus estimates operator performance by leveraging a minimal set of validation examples and, if available, prior beliefs about operator performance. We evaluate Abacus on document processing workloads in the biomedical and legal domains (BioDEX; CUAD) and multi-modal question answering (MMQA). We demonstrate that systems optimized by Abacus achieve 18.7%-39.2% better quality and up to 23.6x lower cost and 4.2x lower latency than the next best system.