AID: Adapting Image2Video Diffusion Models for Instruction-based Video Prediction

Zhen Xing 1 Qi Dai 2 Zejia Weng 1 Zuxuan Wu 1 Yu-Gang Jiang 1

 1 Fudan University  2 Microsoft Research Asia


   

Abstract

Text-guided video prediction (TVP) involves predicting the motion of future frames from the initial frame according to an instruction, which has wide applications in virtual reality, robotics, and content creation. Previous TVP methods make significant breakthroughs by adapting Stable Diffusion for this task. However, they struggle with frame consistency and temporal stability primarily due to the limited scale of video datasets. We observe that pretrained Image2Video diffusion models possess good priors for video dynamics but they lack textual control. Hence, transferring Image2Video models to leverage their video dynamic priors while injecting instruction control to generate controllable videos is both a meaningful and challenging task. To achieve this, we introduce the Multi-Modal Large Language Model (MLLM) to predict future video states based on initial frames and text instructions. More specifically, we design a dual query transformer (DQFormer) architecture, which integrates the instructions and frames into the conditional embeddings for future frame prediction. Additionally, we develop Long-Short Term Temporal Adapters and Spatial Adapters that can quickly transfer general video diffusion models to specific scenarios with minimal training costs. Experimental results show that our method significantly outperforms state-of-the-art techniques on four datasets: Something Something V2, Epic Kitchen-100, Bridge Data, and UCF-101. Notably, AID achieves 91.2% and 55.5% FVD improvements on Bridge and SSv2 respectively, demonstrating its effectiveness in various domains.



Method

Image

The pipeline includes a 3D U-Net for diffusion and a DQFormer for text conditioning. The parameters of the original 3D U-Net are frozen, we only fine-tune the parameters of the newly added adapter during training.







Text-conditioned Video Prediction (SSthV2)

Input frames Text instruction Real Video Synthesized Video (Ours)
Image "Lifting flashlight
up completely
without letting
it drop down"
Image Image

Input frames Text instruction Real Video Synthesized Video (Ours)
Image "Moving camera closer
to dry erase board"
Image Image

Input frames Text instruction Real Video Synthesized Video (Ours)
Image "Taking remote
out of pen stand"
Image Image

Input frames Text instruction Real Video Synthesized Video (Ours)
Image "Moving glue stick
and diskette away
from each other"
Image Image

Input frames Text instruction Real Video Synthesized Video (Ours)
Image "Showing lemon
to the camera"
Image Image

Input frames Text instruction Real Video Synthesized Video (Ours)
Image "Moving charger up" Image Image

Input frames Text instruction Real Video Synthesized Video (Ours)
Image "Pretending or failing
to wipe ink off of
a dry erase board"
Image Image

 

Text-conditioned Video Prediction (BridgeData)

Input frames Text instruction Real Video AID (Ours) Seer Method
Image "Put can in pot" Image Image Image

Input frames Text instruction Real Video AID (Ours) Seer Method
Image "Open brown1fbox
flap"
Image Image Image

Input frames Text instruction Real Video AID (Ours) Seer Method
Image "Open brown1fbox
flap"
Image Image Image

 

Text-conditioned Video Prediction (Epic-kitchen)

Input frames Text instruction Real Video AID (Ours) Seer Method
Image "Open fridge" Image Image Image

Input frames Text instruction Real Video AID (Ours) Seer Method
Image "Wash meat
flap"
Image Image Image

Input frames Text instruction Real Video AID (Ours) Seer Method
Image "Put plate on counter" Image Image Image

Input frames Text instruction Real Video AID (Ours) Seer Method
Image "Continue stirring food" Image Image Image

 

Text-conditioned Video Prediction (UCF-101)

Input frames Text instruction Real Video AID (Ours)
Image "A person is
blow-drying hair"
Image Image

Input frames Text instruction Real Video AID (Ours)
Image "A person is
applying eye makeup"
Image Image

Input frames Text instruction Real Video AID (Ours)
Image "A person is
doing push-ups."
Image Image

Bibtex

            @article{AID,
                title={AID: Adapting Image2Video Diffusion Models for Instruction-guided Video Prediction},
                author={Zhen Xing and Qi Dai and Zejia Weng and Zuxuan Wu and Yu-Gang Jiang}, 
                journal={arXiv preprint arXiv:2406.06465},
                year={2024}
              }