Search papers, labs, and topics across Lattice.
SwiftVR introduces a novel framework for real-time video restoration that addresses the latency and memory challenges of existing diffusion-based models on consumer-grade GPUs. By employing a mask-free shifted-window self-attention mechanism and a lightweight Restoration-aware Autoencoder, SwiftVR achieves high-resolution outputs while maintaining efficient processing speeds. The model demonstrates impressive performance, sustaining 31 FPS at 2560x1440 and 14 FPS at 3840x2160 on an H100, marking a significant advancement in generative video restoration capabilities.
SwiftVR achieves real-time 1080p video restoration on consumer GPUs, a first in the field, while maintaining high perceptual quality and low inference costs.
Real-time video restoration (VR) for live streams requires high-resolution outputs under strict per-frame latency constraints. Existing one-step diffusion-based VR models remain difficult to deploy on consumer-grade GPUs due to two main bottlenecks: quadratic spatial attention at high resolutions and the latency-memory overhead of large video autoencoders. We present SwiftVR, a streaming one-step generative VR framework that reduces both bottlenecks under a causal chunk-wise protocol. For attention, mask-free shifted-window self-attention gathers each spatial window into a dense tensor via deterministic indexing, keeping all attention calls on the dense scaled dot-product attention path without masks, cyclic shifts, padding, or hardware-specific sparse kernels. Because SwiftVR uses only standard dense SDPA calls, the trained model transfers to consumer GPUs without retraining or custom kernels. For autoencoding, a lightweight Restoration-aware Autoencoder enables fast chunk-wise decoding while preserving reconstruction quality. On a single H100, SwiftVR sustains 31~FPS at 2560x1440 and 14~FPS at 3840x2160, whereas all compared diffusion-based VR baselines exceed the memory limit at 4K. On a consumer RTX~5090, SwiftVR reaches 26~FPS at 1920x1080. To our knowledge, SwiftVR is the first generative VR model to achieve real-time 1080p streaming on a consumer-grade GPU, while attaining strong no-reference perceptual quality with lower inference cost. Project is available at https://h-oliday.github.io/SwiftVR.
