arXiv:2410.03478cs.CVcs.LG2024-10ICLR被引 12

用预训练视觉编码器的隐空间预测视频动作,无需额外训练或语言监督。

VEDIT: Latent Prediction Architecture For Procedural Video Representation Learning

  • 基于冻结的视觉编码器隐空间,用扩散模型迭代去噪预测未来动作
  • 在5个任务上实现最高准确率,长时程动作预测提升2.6%~3.1%
  • 适合做视频理解与规划,尤其无语言标注场景

程序化视频表征学习旨在让智能体根据当前视频输入及文本注释预测未来。以往方法依赖大规模视觉编码器和预测模型的预训练,配合语言监督。然而,这种高计算开销的预训练在含噪声文本监督下的有效性尚未充分验证。本文表明:仅使用现成的冻结预训练视觉编码器,搭配精心设计的预测模型,即可在无需预训练预测模块、也不依赖语言或语音识别(ASR)监督的情况下,达到最先进的性能。本方法不直接处理像素,而是利用公开视觉编码器的隐空间表示。通过条件于已观察步骤的片段级嵌入,预测未见步骤的动作,其预测模型通过迭代去噪学习鲁棒表征,借助扩散变换器(Peebles & Xie, 2023)最新进展。在四个数据集(NIV、CrossTask、COIN、Ego4D-v2)上的五项程序学习任务中,模型在长时程动作预测(Verb ED@20提升2.6%,Noun ED@20提升3.1%)、步骤预测(+5.0%)、任务分类(+3.8%)、流程规划(成功率达+2.28%,mAcc提升+3.39%,mIoU提升+0.90%)上均超越现有最佳结果。

原文摘要 · Abstract (English)

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).

视频预测扩散模型隐空间动作规划

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。