用4块RTX 4090实现顶尖视频超分辨率,兼顾画质与效率
LiftVSR: Lifting Image Diffusion to Video Super-Resolution via Hybrid Temporal Modeling with Only 4$\times$RTX 4090s
- 分阶段时序建模:短段内用动态注意力,长段间用记忆缓存
- 仅需4×RTX 4090,推理成本比主流方法降低超60%
- 适合资源有限但追求高质量视频重建的研究者和开发者
扩散模型显著提升了视频超分辨率(VSR)的感知质量,主要依赖精细设计的时序建模以保证帧间一致性。然而,现有方法通常存在时序一致性不足且计算开销巨大(如通常需超过8块NVIDIA A100-80G GPU),尤其在处理长视频时更为明显。本文提出LiftVSR,一个高效VSR框架,通过利用PixArt-α的图像级扩散先验,在仅4×RTX 4090的硬件条件下达到业界领先性能。为平衡长期一致性与效率,我们引入混合时序建模机制,将时序学习分解为两个互补组件:(i) 动态时序注意力(DTA)用于短帧段内的细粒度建模(低复杂度),通过多头查询与键令牌识别多个令牌流,对值令牌中的帧间上下文进行扭曲;(ii) 注意力记忆缓存(AMC)用于跨段的长期时序建模(高一致性),通过缓存单元自适应聚合历史段信息,以最小开销保障长期一致性。为进一步稳定推理过程中的缓存交互,我们提出一种非对称采样策略,缓解因不同扩散采样步数导致的特征不匹配问题。在多个典型VSR基准上的大量实验表明,LiftVSR在显著降低计算成本的同时实现了出色的性能表现。
原文摘要 · Abstract (English)
Diffusion models have significantly advanced video super-resolution (VSR) by enhancing perceptual quality, largely through elaborately designed temporal modeling to ensure inter-frame consistency. However, existing methods usually suffer from limited temporal coherence and prohibitively high computational costs (e.g., typically requiring over 8 NVIDIA A100-80G GPUs), especially for long videos. In this work, we propose LiftVSR, an efficient VSR framework that leverages and elevates the image-wise diffusion prior from PixArt-$α$, achieving state-of-the-art results using only 4$\times$RTX 4090 GPUs. To balance long-term consistency and efficiency, we introduce a hybrid temporal modeling mechanism that decomposes temporal learning into two complementary components: (i) Dynamic Temporal Attention (DTA) for fine-grained temporal modeling within short frame segment ($\textit{i.e.}$, low complexity), and (ii) Attention Memory Cache (AMC) for long-term temporal modeling across segments ($\textit{i.e.}$, consistency). Specifically, DTA identifies multiple token flows across frames within multi-head query and key tokens to warp inter-frame contexts in the value tokens. AMC adaptively aggregates historical segment information via a cache unit, ensuring long-term coherence with minimal overhead. To further stabilize the cache interaction during inference, we introduce an asymmetric sampling strategy that mitigates feature mismatches arising from different diffusion sampling steps. Extensive experiments on several typical VSR benchmarks have demonstrated that LiftVSR achieves impressive performance with significantly lower computational costs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。