Data Flywheels for Cost-Effective AI Agent Optimization
nvidia · Tech · year n/a · source Agentic: yes · promotional bias: med · source-backed · extracted with prompt 0.5.6 (schema v2)
Generated from extraction.json — do not edit by hand. Every claim cites verbatim source evidence; stated means the quote directly asserts the claim, implied means it supports it indirectly.
Problem
Domain: Enterprise AI agent optimization
NVIDIA needed its internal NV Info Agent router to keep routing employee questions to the right expert agent as data, user preferences, and costs changed, without relying forever on a high-latency 70B model.
Constraints:
- Enterprise data and user preferences change over time.
- Large models increase inference cost as usage grows.
- The router must send queries to the correct expert agent.
- Smaller models started with much lower routing accuracy.
- Production logs and user feedback need human/SME curation before becoming ground truth.
Architecture
NVIDIA built a data flywheel around NV Info Agent. Employee queries pass through guardrails into a router LLM that selects an expert agent, each backed by RAG. Production inference logs and user feedback are curated with human/SME review into ground truth, evaluated with NeMo Evaluator and LLM-as-judge, used to fine-tune smaller models through NeMo Customizer, and then promoted when accuracy, latency, and cost meet the target.
Diagram confidence: explicit (topology described by the source).
flowchart LR employee-user["Employee user"] --> guardrails["Guardrails"] guardrails --> router-agent["Router agent"] router-agent --> expert-agents["Expert agents"] expert-agents --> rag-pipelines["RAG pipelines"] router-agent --> production-logs["Production logs"] production-logs --> ground-truth-curation["Ground truth curation"] ground-truth-curation --> model-evaluation["Model evaluation"] model-evaluation --> fine-tuning-pipeline["Fine-tuning pipeline"] fine-tuning-pipeline --> promoted-model["Promoted model"] promoted-model --> router-agent
Components
| Component | Role | Interacts with |
|---|---|---|
Employee user |
Submits internal support questions to NV Info Agent. | Guardrails |
Guardrails |
Protects the interaction for safety, security, and privacy. | Employee user, Router agent |
Router agent |
LLM router that classifies intent/context and selects the correct expert agent. | Guardrails, Expert agents, Production logs, Promoted model |
Expert agents |
Domain-specific agents for HR, finance, IT, product docs, and other employee knowledge domains. | Router agent, RAG pipelines |
RAG pipelines |
Retrieve relevant enterprise information for the chosen expert agent. | Expert agents |
Production logs |
Inference logs and feedback data captured while the router runs. | Router agent, Ground truth curation |
Ground truth curation |
Human/SME process that labels failures and creates training/evaluation examples. | Production logs, Model evaluation |
Model evaluation |
NeMo Evaluator and LLM-as-judge process for comparing candidate models. | Ground truth curation, Fine-tuning pipeline |
Fine-tuning pipeline |
NeMo Customizer process that adapts smaller models for routing. | Model evaluation, Promoted model |
Promoted model |
Chosen smaller or larger model deployed to power the router. | Router agent |
Data flow
An employee submits a query. Guardrails protect the interaction, the router agent chooses an expert agent, and the expert retrieves enterprise knowledge. Feedback and production logs feed a flywheel: curate ground truth, analyze failures, evaluate models, fine-tune smaller variants, compare routing accuracy/latency/cost, and promote the best model as the router.
Why this architecture
The flywheel exists to convert production feedback into smaller, cheaper models that retain large-model routing accuracy. NVIDIA uses curation, evaluation, fine-tuning, and guardrails as an end-to-end loop rather than treating model choice as a one-time selection.
Models used: Llama 3.1 70B · Llama 3.3 70B · 8B model variant · 3B model variant · 1B model variant · LLM-as-a-judge via NeMo Evaluator
Scale
- The NV Info Agent covers internal knowledge across HR benefits, financial earnings, IT help, product documentation, and other employee needs.“agent can help answer queries across a variety of domains from HR benefits, financial earnings, IT help, product documentation or anything everything that an internal employee needs access to.”transcript, stated
- The feedback pass produced 1,224 responses, including 729 satisfactory and 495 unsatisfactory responses.“we were able to curate around,224 data points. 729 of them were incorrect sorry satisfactory responses and 495 of them were unsatisfactory responses.”transcript, stated
- The final ground-truth set contained 685 data points.“we figured out and set up a ground trth data point data set of 8 68 85 data points and we split them into 6040”transcript, stated
- The 70B baseline reached 96% routing accuracy with 26-second first-token latency.“the 70B variant like we saw provided us gave us an accuracy of 96% uh for accurately routing it to the right export agent but it came with an latency of 26 seconds”transcript, stated
- A 1B model reached 94% accuracy with 98% lower inference cost and 70x lower latency.“by deploying let's say a 1B model, you're looking at a 98% savings in terms of lower inference cost and also 10x 70x model size reduction with 70x lower latency.”transcript, stated
Alternatives considered (roads not taken)
Use the 70B router because its baseline accuracy is high
Rejected because: It had high latency and cost compared with smaller models after fine-tuning.
Deploy smaller models without customization
Rejected because: The 8B baseline was below 14% routing accuracy before fine-tuning.
Patterns
multi-agent-orchestration (architecture)
A router agent chooses among multiple expert agents for employee support domains.
data-quality-loop (data)
Production feedback and logs are curated into ground truth for recurring model improvement.
fine-tuning-adaptation (modeling)
Smaller models are fine-tuned on curated router data to approach 70B accuracy.
model-routing (orchestration)
The central router model selects the expert agent for each query.
cost-aware-routing (cost)
Model selection explicitly trades routing accuracy against inference cost and latency.
Tradeoffs
Use a 1B model in some cases
Chose 94% accuracy with much lower cost and latency over maximum 96% routing accuracy from the 70B model.
Rationale: Some missions can accept a small accuracy gap in exchange for 98% lower inference cost and much lower latency.
Use employee feedback as training signal
Chose curated production feedback and logs over static benchmark-only model selection.
Rationale: The production router needs to adapt to real employee queries and failure modes.
Gotchas (what bit them)
Large models can hide an unnecessary cost/latency penalty.
Root cause: Baseline comparisons make the 70B model look like the only accurate router before fine-tuning smaller models.
Resolution: Run a data flywheel to curate examples and fine-tune smaller variants before deciding.
Unsatisfactory responses are not all router failures.
Root cause: Only a subset of user-reported unsatisfactory responses were actually due to incorrect routing.
Resolution: Use LLM-as-judge plus SME/manual analysis to attribute failures before training.
Evaluation
NVIDIA evaluated routing accuracy, first-token latency, inference cost, model size reduction, user feedback, unsatisfactory response attribution, LLM-as-judge results, and SME-confirmed ground truth.
Metrics: Router accuracy · First-token latency · Inference cost · Model size · User satisfaction · Incorrect routing count · Ground-truth data points
Outcomes
- A fine-tuned 8B model matched the 70B router's accuracy.
- A 1B model reached 94% accuracy, only 2 points below the 70B baseline.
- The 1B option offered 98% lower inference cost and 70x lower latency.
- The flywheel produced a curated 685-point ground-truth dataset.
- The system established a recurring loop for evaluation and fine-tuning.
The system supports NVIDIA employee knowledge across multiple internal domains and uses production feedback/logs to update the router.