Mobile Video Diffusion

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Yahia, Haitam Ben, Korzhenkov, Denis, Lelekas, Ioannis, Ghodrati, Amir, Habibian, Amirhossein
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913605991006208
author Yahia, Haitam Ben
Korzhenkov, Denis
Lelekas, Ioannis
Ghodrati, Amir
Habibian, Amirhossein
author_facet Yahia, Haitam Ben
Korzhenkov, Denis
Lelekas, Ioannis
Ghodrati, Amir
Habibian, Amirhossein
contents Video diffusion models have achieved impressive realism and controllability but are limited by high computational demands, restricting their use on mobile devices. This paper introduces the first mobile-optimized video diffusion model. Starting from a spatio-temporal UNet from Stable Video Diffusion (SVD), we reduce memory and computational cost by reducing the frame resolution, incorporating multi-scale temporal representations, and introducing two novel pruning schema to reduce the number of channels and temporal blocks. Furthermore, we employ adversarial finetuning to reduce the denoising to a single step. Our model, coined as MobileVD, is 523x more efficient (1817.2 vs. 4.34 TFLOPs) with a slight quality drop (FVD 149 vs. 171), generating latents for a 14x512x256 px clip in 1.7 seconds on a Xiaomi-14 Pro. Our results are available at https://qualcomm-ai-research.github.io/mobile-video-diffusion/
format Preprint
id arxiv_https___arxiv_org_abs_2412_07583
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Mobile Video Diffusion
Yahia, Haitam Ben
Korzhenkov, Denis
Lelekas, Ioannis
Ghodrati, Amir
Habibian, Amirhossein
Computer Vision and Pattern Recognition
Artificial Intelligence
Video diffusion models have achieved impressive realism and controllability but are limited by high computational demands, restricting their use on mobile devices. This paper introduces the first mobile-optimized video diffusion model. Starting from a spatio-temporal UNet from Stable Video Diffusion (SVD), we reduce memory and computational cost by reducing the frame resolution, incorporating multi-scale temporal representations, and introducing two novel pruning schema to reduce the number of channels and temporal blocks. Furthermore, we employ adversarial finetuning to reduce the denoising to a single step. Our model, coined as MobileVD, is 523x more efficient (1817.2 vs. 4.34 TFLOPs) with a slight quality drop (FVD 149 vs. 171), generating latents for a 14x512x256 px clip in 1.7 seconds on a Xiaomi-14 Pro. Our results are available at https://qualcomm-ai-research.github.io/mobile-video-diffusion/
title Mobile Video Diffusion
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2412.07583