Databricks details Adaptive Instructed-Retriever for efficient AI search
Databricks details its Adaptive Instructed-Retriever, a model designed to optimize search for enterprise data agents by combining parallel and sequential search. It adaptively decides computation requirements for user requests, delivering frontier-quality retrieval with 2x lower latency than leading models for complex, multi-hop questions. The post explains the training methodology using online reinforcement learning and its applicability for customers developing specialized models. This approach allows for customizable quality-latency tradeoffs to fit various production workloads.
- →Adaptive Retrieval for Enterprise AI Agents
- →Achieving Frontier-Quality Search at 2x Lower Latency
- →Customizable Quality-Latency Tradeoffs with ORL Training
Notes (3) ›
- Adaptive Retrieval for Enterprise AI Agents
The post explains Adaptive Instructed-Retriever, designed for enterprise data agents, which combines parallel single-step retrieval and sequential search to balance accuracy and speed for complex, multi-hop questions.
- Achieving Frontier-Quality Search at 2x Lower Latency
The model showcases performance comparable to leading third-party and open-source models while achieving 2x lower latency, significantly improving retrieval quality over single-step search on benchmarks.
- Customizable Quality-Latency Tradeoffs with ORL Training
The article describes the training process using online reinforcement learning (ORL) with a reward design that balances trajectory quality and search cost, allowing for a customizable quality-latency tradeoff. This approach, trainable with AI Runtime (AIR), is practical for customers developing specialized models for their domains.
https://www.databricks.com/blog/adaptive-instructed-retriever-frontier-quality-search-2x-lower-latency
Related releases
- Consort: Open-Source Framework for Test-Driven Development on Branching Databases Databricks Blog ·
- Databricks Go SDK v0.178.0 Enhances ML Features and Adds AI/Sandbox APIs Databricks Go SDK Releases ·
- Databricks SDK for Java v0.154.0 Enhances Feature Engineering & ML Policy APIs Databricks Java SDK Releases ·
- Databricks Python SDK v0.137.0 Enhances Feature Engineering and ML Metadata APIs Databricks Python SDK Releases ·
- Zepto Scales AI Customer Support with Evaluation-First Agents on Databricks & MLflow Databricks Blog ·
- Databricks Lakebase Expands PCI-DSS and HITRUST Compliance to All AWS Regions Databricks Release Notes ·