无需训练即可在手机上生成高质量视频,三招破解算力与内存瓶颈。
On-device Sora: Enabling Training-Free Diffusion-based Text-to-Video Generation for Mobile Devices
- 通过跳步式去噪、时序令牌合并与动态加载,优化扩散模型运行效率。
- 在iPhone 15 Pro上实现媲美高端显卡的视频生成质量,帧率稳定且低延迟。
- 适合移动开发者、AI创作者及希望离线生成视频的普通用户。
我们提出On-device Sora,首个无需训练的基于扩散模型的移动端文生视频方案,可在智能手机级设备上高效运行。针对计算与内存受限的移动设备挑战,该方案对预训练视频生成模型应用三项新技术:首先,线性比例跳步(LPL)通过高效的跳步策略减少扩散过程中的冗余去噪步骤;其次,时序维度令牌合并(TDTM)通过合并时间轴上的连续令牌,降低注意力层中的密集计算开销;第三,并发推理与动态加载(CI-DL)将大模型动态分割为小块,在内存中分批加载以实现并发推理,有效缓解设备内存不足问题。我们在iPhone 15 Pro上实现On-device Sora,实验表明其可在设备端生成高质量视频,效果接近高端GPU生成结果。该工作标志着向在消费级移动与嵌入式设备上普及前沿生成技术迈出关键一步,无需耗时的重训练或模型压缩即可实现高效生成。代码已开源:https://github.com/eai-lab/On-device-Sora。
原文摘要 · Abstract (English)
We present On-device Sora, the first model training-free solution for diffusion-based on-device text-to-video generation that operates efficiently on smartphone-grade devices. To address the challenges of diffusion-based text-to-video generation on computation- and memory-limited mobile devices, the proposed On-device Sora applies three novel techniques to pre-trained video generative models. First, Linear Proportional Leap (LPL) reduces the excessive denoising steps required in video diffusion through an efficient leap-based approach. Second, Temporal Dimension Token Merging (TDTM) minimizes intensive token-processing computation in attention layers by merging consecutive tokens along the temporal dimension. Third, Concurrent Inference with Dynamic Loading (CI-DL) dynamically partitions large models into smaller blocks and loads them into memory for concurrent model inference, effectively addressing the challenges of limited device memory. We implement On-device Sora on the iPhone 15 Pro, and the experimental evaluations show that it is capable of generating high-quality videos on the device, comparable to those produced by high-end GPUs. These results show that On-device Sora enables efficient and high-quality video generation on resource-constrained mobile devices. We envision the proposed On-device Sora as a significant first step toward democratizing state-of-the-art generative technologies, enabling video generation on commodity mobile and embedded devices without resource-intensive re-training for model optimization (compression). The code implementation is available at a GitHub repository(https://github.com/eai-lab/On-device-Sora).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。