Omni-scale Feature Learning for Lightweight Image Dehazing

Zheng Chen,Xiaojun Bi,Shuo Li,Jianyu Yue
DOI: https://doi.org/10.1007/s10489-024-05721-6
IF: 5.3
2024-01-01
Applied Intelligence
Abstract:Single image dehazing is a critical pre-processing step in high-level computer vision tasks. Existing deep models often have complex network structures and low computational efficiency, while lightweight models typically sacrifice dehazing performance. To handle these issues, we propose a novel Lightweight Omni-Scale Dehaze Network (LOS-DehazeNet). LOS-DehazeNet includes two novel blocks: the Omni-Scale Feature Aggregation (OSFA) block and the Omni-Scale Hybrid Attention (OSHAtt) block. Both of them realize the role of omni-scale feature learning. Specifically, the OSFA block leverages two mapping operations to aggregate features, enabling rich interactions among multi-scale convolutional streams. The OSHAtt block combines dual residual connections with three scales of attention mechanisms (channel, spatial, and pixel). This combination successfully alleviates the performance degradation problem of lightweight dehazing models, bringing its performance to be close to that of deep models while maintaining a low number of parameters. Compared with the latest lightweight dehazing model GRFANet, our LOS-DehazeNet not only achieves state-of-the-art dehazing results, but also has the fastest inference speed. Additionally, our model achieves a remarkable 5 × reduction in network parameters. Extensive experiments validate the effectiveness and superiority of LOS-DehazeNet in both quantitative assessments and visual quality.
What problem does this paper attempt to address?