AlayaDB: The Data Foundation for Efficient and Effective Long-context LLM Inference
This addresses efficiency challenges for Model as a Service providers, though it appears incremental as an optimization of existing methods.
The paper tackles the problem of inefficient long-context inference for Large Language Models (LLMs) by introducing AlayaDB, a vector database system that decouples KV cache and attention computation, resulting in fewer hardware resources and higher generation quality compared to existing solutions.
AlayaDB is a cutting-edge vector database system natively architected for efficient and effective long-context inference for Large Language Models (LLMs) at AlayaDB AI. Specifically, it decouples the KV cache and attention computation from the LLM inference systems, and encapsulates them into a novel vector database system. For the Model as a Service providers (MaaS), AlayaDB consumes fewer hardware resources and offers higher generation quality for various workloads with different kinds of Service Level Objectives (SLOs), when comparing with the existing alternative solutions (e.g., KV cache disaggregation, retrieval-based sparse attention). The crux of AlayaDB is that it abstracts the attention computation and cache management for LLM inference into a query processing procedure, and optimizes the performance via a native query optimizer. In this work, we demonstrate the effectiveness of AlayaDB via (i) three use cases from our industry partners, and (ii) extensive experimental results on LLM inference benchmarks.