AWS Finds Multi-Turn Training Improves Search Agent Retrieval on Three Tests
AWS Finds Multi-Turn Training Improves Search Agent Retrieval on Three Tests
AI Agents & Automation

AWS Finds Multi-Turn Training Improves Search Agent Retrieval on Three Tests

AWS Machine LearningOct 2, 2026

AWS says fine-tuning Qwen3.6-27B with SageMaker AI multi-turn reinforcement learning improved retrieval on three of four tests and sharply reduced some failures.

AWS published results on October 2, 2026, from fine-tuning a search agent with multi-turn reinforcement learning (MTRL) in Amazon SageMaker AI. The experiment used Qwen3.6-27B in an enterprise-search setup, where the agent could choose between keyword-based BM25 search and vector search, then refine its queries over multiple turns.

The training approach scores the outcome of a complete search sequence rather than judging each response in isolation. AWS used normalized discounted cumulative gain at rank 10 (nDCG@10) as its retrieval reward and penalized runs that exceeded turn or token limits. Training data came from six datasets, including FRAMES, BRIGHT, and an enterprise retrieval-augmented generation collection; four separate datasets were held out for testing.

In AWS’s evaluation, the fine-tuned model improved nDCG@10 on WixQA, Wands, and BrowseComp-Plus, while its score dipped slightly on FreshStack. The largest reported relative gains were 23.7% on BrowseComp-Plus and 18.4% on WixQA. On BrowseComp-Plus, the failure rate fell from 22.89% for the base model to 0.68% after fine-tuning. The WixQA failure rate also declined, from 0.67% to 0.17%; Wands recorded no failures for either version.

AWS says the training job changed three main settings from their defaults: one training epoch, a global batch size of 128, and rollout concurrency of 32. The post describes SageMaker’s MTRL service as serverless and priced per token, with support for inspecting training trajectories and evaluating models before deployment to SageMaker endpoints or Amazon Bedrock.

The results are AWS’s own experiment, not an independent comparison, and the FreshStack regression shows the gains were not uniform across tasks. The example nevertheless illustrates how teams might train a smaller, specialized agent against their own retrieval tools and task-specific outcome measures.

Industry Impact

More reliable enterprise search agents

Enterprise search and AI software

Overview

AWS reports that multi-turn reinforcement learning improved retrieval on three of four held-out tests for a search agent using keyword and vector search. For ML engineers and developers, the results offer a practical example of training agents against end-to-end search outcomes and task-specific tools. Product managers and business leaders should note that gains were not uniform: FreshStack declined slightly, and AWS's evaluation was not independent. Testing on an organization's own queries and failure modes remains essential before deployment.

Key Impact Points

The fine-tuned agent improved retrieval scores on WixQA, Wands, and BrowseComp-Plus, with the largest reported relative gain on BrowseComp-Plus.

BrowseComp-Plus failures fell from 22.89% to 0.68%, showing a potential reliability benefit alongside retrieval improvements.

FreshStack results dipped slightly, underscoring the need to validate performance across the specific tasks an organization cares about.

Opportunities

Task-specific search agent tuning

Enterprise search and AI software

Organizations with enterprise search tools could use multi-turn training to teach agents when to use keyword or vector retrieval and how to refine queries against their own outcome measures. AWS describes a serverless SageMaker AI service priced per token, with trajectory inspection and pre-deployment evaluation, which may lower operational barriers for teams already using that platform. The business case is better retrieval and fewer failures on targeted tasks, but the results are AWS's own experiment and include a regression on FreshStack. Teams should benchmark against representative internal queries and account for token costs before expanding deployment.

  • Tune agents for organization-specific search tasks

    Problem: General search agents may fail to retrieve relevant information reliably across specialized enterprise tasks.

    Solution: Fine-tune an agent through multi-turn reinforcement learning using organization-specific retrieval tools and outcome measures.

    Target Users: Enterprise teams building or operating search and retrieval-augmented generation systems.

    Business Value: Improve retrieval quality and reduce failures on validated tasks; results should be measured against representative internal tests.

    Revenue Potential: Potentially supports more effective enterprise search offerings, but the article provides no revenue estimates.

Reference link

https://aws.amazon.com/blogs/machine-learning/fine-tune-a-search-agent-with-multi-turn-rl-on-amazon-sagemaker-ai

Image Source Link

https://substackcdn.com/image/fetch/$s_!0hiB!,w_1200,h_675,c_fill,f_jpg,q_auto:good,fl_progressive:steep,g_auto/https%3A%2F%2Fsubstack-post-media.s3.amazonaws.com%2Fpublic%2Fimages%2F2388d978-afec-47c1-aea8-6a09234989b8_1376x768.jpeg

Community Discussion (0)

No discussions yet. Be the first to start the conversation!