Search papers, labs, and topics across Lattice.
This paper introduces a novel test-time computation framework for VLM policy optimization that decouples state evaluation from action generation, providing a more direct supervisory signal for decision-making. The method explicitly models the advantage of an action plan using a scalable critic to estimate the reduction in distance to the goal. By employing beam search to explore multiple future paths and using a confidence-based trigger for early exit, the approach achieves a 24.6% improvement in success rate and a 56.5% reduction in inference time on multi-stage robotic manipulation tasks compared to state-of-the-art baselines.
VLMs get a 24% performance boost and run 56% faster on robot manipulation tasks by explicitly modeling action advantages and exploring multiple future paths, instead of relying on noisy foresight predictions.
Solving complex, long-horizon robotic manipulation tasks requires a deep understanding of physical interactions, reasoning about their long-term consequences, and precise high-level planning. Vision-Language Models (VLMs) offer a general perceive-reason-act framework for this goal. However, previous approaches using reflective planning to guide VLMs in correcting actions encounter significant limitations. These methods rely on inefficient and often inaccurate implicit learning of state-values from noisy foresight predictions, evaluate only a single greedy future, and suffer from substantial inference latency. To address these limitations, we propose a novel test-time computation framework that decouples state evaluation from action generation. This provides a more direct and fine-grained supervisory signal for robust decision-making. Our method explicitly models the advantage of an action plan, quantified by its reduction in distance to the goal, and uses a scalable critic to estimate. To address the stochastic nature of single-trajectory evaluation, we employ beam search to explore multiple future paths and aggregate them during decoding to model their expected long-term returns, leading to more robust action generation. Additionally, we introduce a lightweight, confidence-based trigger that allows for early exit when direct predictions are reliable, invoking reflection only when necessary. Extensive experiments on diverse, unseen multi-stage robotic manipulation tasks demonstrate a 24.6% improvement in success rate over state-of-the-art baselines, while significantly reducing inference time by 56.5%.