atlas

Research map / Vision-language and multimodal / Visual question answering

Vision-Language Models for Robot Success Detection

AAAI 2024 · Vision-language and multimodal › Visual question answering · approach: Vision-language model VQA evaluation · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Fine-tunes MiniGPT-4 (Vicuna-13B) as a binary VQA success detector for robot trajectories from Berkeley Bridge and Berkeley UR5, studying few-shot accuracy, cross-dataset transfer, co-finetuning, and visual/language robustness.

Problem

Robot task success detection needs to generalize across visual, language, and task variation; the paper asks whether fine-tuned VLMs can serve as accurate success detectors, especially in low-data regimes and across datasets.

Approach

Formulate success detection as a binary VQA problem and fine-tune an open-source VLM (MiniGPT-4) on a handful of in-distribution robot trajectories, then measure how accuracy transfers across datasets, co-finetuning mixes, and visual versus language perturbations.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
fine-tune with 5 trajectories instead of no fine-tuning (ablation)MiniGPT-4 · UR5 cloth sweepbalanced accuracy: 49% -> 92%Helped
fine-tune with 5 trajectories instead of no fine-tuning (ablation)MiniGPT-4 · UR5 cup stackingbalanced accuracy: 56% -> 94%Helped
increase training trajectories from 5 to 10 (ablation)MiniGPT-4 · UR5 cloth sweepbalanced accuracy: 92% -> 86%Hurt
increase training trajectories from 5 to 10 (ablation)MiniGPT-4 · UR5 cup stackingbalanced accuracy: 94% -> 91%Hurt
fine-tune with 25 rather than 5 trajectories (ablation)MiniGPT-4 · UR5 cloth sweepbalanced accuracy: 92% -> 95%Helped
fine-tune with 25 rather than 5 trajectories (ablation)MiniGPT-4 · UR5 cup stackingbalanced accuracy: 94% -> 97%Helped
fine-tune on Bridge rather than UR5 trajectories (ablation)MiniGPT-4 · UR5 cloth sweepbalanced accuracy: 95% -> 52%Hurt
co-fine-tune with 400 Bridge trajectories (ablation)MiniGPT-4 · UR5 cloth sweepbalanced accuracy: 95% -> 90%Hurt
apply visual augmentations to test images (ablation)MiniGPT-4 · UR5 cloth sweepbalanced accuracy: 95% -> 75%Hurt
paraphrase the task prompt (ablation)MiniGPT-4 · UR5 cloth sweepbalanced accuracy: 95% -> 94%No clear effect
train on 400 Bridge trajectories and evaluate on unseen Bridge tasksMiniGPT-4 · Bridge unseen tasksbalanced accuracy: 87.6% -> 70.3%Mixed

Surprising findings

Limitations

Experimental setup

Datasets: Berkeley Bridge Berkeley UR5 Demonstration Dataset

Benchmarks: Berkeley UR5 cloth sweeping Berkeley UR5 cup stacking Berkeley Bridge unseen tasks

Metrics: balanced accuracy false positive rate false negative rate precision

Baselines: zero-shot MiniGPT-4 RoboFUME

Builds on: MiniGPT-4 RoboFUME

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy