HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

DeblurDiNAT: A Compact Model with Exceptional Generalization and Visual Fidelity on Unseen Domains

Hanzhou Liu Binghan Li Chengkai Liu Mi Lu

DeblurDiNAT: A Compact Model with Exceptional Generalization and Visual Fidelity on Unseen Domains

Abstract

Recent deblurring networks have effectively restored clear images from the blurred ones. However, they often struggle with generalization to unknown domains. Moreover, these models typically focus on distortion metrics such as PSNR and SSIM, neglecting the critical aspect of metrics aligned with human perception. To address these limitations, we propose DeblurDiNAT, a deblurring Transformer based on Dilated Neighborhood Attention. First, DeblurDiNAT employs an alternating dilation factor paradigm to capture both local and global blurred patterns, enhancing generalization and perceptual clarity. Second, a local cross-channel learner aids the Transformer block to understand the short-range relationships between adjacent channels. Additionally, we present a linear feed-forward network with a simple while effective design. Finally, a dual-stage feature fusion module is introduced as an alternative to the existing approach, which efficiently process multi-scale visual information across network levels. Compared to state-of-the-art models, our compact DeblurDiNAT demonstrates superior generalization capabilities and achieves remarkable performance in perceptual metrics, while maintaining a favorable model size.

Code Repositories

hanzhouliu/deblurdinat
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
deblurring-on-goproDeblurDiNAT-L
PSNR: 33.63
SSIM: 0.967
deblurring-on-hide-trained-on-goproDeblurDiNAT-L
PSNR (sRGB): 31.47
Params (M): 16.1
SSIM (sRGB): 0.944
deblurring-on-realblur-j-trained-on-goproDeblurDiNAT-L
PSNR (sRGB): 28.98
SSIM (sRGB): 0.885
deblurring-on-realblur-r-trained-on-goproDeblurDiNAT-L
PSNR (sRGB): 36.09
SSIM (sRGB): 0.955

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