When to Act, Ask, or Learn: Uncertainty-Aware Policy Steering
On the map: Diffusion models → Diffusion and flow policies for robotics
Abstract (arXiv)
Policy steering is an emerging way to adapt robot behaviors at deployment-time: a learned verifier analyzes low-level action samples proposed by a pre-trained policy (e.g., diffusion policy) and selects only those aligned with the task. While Vision-Language Models (VLMs) are promising general-purpose verifiers due to their reasoning capabilities, existing frameworks often assume these models are well-calibrated. In practice, the overconfident judgment from VLM can degrade the steering performance under both high-level semantic uncertainty in task specifications and low-level action uncertainty or incapability of the pre-trained policy. We propose uncertainty-aware policy steering (UPS), a framework that jointly reasons about semantic task uncertainty and low-level action feasibility, and selects an uncertainty resolution strategy: execute a high-confidence action, clarify task ambiguity via natural language queries, or ask for action interventions to correct the low-level policy when it is deemed incapable at the task. We leverage conformal prediction to calibrate the composition of the VLM and the pre-trained base policy, providing statistical assurances that the verifier selects the correct strategy. After collecting interventions during deployment, we employ residual learning to improve the capability of the pre-trained policy, enabling the system to learn continually but with minimal expensive human feedback. We demonstrate our framework through experiments in simulation and on hardware, showing that UPS can disentangle confident, ambiguous, and incapable scenarios and minimizes expensive user interventions compared to uncalibrated baselines and prior human- or robot-gated continual learning approaches. Videos can be found at https://jessie-yuan.github.io/ups/
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 29 of its 47 references that are in the map. First posted on arXiv 25 Feb 2026.
Builds on
29 of 47 references.
Closest ancestors · 1
Same direction, newest first
Other references · 28
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2024π0: A Vision-Language-Action Flow Model for General Robot Controlcited by 2,297
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 699
- 2020Classification with Valid and Adaptive Coveragecited by 390
- 2024Hallucination of Multimodal Large Language Models: A Surveycited by 380
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planningcited by 357
- 2023Robots That Ask For Help: Uncertainty Alignment for Large Language Model Plannerscited by 298
- 2018HG-DAgger: Interactive Imitation Learning with Human Expertscited by 245
- 2018Residual Policy Learningcited by 146
- 2021ThriftyDAgger: Budget-Aware Novelty and Risk Gating for Interactive Imitation Learningcited by 98
16 more
- 2018EnsembleDAgger: A Bayesian Approach to Safe Imitation Learningcited by 96
- 2025Self-Improving Vision-Language-Action Models with Data Generation via Residual RLcited by 72
- 2025RoboMonkey: Scaling Test-Time Sampling and Verification for Vision-Language-Action Modelscited by 56
- 2023Calibrating LLM-Based Evaluatorcited by 53
- 2025From Foresight to Forethought: VLM-In-the-Loop Policy Steering via Latent Alignmentcited by 51
- 2024TRANSIC: Sim-to-Real Policy Transfer by Learning from Online Correctioncited by 50
- 2025Reasoning about Uncertainty: Do Reasoning Models Know When They Don't Know?cited by 35
- 2024Multi-Task Interactive Robot Fleet Learning with Visual World Modelscited by 33
- 2025Do What You Say: Steering Vision-Language-Action Models via Runtime Reasoning-Action Alignment Verificationcited by 25
- 2025Evaluating Uncertainty and Quality of Vision-Language-Action-enabled Robotscited by 17
- 2025World Models That Know When They Don't Know - Controllable Video Generation with Calibrated Uncertaintycited by 16
- 2025Let's Think in Two Steps: Mitigating Agreement Bias in MLLMs with Self-Grounded Verificationcited by 12
- 2024Conformalized Interactive Imitation Learning: Handling Expert Shift and Intermittent Feedbackcited by 11
- 2025Confidence Calibration in Vision-Language-Action Modelscited by 10
- 2024Risk-Calibrated Human-Robot Interaction via Set-Valued Intent Predictioncited by 8
- 2024LBAP: Improved Uncertainty Alignment of LLM Planners using Bayesian Inferencecited by 2
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Hide-and-Seek in Trajectories: Discovering Failure Signals for VLA Runtime Monitoringcited by 6
- 2026PATCH: Action-Chunk-Conditioned Latent Patch Innovation Monitoring for Robot Manipulationcited by 2
- 2026Visual Verification Enables Inference-time Steering and Autonomous Policy Improvementcited by 1
- 2026Learning When to Act: Interval-Aware Reinforcement Learning with Predictive Temporal Structurecited by 0
- 2026Visual Verification Enables Inference-time Steering and Autonomous Policy Improvement16 Jun 2026
- 2026PATCH: Action-Chunk-Conditioned Latent Patch Innovation Monitoring for Robot Manipulation15 Jun 2026
- 2026Hide-and-Seek in Trajectories: Discovering Failure Signals for VLA Runtime Monitoring29 May 2026
- 2026Learning When to Act: Interval-Aware Reinforcement Learning with Predictive Temporal Structure23 Mar 2026
Similar, not cited
Close in wording, no citation either way
- 2026VLS: Steering Pretrained Robot Policies via Vision-Language Models
- 2025GUIDES: Guidance Using Instructor-Distilled Embeddings for Pre-trained Robot Policy Enhancement ↗
- 2026Prioritized Rollouts for Efficient World Model-based Vision-Language-Action Policy Optimization
- 2026Steerable Vision-Language-Action Policies for Embodied Reasoning and Hierarchical Control
- 2026Learning What to Say to Your VLA: Mostly Harmless Vision Language Action Model Steering
- 2026Uncertainty Quantification for Flow-Based Vision-Language-Action Models
- 2026TTT-VLA: Test-Time Latent Prompt Optimization for Vision-Language-Action Models
- 2026FineVLA: Fine-Grained Instruction Alignment for Steerable Vision-Language-Action Policies