Depth-agnostic Single Image Dehazing
Honglei Xu, Yan Shu, Shaohui Liu
TL;DR
The paper tackles single-image dehazing by addressing the domain gap caused by depth priors in synthetic data. It introduces DA-HAZE, a depth-agnostic dataset generated via the Atmospheric Scattering Model with randomized depth mappings, and enhances generalization with a Global Shuffle Strategy that pairs haze-free images with multiple depth estimates. A Convolutional Skip Connection is proposed to improve feature fusion in U-Net-based dehazing while keeping computational costs low. Empirical results show that models trained on DA-HAZE with GSS achieve stronger performance on real-world benchmarks and exhibit reduced cross-distribution discrepancies, while CSC consistently boosts dehazing quality across methods and datasets.
Abstract
Single image dehazing is a challenging ill-posed problem. Existing datasets for training deep learning-based methods can be generated by hand-crafted or synthetic schemes. However, the former often suffers from small scales, while the latter forces models to learn scene depth instead of haze distribution, decreasing their dehazing ability. To overcome the problem, we propose a simple yet novel synthetic method to decouple the relationship between haze density and scene depth, by which a depth-agnostic dataset (DA-HAZE) is generated. Meanwhile, a Global Shuffle Strategy (GSS) is proposed for generating differently scaled datasets, thereby enhancing the generalization ability of the model. Extensive experiments indicate that models trained on DA-HAZE achieve significant improvements on real-world benchmarks, with less discrepancy between SOTS and DA-SOTS (the test set of DA-HAZE). Additionally, Depth-agnostic dehazing is a more complicated task because of the lack of depth prior. Therefore, an efficient architecture with stronger feature modeling ability and fewer computational costs is necessary. We revisit the U-Net-based architectures for dehazing, in which dedicatedly designed blocks are incorporated. However, the performances of blocks are constrained by limited feature fusion methods. To this end, we propose a Convolutional Skip Connection (CSC) module, allowing vanilla feature fusion methods to achieve promising results with minimal costs. Extensive experimental results demonstrate that current state-of-the-art methods. equipped with CSC can achieve better performance and reasonable computational expense, whether the haze distribution is relevant to the scene depth.
