Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Gradient as Conditions: Rethinking HOG for All-in-one Image Restoration

About

All-in-one image restoration (AIR) aims to address diverse degradations within a unified model by leveraging informative degradation conditions to guide the restoration process. However, existing methods often rely on implicitly learned priors, which may entangle feature representations and hinder performance in complex or unseen scenarios. Histogram of Oriented Gradients (HOG) as a classical gradient representation, we observe that it has strong discriminative capability across diverse degradations, making it a powerful and interpretable prior for AIR. Based on this insight, we propose HOGformer, a Transformer-based model that integrates learnable HOG features for degradation-aware restoration. The core of HOGformer is a Dynamic HOG-aware Self-Attention (DHOGSA) mechanism, which adaptively models long-range spatial dependencies conditioned on degradation-specific cues encoded by HOG descriptors. To further adapt the heterogeneity of degradations in AIR, we propose a Dynamic Interaction Feed-Forward (DIFF) module that facilitates channel-spatial interactions, enabling robust feature transformation under diverse degradations. Besides, we propose a HOG loss to explicitly enhance structural fidelity and edge sharpness. Extensive experiments on a variety of benchmarks, including adverse weather and natural degradations, demonstrate that HOGformer achieves state-of-the-art performance and generalizes well to complex real-world scenarios.Code is available at https://github.com/Fire-friend/HOGformer.

Jiawei Wu, Zhifei Yang, Zhe Wang, Zhi Jin• 2025

Related benchmarks

TaskDatasetResultRank
Image DeblurringGoPro
PSNR29.95
221
Low-light Image EnhancementLOL
PSNR25.57
122
Image RestorationCDD 11 (test)
PSNR (Low)1.478
17
Image DehazingRESIDE
PSNR36.6
16
Image DerainingRain 5 sets
PSNR31.63
16
Image DesnowingSnow100k
PSNR34.08
14
All-in-one Image RestorationAverage Rain, Low-light, Snow, Haze, Blur
PSNR31.57
13
Image RestorationReal Rain
MANIQA0.374
7
Image RestorationReal Snow
MANIQA0.396
7
Image RestorationReal Dark
MANIQA Score0.367
7
Showing 10 of 11 rows

Other info

Follow for update