Hihaze: One-Step Diffusion for High-Resolution Real-World Image Dehazing
Organizations: Shanghai Jiao Tong University
Abstract
Diffusion models have recently achieved encouraging results in real-world image dehazing. However, restoring high-resolution images remains challenging due to spatially non-uniform haze and inconsistent predictions across local patches. To address these issues, we propose Hihaze, a one-step diffusion framework for high-resolution real-world dehazing. Hihaze adopts two haze-aware modulation modules: attentive skip compensation (ASC) and a spatial degradation adapter (SDA). ASC selectively transfers reliable encoder information to the variational autoencoder (VAE) decoder, preserving structural details while suppressing haze-degraded features. SDA extracts multi-scale spatial degradation representations and injects them into the U-Net to guide restoration according to local haze variations. Furthermore, we design a training-free global-guided patch fusion (GPF) method. It uses a global prediction to align patch appearance and downweight unreliable predictions, followed by structure-guided detail compensation. Notably, this training-free design also makes GPF a plug-and-play module for other dehazing models. Experiments across four real-world benchmarks show that Hihaze surpasses state-of-the-art approaches in both visual quality and inference speed. Specifically, it delivers PSNR improvements of 3.62dB on Dense-Haze and 3.69dB on NH-HAZE compared with the strongest baselines. Code: https://github.com/zhengchen1999/Hihaze.
Figures & tables
| Non-diffusion | Diffusion / Flow | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Dataset | Metric | PTTD | IPC-Dehaze | BiLaLoRA | DiffUIR | FCDM | Diff-Dehazer | DiffDehaze | HazeFlow | Ours |
| NH-HAZE | PSNR | 14.53 | 11.18 | 12.07 | 11.79 | 14.04 | 12.99 | 14.26 | 14.38 | 18.22 |
| SSIM | 0.6119 | 0.5850 | 0.5989 | 0.4788 | 0.5955 | 0.6079 | 0.3067 | 0.6343 | 0.6664 | |
| LPIPS | 0.3934 | 0.4445 | 0.4398 | 0.5424 | 0.4264 | 0.3936 | 0.6763 | 0.4180 | 0.2863 | |
| DISTS | 0.1938 | 0.2400 | 0.2106 | 0.3462 | 0.2159 | 0.2094 | 0.3374 | 0.2021 | 0.1272 | |
| FID | 187.93 | 200.18 | 199.34 | 215.83 | 198.70 | 151.51 | 170.95 | 162.45 | 90.10 | |
| Metric | IPC-Dehaze | DiffUIR | FCDM | Diff-Dehazer | DiffDehaze | HazeFlow | Ours |
|---|---|---|---|---|---|---|---|
| Runtime (s) | 185.4 | 43.5 | 146.6 | 32.5 | 1,046.8 | 15.2 | 12.9 |
| FLOPs (T) | 826.391 | 236.731 | 2,022.874 | 453.871 | 9,386.778 | 250.991 | 458.955 |
| Params (M) | 43.164 | 36.261 | 155.334 | 1,506.159 | 1,666.750 | 87.087 | 975.822 |
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Configuration | Backbone and ASC | SDA encoder | SDA adaptation branches |
|---|---|---|---|
| Training iterations | |||
| Crop size | |||
| Total batch size | 4 | 1 | 4 |
| Learning rate | |||
| Weight decay | |||
| Learning-rate schedule | Constant | Cosine to of peak | Cosine to zero |
| Non-diffusion | Diffusion / Flow | ||||||
|---|---|---|---|---|---|---|---|
| Dataset | Metric | DehazeFormer | RIDCP | CORUN | CoA | DehazeSB | Ours |
| I-HAZE | PSNR | 16.38 | 15.63 | 17.10 | 13.82 | 16.04 | 19.02 |
| SSIM | 0.7835 | 0.7116 | 0.7658 | 0.4888 | 0.6135 | 0.7888 | |
| LPIPS | 0.3360 | 0.4158 | 0.4093 | 0.6303 | 0.5019 | 0.3217 | |
| DISTS | 0.2188 | 0.1645 | 0.2142 | 0.2512 | 0.1783 | 0.0946 | |
| FID | 135.46 | 204.88 | 136.26 | 214.98 | 202.65 | 78.99 | |
| Dataset | Setting | PSNR | SSIM | LPIPS | DISTS | FID | Q-Align |
|---|---|---|---|---|---|---|---|
| O-HAZE | Raw patch | 16.63 | 0.6484 | 0.4291 | 0.2054 | 152.20 | 3.3200 |
| + GPF | 17.95 | 0.6782 | 0.4295 | 0.1732 | 133.54 | 3.4142 | |
| NH-HAZE | Raw patch | 14.38 | 0.6343 | 0.4180 | 0.2021 | 162.45 | 3.1193 |
| + GPF | 14.72 | 0.6396 | 0.4105 | 0.1901 | 157.43 | 3.0556 | |
| Dense-Haze | Raw patch | 12.10 | 0.5320 | 0.6374 | 0.3834 | 236.11 | 2.4468 |
| + GPF | 12.57 | 0.5416 | 0.6430 | 0.3828 | 225.49 | 2.4267 |
| Dataset | Setting | PSNR | SSIM | LPIPS | DISTS | FID | Q-Align |
|---|---|---|---|---|---|---|---|
| O-HAZE | Raw patch | 16.62 | 0.6332 | 0.4085 | 0.1935 | 144.15 | 3.2273 |
| + GPF | 16.78 | 0.6456 | 0.4089 | 0.1783 | 109.77 | 3.5442 | |
| NH-HAZE | Raw patch | 12.99 | 0.6079 | 0.3936 | 0.2094 | 151.51 | 3.4304 |
| + GPF | 13.11 | 0.6156 | 0.3841 | 0.1981 | 145.86 | 3.4129 | |
| Dense-Haze | Raw patch | 10.94 | 0.4656 | 0.7149 | 0.5034 | 256.24 | 2.1314 |
| + GPF | 11.25 | 0.4704 | 0.6946 | 0.4834 | 244.70 | 2.1626 |
Image Dehazing
Image dehazing restores clear scene details and colors from images obscured by haze, fog, or smoke.
See all 35 papers