R2V Agent: Teaching SLMs When to Ask for Help

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hemadri, Raghu Vamshi, Mohammed, Humaira Firdowse, Maheshwary, Rishabh, Daruru, Srivatsava, Davasam, Sagar, Yadav, Vikas, Sunkara, Srinivas, Rajeswar, Sai
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911690567712768
author Hemadri, Raghu Vamshi
Mohammed, Humaira Firdowse
Maheshwary, Rishabh
Daruru, Srivatsava
Davasam, Sagar
Yadav, Vikas
Sunkara, Srinivas
Rajeswar, Sai
author_facet Hemadri, Raghu Vamshi
Mohammed, Humaira Firdowse
Maheshwary, Rishabh
Daruru, Srivatsava
Davasam, Sagar
Yadav, Vikas
Sunkara, Srinivas
Rajeswar, Sai
contents Efficient agentic systems should incur expensive frontier-model costs only on decisions where a cheaper local model is likely to fail. Existing LLM cascades usually route whole queries before execution, but task difficulty shifts mid-trajectory - after flaky tool calls, truncated observations, or compounding local errors - making pre-execution routing brittle. We introduce \textbf{R2V-Agent}, a risk-calibrated SLM-LLM routing framework for interactive agents. R2V combines four components: a distilled small language model (SLM) policy, a stronger teacher LLM, a lightweight process verifier that scores candidate actions at each step, and a calibrated step-level router. The router is our central contribution: after the SLM is trained, it estimates residual failure risk at each step and escalates only when teacher intervention is warranted. To make the routing problem well-defined, we first train a stable local SLM using a standard offline pipeline: behavioral cloning (BC) on teacher trajectories, followed by verifier-guided Direct Preference Optimization (DPO) with consistency regularization. The router is then trained on this fixed policy's residual failures using Brier-calibrated probability estimation and a Conditional Value-at-Risk (CVaR)-constrained objective that penalizes worst-case failures across perturbation seeds. Across HumanEval+, TextWorld, and TerminalBench with four SLM backbones, R2V improves the reliability-cost frontier: it achieves $94.3\%$ HumanEval+ success with $0.60\%$ LLM escalation, recovers TextWorld from $64.6\%$ SLM-only success to $98.2\%$ at $41.7\%$ escalation, and reaches $93.3\%$ TerminalBench success at $33.9\%$ LLM calls, roughly half the heuristic-router cost.
format Preprint
id arxiv_https___arxiv_org_abs_2605_16604
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle R2V Agent: Teaching SLMs When to Ask for Help
Hemadri, Raghu Vamshi
Mohammed, Humaira Firdowse
Maheshwary, Rishabh
Daruru, Srivatsava
Davasam, Sagar
Yadav, Vikas
Sunkara, Srinivas
Rajeswar, Sai
Machine Learning
Efficient agentic systems should incur expensive frontier-model costs only on decisions where a cheaper local model is likely to fail. Existing LLM cascades usually route whole queries before execution, but task difficulty shifts mid-trajectory - after flaky tool calls, truncated observations, or compounding local errors - making pre-execution routing brittle. We introduce \textbf{R2V-Agent}, a risk-calibrated SLM-LLM routing framework for interactive agents. R2V combines four components: a distilled small language model (SLM) policy, a stronger teacher LLM, a lightweight process verifier that scores candidate actions at each step, and a calibrated step-level router. The router is our central contribution: after the SLM is trained, it estimates residual failure risk at each step and escalates only when teacher intervention is warranted. To make the routing problem well-defined, we first train a stable local SLM using a standard offline pipeline: behavioral cloning (BC) on teacher trajectories, followed by verifier-guided Direct Preference Optimization (DPO) with consistency regularization. The router is then trained on this fixed policy's residual failures using Brier-calibrated probability estimation and a Conditional Value-at-Risk (CVaR)-constrained objective that penalizes worst-case failures across perturbation seeds. Across HumanEval+, TextWorld, and TerminalBench with four SLM backbones, R2V improves the reliability-cost frontier: it achieves $94.3\%$ HumanEval+ success with $0.60\%$ LLM escalation, recovers TextWorld from $64.6\%$ SLM-only success to $98.2\%$ at $41.7\%$ escalation, and reaches $93.3\%$ TerminalBench success at $33.9\%$ LLM calls, roughly half the heuristic-router cost.
title R2V Agent: Teaching SLMs When to Ask for Help
topic Machine Learning
url https://arxiv.org/abs/2605.16604