HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Learning Truncated Causal History Model for Video Restoration

Amirhosein Ghasemabadi; Muhammad Kamran Janjua; Mohammad Salameh; Di Niu

Learning Truncated Causal History Model for Video Restoration

Abstract

One key challenge to video restoration is to model the transition dynamics of video frames governed by motion. In this work, we propose TURTLE to learn the truncated causal history model for efficient and high-performing video restoration. Unlike traditional methods that process a range of contextual frames in parallel, TURTLE enhances efficiency by storing and summarizing a truncated history of the input frame latent representation into an evolving historical state. This is achieved through a sophisticated similarity-based retrieval mechanism that implicitly accounts for inter-frame motion and alignment. The causal design in TURTLE enables recurrence in inference through state-memorized historical features while allowing parallel training by sampling truncated video clips. We report new state-of-the-art results on a multitude of video restoration benchmark tasks, including video desnowing, nighttime video deraining, video raindrops and rain streak removal, video super-resolution, real-world and synthetic video deblurring, and blind video denoising while reducing the computational cost compared to existing best contextual methods on all these tasks.

Code Repositories

Ascend-Research/Turtle
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
deblurring-on-beam-splitter-deblurring-bsdTurtle
PSNR: 33.58
deblurring-on-goproTurtle
PSNR: 34.5
SSIM: 0.972
rain-removal-on-nighrainTurtle
PSNR: 29.26
video-denoising-on-set8-sigma50Turtle
PSNR: 30.29
video-deraining-on-vrdsTurtle
PSNR: 32.01
SSIM: 0.9590

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Learning Truncated Causal History Model for Video Restoration | Papers | HyperAI