MAPF-Net: lightweight network for dehazing via multi-scale attention and physics-aware feature fusion

被引:1
作者
Huang, Guangye [1 ]
Zhang, Jindong [1 ,2 ]
机构
[1] Jilin Univ, Coll Comp Sci & Technol, Changchun 130012, Peoples R China
[2] Jilin Univ, Key Lab Symbol Computat & Knowledge Engn, Minist Educ, Changchun 130012, Peoples R China
关键词
Single image dehazing; Lightweight network; Multi-scale attention aggregation; Physics-aware feature fusion; IMAGE; HAZE;
D O I
10.1007/s11227-025-07089-x
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Single image dehazing is an important preprocessing step in advanced computer vision tasks. Existing deep learning models often have complex network structures, leading to low computational efficiency, while lightweight models usually compromise on dehazing performance. To address these issues, this paper proposes a lightweight network for dehazing with multi-scale attention and physics-aware feature fusion (MAPF-Net), aiming to achieve significant improvement in dehazing performance while maintaining a lightweight model. MAPF-Net includes two novel blocks: the multi-scale attention aggregation (MAA) module and the physics-aware feature fusion (PAFF) module. Specifically, the MAA module utilizes multi-scale large-kernel convolutions to extract features and aggregates multi-scale features through dual-residual parallel attention. This design not only enriches feature representation by fusing multi-scale contextual information but also mitigates gradient degradation via residual pathways, thereby maintaining discriminative capabilities for both dense and sparse haze regions. The PAFF module integrates atmospheric scattering priors into a learnable feature fusion framework, where multi-head self-attention captures global dependencies of haze patterns while pixel attention refines local texture details. It adopts a frequency-aware feature-weighted fusion strategy that balances low-frequency structural preservation and high-frequency texture recovery through learnable channel-wise coefficients. Comprehensive experiments conducted on multiple benchmark datasets, along with ablation studies, validate the effectiveness of the proposed method. The outcomes show that MAPF-Net uses just 0.687M parameters and 7.547G FLOPs to deliver state-of-the-art performance.
引用
收藏
页数:27
相关论文
共 54 条
[1]   NH-HAZE: An Image Dehazing Benchmark with Non-Homogeneous Hazy and Haze-Free Images [J].
Ancuti, Codruta O. ;
Ancuti, Cosmin ;
Timofte, Radu .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020), 2020, :1798-1805
[2]  
Ancuti CO, 2019, IEEE IMAGE PROC, P1014, DOI [10.1109/icip.2019.8803046, 10.1109/ICIP.2019.8803046]
[3]   Single Image Dehazing Using Haze-Lines [J].
Berman, Dana ;
Treibitz, Tali ;
Avidan, Shai .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2020, 42 (03) :720-734
[4]   Non-Local Image Dehazing [J].
Berman, Dana ;
Treibitz, Tali ;
Avidan, Shai .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :1674-1682
[5]   Generalized Foggy-Scene Semantic Segmentation by Frequency Decoupling [J].
Bi, Qi ;
You, Shaodi ;
Gevers, Theo .
2024 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, CVPRW, 2024, :1389-1399
[6]   DehazeNet: An End-to-End System for Single Image Haze Removal [J].
Cai, Bolun ;
Xu, Xiangmin ;
Jia, Kui ;
Qing, Chunmei ;
Tao, Dacheng .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2016, 25 (11) :5187-5198
[7]   Gated Context Aggregation Network for Image Dehazing and Deraining [J].
Chen, Dongdong ;
He, Mingming ;
Fan, Qingnan ;
Liao, Jing ;
Zhang, Liheng ;
Hou, Dongdong ;
Yuan, Lu ;
Hua, Gang .
2019 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV), 2019, :1375-1383
[8]   PSD: Principled Synthetic-to-Real Dehazing Guided by Physical Priors [J].
Chen, Zeyuan ;
Wang, Yangchao ;
Yang, Yang ;
Liu, Dong .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :7176-7185
[9]  
Chen ZH, 2024, Arxiv, DOI arXiv:2403.08557
[10]   Multi-Scale Boosted Dehazing Network with Dense Feature Fusion [J].
Dong, Hang ;
Pan, Jinshan ;
Xiang, Lei ;
Hu, Zhe ;
Zhang, Xinyi ;
Wang, Fei ;
Yang, Ming-Hsuan .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :2154-2164