Semi-supervised single image dehazing based on dual-teacher-student network with knowledge transfer

被引:2
作者
Liu, Jianlei [1 ]
Hou, Qianwen [1 ]
Wang, Shilong [1 ]
Zhang, Xueqing [1 ]
机构
[1] Qufu Normal Univ, Sch Cyber Sci & Engn, 57 Jingxuan West Rd, Jining 273165, Peoples R China
基金
中国国家自然科学基金;
关键词
Single image dehazing; Semi-supervised learning; Mean teacher; Knowledge transfer; Contrastive learning;
D O I
10.1007/s11760-024-03216-y
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
While significant progress has been made in image dehazing techniques, the lack of large-scale labeled datasets remains one of the limiting factors for enhancing the performance of image dehazing algorithms. Therefore, based on the mean teacher model, we propose a semi-supervised dehazing framework with a dual-teacher-student (DTS) architecture. DTS is composed of a pretrained teacher network (P-teacher), a mean teacher network (M-teacher), and a student network. The P-teacher facilitates the student network in learning intermediate layer features that resemble haze-free images through knowledge transfer. The M-teacher guides the student network in image dehazing in unsupervised manner. The P-teacher, M-teacher, and student networks share the same network architecture known as the multiscale feature fusion attention-enhanced network (MFFA-Net). The MFFA-Net consists of a multiscale feature fusion network (MFF-Net) and an attention network (A-Net). The MFF-Net is responsible for fusing features from different levels. The A-Net is capable of compensating for information loss during downsampling in the MFF-Net and dynamically adjusting the focus on different regions. Extensive experimental results demonstrate that the dehazing method proposed in this paper outperforms several state-of-the-art algorithms on multiple datasets.The code has been released on https://github.com/houqianwen/MFFA-Net.
引用
收藏
页码:5073 / 5087
页数:15
相关论文
共 45 条
[1]   O-HAZE: a dehazing benchmark with real hazy and haze-free outdoor images [J].
Ancuti, Codruta O. ;
Ancuti, Cosmin ;
Timofte, Radu ;
De Vleeschouwer, Christophe .
PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2018, :867-875
[2]   I-HAZE: A Dehazing Benchmark with Real Hazy and Haze-Free Indoor Images [J].
Ancuti, Cosmin ;
Ancuti, Codruta O. ;
Timofte, Radu ;
De Vleeschouwer, Christophe .
ADVANCED CONCEPTS FOR INTELLIGENT VISION SYSTEMS, ACIVS 2018, 2018, 11182 :620-631
[3]  
Ba J, 2014, ACS SYM SER
[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]   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
[6]  
Chen Ting, 2019, INT C MACH LEARN, DOI DOI 10.22489/CINC.2017.065-469
[7]   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
[8]  
Cheng D., 2022, ARXIV
[9]   U2D2Net: Unsupervised Unified Image Dehazing and Denoising Network for Single Hazy Image Enhancement [J].
Ding, Bosheng ;
Zhang, Ruiheng ;
Xu, Lixin ;
Liu, Guanyu ;
Yang, Shuo ;
Liu, Yumeng ;
Zhang, Qi .
IEEE TRANSACTIONS ON MULTIMEDIA, 2024, 26 :202-217
[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