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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Deblurring | GoPro | PSNR29.95 | 221 | |
| Low-light Image Enhancement | LOL | PSNR25.57 | 122 | |
| Image Restoration | CDD 11 (test) | PSNR (Low)1.478 | 17 | |
| Image Dehazing | RESIDE | PSNR36.6 | 16 | |
| Image Deraining | Rain 5 sets | PSNR31.63 | 16 | |
| Image Desnowing | Snow100k | PSNR34.08 | 14 | |
| All-in-one Image Restoration | Average Rain, Low-light, Snow, Haze, Blur | PSNR31.57 | 13 | |
| Image Restoration | Real Rain | MANIQA0.374 | 7 | |
| Image Restoration | Real Snow | MANIQA0.396 | 7 | |
| Image Restoration | Real Dark | MANIQA Score0.367 | 7 |