video-prediction-policy-2 README

The project's README from github.com/roboterax/video-prediction-policy-2. As one markdown file: README.md.

Video Prediction Policy 2: Predict Better, Act Better

Official implementation of "Video Prediction Policy 2: Predict Better, Act Better".

arXiv: 2610.10270 Project Page ModelScope — Models Hugging Face — Models

News

Overview

We introduce Video Prediction Policy 2 (VPP2), a world-action model (WAM) with strong zero-shot generalization in both video prediction and action generation.

Our pipeline has three stages:

  1. Event-level video pretraining. Continue pretraining the base video model on a large, diverse dataset of manipulation videos with detailed captions to improve generalization across open-ended tasks.
  2. Single-step visual planning. Post-train and distill the video model into a single-step visual planner with a fixed prediction horizon.
  3. Action learning. Add an action module through a mixture-of-transformers (MoT) architecture to learn an implicit inverse dynamics model.

VPP2 paper overview: event-level video prediction, instruction following, and robot policy learning.

Installation

git clone https://github.com/roboterax/video-prediction-policy-2.git
cd video-prediction-policy-2
conda create -n vpp2 python=3.10 -y
conda activate vpp2

Install the CUDA/PyTorch dependencies using the installation guide, then install the dependencies from the repository root:

python -m pip install -r requirements-train.txt -c environment-reference.txt

The scripts load VPP2 directly from this checkout; installing the vpp2 package is unnecessary. For inference and data processing only, use requirements.txt.

Get started

Benchmark What is included Guide
RoboDojo Joint video–action training, checkpoint export, policy serving, and closed-loop evaluation RoboDojo guide
LIBERO Video-stage training, action training, and standard benchmark evaluation LIBERO guide
LIBERO-OOD / PRO Generalization evaluation with the LIBERO policy checkpoint OOD and PRO setup

RoboDojo configurations live in configs/robodojo/; LIBERO configurations live in configs/libero/. Follow the guide for the benchmark you want to use.

Model Weights

Weights are available on Hugging Face and ModelScope. The Hugging Face repository is public; ModelScope requires an account with access.

Checkpoint Use Hugging Face ModelScope
RoboDojo history-conditioned Video-10k Starting Video model for joint + Action2B training Download ModelScope
RoboDojo joint Video-100k Video model paired with Action2B-100k for evaluation Download ModelScope
RoboDojo Action2B-100k RoboDojo policy evaluation Download ModelScope
LIBERO Video-10k Video model for Action training and evaluation Download ModelScope
LIBERO Action-30k LIBERO, LIBERO-OOD and LIBERO-PRO evaluation Download ModelScope

Evaluation uses the matching Video and Action checkpoints together. Shared Wan2.1 VAE, CLIP, UMT5 and tokenizer assets are also available in the model repositories under checkpoints/Wan2.1-I2V-14B-480P/. Follow the benchmark guides to download the complete assets and normalization files:

TODO

Resource Link
Paper arXiv:2610.10270
Project page VPP2 Project Page
XPolicyLab adapter VPP2 RoboDojo adapter PR
ModelScope haodong123/VPP2_preview
Hugging Face Haodong082399/VPP2

Documentation

Citation

If you use VPP2 in your research, please cite our paper (BibTeX):

@misc{guo2026videopredictionpolicy2,
  title = {Video Prediction Policy 2: Predict Better, Act Better},
  author = {Yanjiang Guo and Haodong Yan and Zhide Zhong and Zhongru Zhang and
            Qingyuan Yang and Qingzhou Lu and Xiaoyu Chen and Yen-Jen Wang and
            Shuying Deng and Chenghan Yang and Puzhen Yuan and Chenxin Liu and
            Tun Ban and Xiang Zhu and Yichen Liu and Kun Feng and Haoang Li and
            Jianyu Chen},
  year = {2026},
  eprint = {2610.10270},
  archivePrefix = {arXiv},
  primaryClass = {cs.CV},
  url = {https://arxiv.org/abs/2610.10270}
}

License

The code is released under the MIT License. Third-party components retain their original licenses; see third-party notices. Model weights, datasets, and simulator assets have their own terms.

video-prediction-policy-2 (roboterax/video-prediction-policy-2)

Searching Database…

Что это?Video Prediction Policy 2 (VPP2) — модель «мир-действие» (WAM) для роботизированной манипуляции: она предсказывает будущие видеокадры и превращает их в действия робота, демонстрируя сильную zero-shot генерализацию между задачами.
Это готовый контроллер робота или исследовательский код, который я обучаю сам?Это исследовательский код, построенный вокруг трёхэтапного конвейера обучения (предобучение на видео, дистилляция визуального планирования, обучение действий), а не контроллер, который работает сразу из коробки.
Как его запустить?Установите CUDA и PyTorch по руководству по установке, затем поставьте зависимости репозитория из его корня; скрипты загружают VPP2 прямо из чекаута, поэтому сам пакет vpp2 отдельно устанавливать не нужно.
Нужны ли предобученные чекпоинты, чтобы его протестировать?Да, парные чекпоинты Video и Action нужны вместе, плюс общие ассеты Wan2.1 VAE, CLIP, UMT5 и токенизатор; веса размещены на Hugging Face (публично) и ModelScope (нужен одобренный аккаунт).
Всё ли из статьи уже выпущено?Пока нет — чекпоинт предобученного на роботизированных видео backbone, а также код и конфиги крупномасштабного предобучения видео на уровне событий ещё не вышли.
Какие бенчмарки он поддерживает из коробки?RoboDojo (совместное обучение видео и действий, экспорт чекпоинтов, обслуживание политик, closed-loop оценка) и LIBERO, включая варианты для генерализации LIBERO-OOD и LIBERO-PRO.
Как VPP2 выступает на бенчмарках?Он занимает первое место в общем зачёте официального лидерборда RoboDojo-Sim, показывая лучшие результаты: Score 39.26 и долю успеха 32.26% по трекам agent и non-agent.
Что делает модуль действий mixture-of-transformers?Это модуль действий, добавленный после одношагового визуального планировщика: он обучает неявную модель обратной динамики и превращает предсказанные видеокадры в исполнимые действия робота.
Какие чекпоинты моделей предоставляются?Отдельные чекпоинты Video и Action для RoboDojo (Video-10k, Video-100k, Action2B-100k) и LIBERO (Video-10k, Action-30k), плюс общие ассеты Wan2.1 VAE, CLIP, UMT5 и токенизатор — всё размещено на Hugging Face и ModelScope.
Может ли VPP2 обобщать за пределами комплектных бенчмарков?Он рассчитан на сильную zero-shot генерализацию на открытые задачи манипуляции благодаря предобучению на большом разнообразном наборе видео манипуляций с подписями, хотя RoboDojo и LIBERO — пока единственные официально поддерживаемые бенчмарки для оценки.
Creator
roboterax
Size
6.7 MB
Stars
264
Updated
CommitOct 12
Age
11d
Commits · 90 Days