Getty Images

Databricks launches model to boost data retrieval for AI

Adding sequential search to improve the speed and accuracy of responses to complex queries evolves existing capabilities and keeps Databricks in step with competitors.

Databricks is once again aiming to improve the accuracy of data retrieval systems for AI.

In January, Databricks launched Instructed Retriever, an alternative to the traditional retrieval-augmented generation (RAG) pipelines that many enterprises deploy to try to find and feed data to AI agents.

With AI projects failing far more than they succeeded heading into 2026, RAG, which searches for data based on users' queries, showed it was not good enough on its own at providing agents with relevant data. Databricks designed Instructed Retriever, powered by the Instructed-Retriever-1 model, to discover more contextually appropriate data by augmenting user queries with additional parameters to make data retrieval more precise.

To further improve data retrieval for AI, Databricks is introducing Adaptive Instructed-Retriever, a new retrieval model specifically designed for complex searches.

Instructed Retriever, which deploys parallel single-step retrieval capabilities to simultaneously search multiple parameters, works well for one-step queries but does not perform as fast or accurately when responses require multi-step searches. Adaptive Instructed-Retriever adds sequential search to parallel single-step retrieval to enable multi-step queries without adding latency.

"Adaptive Instructed-Retriever is a significant upgrade because it enables agents to efficiently locate relevant … enterprise assets across large workspaces," William McKnight, president of McKnight Consulting, told TechTarget. "Its key advantage is it can use fast, single-step retrieval for straightforward requests while escalating to multi-step sequential search for more complex ones."

Maintained speed, more relevant search results and lower costs could all result from using the new model, he continued.

"Unlike queries that are focused due to a single-source -- like a data lake query -- agent data retrieval is harder because it can be for anything … and the ways of ensuring which is the 'right data' for a particular agent task are nascent," McKnight said. "Adaptive retrieval addresses this by matching retrieval depth to query complexity to improve the accuracy, speed and cost of agent-driven data discovery."

Improving accuracy

When Databricks launched Instructed Retriever, it benchmarked the tool against traditional RAG pipelines for AI to demonstrate its proficiency. Adaptive Instructed-Retriever was similarly measured against competing capabilities to show its effectiveness.

Adaptive Instructed-Retriever is a significant upgrade because it enables agents to efficiently locate relevant … enterprise assets across large workspaces.
William McKnightPresident, McKnight Consulting

In tests conducted by Databricks -- third-party evidence was not provided by the vendor -- the new model's recall capabilities were in line with those of models from Anthropic, DeepSeek and OpenAI. However, Adaptive Instructed-Retriever executed its searches significantly faster.

"This release is incremental for users but significant for the product," Donald Farmer, founder and principal of TreeHive Strategy, told TechTarget. "Adaptive Instructed-Retriever adds multi-step search for questions that the previous Instructed Retriever would get wrong, so a user gets better answers with a fixed latency budget at low cost."

Farmer noted that enterprises continue to struggle to discover and operationalize the right data for agents because finding information across disparate systems is difficult, especially when query instructions can differ from user to user. As a result, agile searches rather than rigid ones are beneficial.

"In the enterprise, retrieval is not one problem," Farmer said. "Data is spread over tables, notebooks, dashboards and documents, and getting the right answer often depends on instructions that require interpretation. Plain similarity search can't help."

Differentiating with governance

Although beneficial for users and an important advancement of Databricks' data retrieval capabilities for AI, Adaptive Instructed-Retriever itself is not a competitive differentiator for the vendor, according to McKnight.

For example, Snowflake similarly offers a model that delivers text-to-SQL and semantic-based retrieval across widespread enterprise data estates. However, what will appeal to Databricks' users is Adaptive Instructed-Retriever synchronization with Unity Catalog, the vendor's centralized governance layer.

"Databricks differentiates [by anchoring] dynamic search, predicate extraction and governance directly into Unity Catalog, which serves as the optimized backbone for its conversational data tools, including Databricks Genie, Databricks Assistant and the Mosaic AI Agent Framework," McKnight said.

Farmer likewise noted that Databricks is not the first vendor to provide a small model specialized for data retrieval. Google Cloud and Microsoft -- among others -- offer tools that plan and run parallel subqueries to discover and operationalize relevant data.

Databricks, however, goes further with Adaptive Instructed-Retriever by training the model to know when to stop searching, according to Farmer.

"When an AI agent searches for data, it can take one look and answer, or it can keep looking," he said. "More looks usually give a better answer, but each one takes time and costs money. The hard part is knowing when to stop. Databricks trained a small model to make the stopping decision itself."

In addition, Farmer posited that the savings which the new model could lead to might force competing vendors to respond.

"Customers do not choose a data platform for a retrieval model, and the competitors' own retrieval layers are probably good enough, but the argument for cost-control is compelling," he said. "So, every vendor will need a cheap, fast retrieval layer for agents. But they'll do it their own way, I expect."

Next steps

Databricks developed Adaptive Instructed-Retriever after observing customers that have begun to put agentic AI tools into production ask increasingly intricate questions of those agents, according to Michael Bendersky, the vendor's director of research.

"Agentic search requires multiple rounds of retrieval using tools, reviewing the information and then searching again," he told TechTarget. "In these environments, the agents search, retrieve, and then examine again until they find the right answer. At the same time, you don't want the user to wait longer than necessary for the system to finish the search."

Over the next few months, Databricks will add serverless GPU compute capabilities -- now in public preview -- to Adaptive Instructed-Retriever to optimize the model's usage by automatically scaling up when needed and scaling down when workload demands decrease, Bendersky continued. In addition, he said that Databricks plans to continue to reduce barriers enterprises face when moving AI into production by addressing context for agents, AI governance, cost control and the openness of its platform.

Continuing to address the quality and performance of existing capabilities to enable customers to build and deploy reliable AI capabilities is a wise strategy for Databricks, according to McKnight. Specifically, improving the accuracy rate of its model serving capabilities and simplifying the use of its platform would be beneficial.

"To support enterprise-grade AI agents, Databricks must resolve its real-time concurrency bottlenecks and high model serving failure rates to ensure stable, low-latency agent execution loops," McKnight said. "Additionally, they need to simplify the overall user experience by shifting toward a zero-administration serverless platform and offering prebuilt operational templates … to accelerate business ROI."

Farmer, meanwhile, suggested that Databricks further improve its data retrieval capabilities by better enabling them to choose which tables, metrics and semantic definitions to use when searching for which assets to feed AI tools.

"The hard problem for data agents is not finding documents, but choosing among thousands of tables, metrics and semantic definitions," he said. "I think that's a future demand [from customers]."

Eric Avidon is a senior news writer for Informa TechTarget and a journalist with more than three decades of experience. He covers analytics and data management.

Dig Deeper on Data Management