atlas

Research map / Vision-language and multimodal / CLIP and zero-shot

Vision-Language Models are Zero-Shot Reward Models for Reinforcement Learning

ICLR 2024 · Vision-language and multimodal › CLIP and zero-shot · approach: Adapting VLMs for zero-shot classification · 6 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.

Uses pretrained CLIP as a zero-shot reward model (VLM-RM) with optional goal-baseline regularization to train RL agents (SAC/DQN) on classic control and MuJoCo humanoid tasks specified by one text prompt.

Problem

Reward specification in vision-based RL is either manual and infeasible or requires expensive human feedback to learn reward models; the paper asks whether pretrained VLMs can serve as zero-shot, language-specified reward functions.

Approach

Use the cosine similarity between a CLIP image embedding of the current state and a CLIP text embedding of a task prompt as the RL reward, with no fine-tuning, and optionally project out irrelevant embedding directions using a neutral baseline prompt (goal-baseline regularization).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add realistic textures to MountainCar observations (ablation)CLIP reward model · MountainCartask success rateHelped
add goal-baseline regularization to the CLIP reward (ablation)VLM-RM · CartPolereward landscape qualityNo clear effect
add goal-baseline regularization to the CLIP reward (ablation)VLM-RM · humanoid kneelingEPIC distance to human labelsHelped
add realistic textures to the humanoid environment (ablation)VLM-RM · Humanoid-v4: kneelingreward model qualityHelped
reposition the camera to give a better view of the humanoid (ablation)VLM-RM · Humanoid-v4: kneelingreward model qualityHelped
increase CLIP model scaleVLM-RM · Humanoid-v4: kneelingsuccess rate: 0% success rate for all smaller models -> successfully learn the kneeling task with VIT-bigG-14Helped
add goal-baseline regularization (ablation)VLM-RM · humanoid tasks that failed with α=0task performanceNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: MuJoCo Humanoid-v4 environment CartPole and MountainCar (OpenAI Gym) LAION-5B (CLIP pretraining, not used directly)

Benchmarks: CartPole MountainCar Humanoid-v4 (kneeling, lotus position, standing up, arms raised, doing splits, hands on hips, arms crossed, standing on

Metrics: human-evaluated success rate ground truth episode reward EPIC distance

Baselines: CLIP RN50 OpenCLIP ViT-L-14 OpenCLIP ViT-H-14 ground truth environment reward

Builds on: CLIP Soft Actor-Critic DQN stable-baselines3 EPIC distance

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