VEDIT: Latent Prediction Architecture For Procedural Video Representation Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Lin, Han, Nagarajan, Tushar, Ballas, Nicolas, Assran, Mido, Komeili, Mojtaba, Bansal, Mohit, Sinha, Koustuv
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917794208022528
author Lin, Han
Nagarajan, Tushar
Ballas, Nicolas
Assran, Mido
Komeili, Mojtaba
Bansal, Mohit
Sinha, Koustuv
author_facet Lin, Han
Nagarajan, Tushar
Ballas, Nicolas
Assran, Mido
Komeili, Mojtaba
Bansal, Mohit
Sinha, Koustuv
contents Procedural video representation learning is an active research area where the objective is to learn an agent which can anticipate and forecast the future given the present video input, typically in conjunction with textual annotations. Prior works often rely on large-scale pretraining of visual encoders and prediction models with language supervision. However, the necessity and effectiveness of extending compute intensive pretraining to learn video clip sequences with noisy text supervision have not yet been fully validated by previous works. In this work, we show that a strong off-the-shelf frozen pretrained visual encoder, along with a well designed prediction model, can achieve state-of-the-art (SoTA) performance in forecasting and procedural planning without the need for pretraining the prediction model, nor requiring additional supervision from language or ASR. Instead of learning representations from pixel space, our method utilizes the latent embedding space of publicly available vision encoders. By conditioning on frozen clip-level embeddings from observed steps to predict the actions of unseen steps, our prediction model is able to learn robust representations for forecasting through iterative denoising - leveraging the recent advances in diffusion transformers (Peebles & Xie, 2023). Empirical studies over a total of five procedural learning tasks across four datasets (NIV, CrossTask, COIN and Ego4D-v2) show that our model advances the strong baselines in long-horizon action anticipation (+2.6% in Verb ED@20, +3.1% in Noun ED@20), and significantly improves the SoTA in step forecasting (+5.0%), task classification (+3.8%), and procedure planning tasks (up to +2.28% in success rate, +3.39% in mAcc, and +0.90% in mIoU).
format Preprint
id arxiv_https___arxiv_org_abs_2410_03478
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle VEDIT: Latent Prediction Architecture For Procedural Video Representation Learning
Lin, Han
Nagarajan, Tushar
Ballas, Nicolas
Assran, Mido
Komeili, Mojtaba
Bansal, Mohit
Sinha, Koustuv
Computer Vision and Pattern Recognition
Machine Learning
Procedural video representation learning is an active research area where the objective is to learn an agent which can anticipate and forecast the future given the present video input, typically in conjunction with textual annotations. Prior works often rely on large-scale pretraining of visual encoders and prediction models with language supervision. However, the necessity and effectiveness of extending compute intensive pretraining to learn video clip sequences with noisy text supervision have not yet been fully validated by previous works. In this work, we show that a strong off-the-shelf frozen pretrained visual encoder, along with a well designed prediction model, can achieve state-of-the-art (SoTA) performance in forecasting and procedural planning without the need for pretraining the prediction model, nor requiring additional supervision from language or ASR. Instead of learning representations from pixel space, our method utilizes the latent embedding space of publicly available vision encoders. By conditioning on frozen clip-level embeddings from observed steps to predict the actions of unseen steps, our prediction model is able to learn robust representations for forecasting through iterative denoising - leveraging the recent advances in diffusion transformers (Peebles & Xie, 2023). Empirical studies over a total of five procedural learning tasks across four datasets (NIV, CrossTask, COIN and Ego4D-v2) show that our model advances the strong baselines in long-horizon action anticipation (+2.6% in Verb ED@20, +3.1% in Noun ED@20), and significantly improves the SoTA in step forecasting (+5.0%), task classification (+3.8%), and procedure planning tasks (up to +2.28% in success rate, +3.39% in mAcc, and +0.90% in mIoU).
title VEDIT: Latent Prediction Architecture For Procedural Video Representation Learning
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2410.03478