PIVOT: Iterative Visual Prompting Elicits Actionable Knowledge for VLMs

TL;DR

PIVOT uses iterative visual prompting to enable zero-shot spatial reasoning and robotic control with VLMs, achieving 75% success in navigation tasks.

cs.RO 🔴 Advanced 2024-02-13 43 views
Soroush Nasiriany Fei Xia Wenhao Yu Ted Xiao Jacky Liang Ishita Dasgupta Annie Xie Danny Driess Ayzaan Wahid Zhuo Xu Quan Vuong Tingnan Zhang Tsang-Wei Edward Lee Kuang-Huei Lee Peng Xu Sean Kirmani Yuke Zhu Andy Zeng Karol Hausman Nicolas Heess Chelsea Finn Sergey Levine Brian Ichter
vision-language models spatial reasoning robot control visual prompting zero-shot learning

Key Findings

Methodology

PIVOT employs an iterative visual prompting framework, transforming spatial tasks into visual question answering. It generates candidate proposals, projects them onto images with annotations, and queries VLMs for relevance scores. Using a distribution fitting approach, it refines proposals across iterations, akin to cross-entropy optimization, without model fine-tuning. Experiments on real robots and simulated environments demonstrate its effectiveness in navigation, manipulation, and localization, with success rates up to 75%. The process involves multi-round sampling, visual annotation, and model-based selection, enabling precise control in complex tasks.

Key Results

  • In robotic navigation, success rates reached 75%, outperforming baseline methods at 25%. Iterative and parallel strategies significantly improved efficiency and robustness.
  • In manipulation tasks, success rate for grasping increased from 50% to 67%, with average steps reduced from 4.5 to 3, validating the method's precision.
  • On spatial localization benchmarks like RefCOCO, PIVOT achieved over 85% accuracy, demonstrating strong spatial reasoning capabilities with minimal iterations.

Significance

This work advances the application of large vision-language models in embodied and spatial reasoning tasks, enabling zero-shot control without task-specific data. It broadens the scope of VLMs from high-level question answering to low-level robotic control, promising a new paradigm for autonomous systems. The approach leverages pre-trained models' inherent decision-making ability, reducing reliance on costly fine-tuning and domain-specific datasets, thus accelerating deployment in real-world scenarios.

Technical Contribution

The paper introduces a novel iterative visual prompting framework that maps continuous spatial tasks into a visual question answering format. By iteratively refining proposals through sampling, annotation, and model evaluation, it circumvents the need for fine-tuning. The method integrates distribution fitting and multi-instance parallel calls to enhance robustness. It demonstrates that large pre-trained VLMs like GPT-4V and Gemini can perform complex spatial reasoning and control tasks in a zero-shot manner, opening new avenues for generalizable embodied AI.

Novelty

This is the first work to combine iterative visual prompting with proposal distribution fitting for continuous spatial control using VLMs. Unlike prior approaches relying on fine-tuning or explicit perception modules, it exploits the model’s inherent decision-making in a multi-round, visual-guided manner. The method transforms spatial control into a visual question answering problem, representing a significant innovation in zero-shot embodied AI.

Limitations

  • The current approach depends heavily on the quality of initial proposals and the model's spatial understanding, which can be limited in cluttered or occluded environments.
  • Computational overhead from multiple iterations and parallel calls restricts real-time deployment.
  • The method's success diminishes with highly dynamic scenes or when the initial proposal distribution poorly covers the target space.

Future Work

Future directions include integrating reinforcement learning to improve proposal sampling, reducing computational costs, and enhancing robustness in complex, dynamic environments. Combining multi-modal data and adaptive proposal strategies could further improve accuracy. Extending to multi-agent scenarios and real-time control remains a key goal, aiming for autonomous systems capable of complex spatial reasoning in unstructured settings.

AI Executive Summary

Recent advances in vision-language models (VLMs) such as GPT-4V and Gemini have unlocked remarkable capabilities in multi-modal understanding, question answering, and reasoning. However, their application to embodied tasks like robotic control and spatial localization has been limited by their primary output modality—text. Traditional methods rely heavily on fine-tuning or perception modules, which constrain generalization and scalability.

This paper introduces PIVOT, an innovative framework that transforms spatial control problems into visual question answering tasks. By iteratively annotating images with candidate actions or locations, and querying VLMs to evaluate their relevance, PIVOT effectively guides robots in complex tasks without any additional training. The core idea hinges on generating proposals, projecting them onto images with visual markers, and refining these proposals across multiple iterations using a distribution fitting approach similar to cross-entropy optimization.

Experiments on real-world robotic platforms—mobile manipulators, Franka arms—and simulated environments demonstrate that PIVOT achieves success rates up to 75% in navigation and 67% in manipulation tasks. Notably, the method requires no task-specific fine-tuning, relying solely on pre-trained models like GPT-4V and Gemini. The iterative process significantly improves robustness and precision, enabling zero-shot control in diverse settings.

This work marks a significant step toward generalizable embodied AI, leveraging the decision-making strength of large VLMs for continuous spatial reasoning. While current limitations include computational costs and sensitivity to initial proposals, future work aims to optimize efficiency, extend to dynamic environments, and incorporate reinforcement learning. Overall, PIVOT opens new horizons for deploying large-scale internet models in real-world robotics and spatial inference, promising a flexible, scalable, and powerful approach to autonomous control.

Deep Analysis

Background

The evolution of vision-language models (VLMs) such as CLIP, GPT-4V, and Gemini has revolutionized multi-modal understanding, enabling tasks like visual question answering, image captioning, and reasoning. Early works focused on high-level semantic tasks, but recent advances have shown their potential for zero-shot generalization across diverse domains. Despite these breakthroughs, applying VLMs to embodied AI—specifically robotic control and spatial reasoning—remains challenging due to their primary text output modality and difficulty in producing precise continuous outputs. Prior approaches often relied on fine-tuning or separate perception modules, which limited scalability and adaptability. This paper addresses these gaps by proposing a novel visual prompting framework that leverages the inherent decision-making ability of VLMs, aiming for zero-shot control in complex spatial tasks.

Core Problem

The core challenge lies in enabling VLMs to output continuous spatial data—such as coordinates, trajectories, or actions—without task-specific fine-tuning. Existing models excel at selecting discrete categories but struggle with precise, grounded continuous outputs essential for robotic manipulation and navigation. Additionally, the difficulty in representing spatial concepts visually and the high computational cost of iterative refinement pose significant hurdles. Overcoming these limitations requires a method that can effectively translate spatial tasks into a form compatible with pre-trained models, leveraging their decision-making capacity while maintaining efficiency and generalization. Addressing this problem is crucial for deploying autonomous systems capable of versatile, real-world operations without extensive retraining.

Innovation

The paper's key innovations include: 1) transforming spatial control tasks into visual question answering by annotating images with candidate proposals, 2) employing an iterative refinement process that samples, projects, and evaluates proposals across multiple rounds, and 3) fitting proposal distributions dynamically to guide the search towards optimal solutions without fine-tuning. This approach differs from prior methods by directly leveraging the pre-trained VLM's decision-making in a multi-round, visual-guided manner, enabling zero-shot spatial reasoning. The integration of proposal sampling, visual annotation, and distribution fitting creates a flexible framework adaptable to diverse tasks, marking a significant step forward in embodied AI.

Methodology

  • �� Define tasks as a visual question answering problem, inputting images and natural language instructions.
  • �� Generate candidate proposals (actions/positions) from a probability distribution, initially broad.
  • �� Map proposals onto the image via projection, drawing visual markers and labeling them with textual identifiers.
  • �� Input the annotated image and task description into the VLM, obtaining relevance scores or selections.
  • �� Fit a new proposal distribution based on the model's outputs, focusing on promising regions.
  • �� Repeat the sampling, annotation, and evaluation process iteratively, refining the proposals.
  • �� Use parallel calls to improve robustness, aggregating multiple candidate solutions.
  • �� Continue until convergence or maximum iterations, then execute the best action.
  • �� Validate in real robotic platforms and simulated environments, measuring success rates and efficiency.

Experiments

The experiments involve robotic navigation and manipulation tasks on mobile manipulators, Franka arms, and simulated domains like RAVENS. Metrics include success rate, steps to completion, and grasp success. The evaluation compares single-pass, multi-iteration, and parallel strategies, demonstrating that iterative refinement and parallel calls significantly enhance performance. Additional tests on spatial localization (RefCOCO) assess the model's reasoning accuracy. Ablation studies analyze the impact of proposal quantity, prompt design, and iteration count, confirming the robustness and scalability of the approach across diverse settings.

Results

PIVOT achieves up to 75% success in navigation tasks, outperforming baseline methods by 50%. In manipulation, success rates for grasping reach 67%, with average steps reduced from 4.5 to 3.0. Spatial localization accuracy exceeds 85% on RefCOCO, with minimal iteration. The iterative process consistently improves performance, and parallel calls mitigate local minima, demonstrating robustness. The results validate that large VLMs can perform complex spatial reasoning and control without fine-tuning, highlighting the method's generalizability.

Applications

This approach enables zero-shot robotic control, spatial reasoning, and visual grounding in real-world applications such as autonomous navigation, industrial automation, and assistive robotics. It requires only pre-trained VLMs and standard visual inputs, making it adaptable to various platforms. Long-term, integrating reinforcement learning and multi-modal data could further enhance autonomy, enabling robots to learn from interactions and adapt to dynamic environments, ultimately transforming autonomous systems across industries.

Limitations & Outlook

Current limitations include reliance on proposal quality, sensitivity to initial distributions, and high computational costs due to multiple iterations. Performance drops in cluttered or occluded scenes, and real-time application remains challenging. Future work should focus on improving proposal sampling, reducing inference overhead, and enhancing robustness in dynamic, complex environments. Addressing these issues is essential for deploying PIVOT in practical, real-time systems.

Plain Language Accessible to non-experts

想象你在玩一个拼图游戏,你需要把碎片拼成完整的图片。你一开始可能不知道每个碎片该放在哪里,但你可以试着把一些碎片放到不同位置,然后观察是否匹配。每次你都调整位置,直到拼出完整的图。PIVOT就像这样,它会在图片上标记一些可能的动作或位置,然后问模型“这个动作合适吗?”模型会告诉你,然后你根据建议不断调整。通过多次试错和优化,最终找到最合适的动作,让机器人完成任务。这就像你在拼图,反复试错,直到拼出完美的图。

ELI14 Explained like you're 14

想象你在玩一个超级聪明的机器人游戏,你给它一个任务,比如“把玩具放到桌子上”。这个机器人没有提前学会怎么做,但它可以用眼睛看着房间,然后用手指在图片上标记一些可能的路径或动作,比如箭头指向桌子。每次它都让模型帮忙判断哪个路径最合适,然后根据模型的建议调整路径。它会反复试几次,逐渐找到最好的动作,让机器人成功完成任务。这就像你在学校里反复试错,老师不断给你建议,直到你学会怎么做。这个方法不需要提前教机器人具体技能,只用观察和问答,就能让它变得越来越聪明。

Abstract

Vision language models (VLMs) have shown impressive capabilities across a variety of tasks, from logical reasoning to visual understanding. This opens the door to richer interaction with the world, for example robotic control. However, VLMs produce only textual outputs, while robotic control and other spatial tasks require outputting continuous coordinates, actions, or trajectories. How can we enable VLMs to handle such settings without fine-tuning on task-specific data? In this paper, we propose a novel visual prompting approach for VLMs that we call Prompting with Iterative Visual Optimization (PIVOT), which casts tasks as iterative visual question answering. In each iteration, the image is annotated with a visual representation of proposals that the VLM can refer to (e.g., candidate robot actions, localizations, or trajectories). The VLM then selects the best ones for the task. These proposals are iteratively refined, allowing the VLM to eventually zero in on the best available answer. We investigate PIVOT on real-world robotic navigation, real-world manipulation from images, instruction following in simulation, and additional spatial inference tasks such as localization. We find, perhaps surprisingly, that our approach enables zero-shot control of robotic systems without any robot training data, navigation in a variety of environments, and other capabilities. Although current performance is far from perfect, our work highlights potentials and limitations of this new regime and shows a promising approach for Internet-Scale VLMs in robotic and spatial reasoning domains. Website: pivot-prompt.github.io and HuggingFace: https://huggingface.co/spaces/pivot-prompt/pivot-prompt-demo.

cs.RO cs.CL cs.CV cs.LG