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 条
[21]   AOD-Net: All-in-One Dehazing Network [J].
Li, Boyi ;
Peng, Xiulian ;
Wang, Zhangyang ;
Xu, Jizheng ;
Feng, Dan .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :4780-4788
[22]   All-In-One Image Restoration for Unknown Corruption [J].
Li, Boyun ;
Liu, Xiao ;
Hu, Peng ;
Wu, Zhongqin ;
Lv, Jiancheng ;
Peng, Xi .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :17431-17441
[23]   USID-Net: Unsupervised Single Image Dehazing Network via Disentangled Representations [J].
Li, Jiafeng ;
Li, Yaopeng ;
Zhuo, Li ;
Kuang, Lingyan ;
Yu, Tianjian .
IEEE TRANSACTIONS ON MULTIMEDIA, 2023, 25 :3587-3601
[24]   GridDehazeNet plus : An Enhanced Multi-Scale Network With Intra-Task Knowledge Transfer for Single Image Dehazing [J].
Liu, Xiaohong ;
Shi, Zhihao ;
Wu, Zijun ;
Chen, Jun ;
Zhai, Guangtao .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2023, 24 (01) :870-884
[25]  
McCartney E. J., 1976, Optics of the atmosphere. Scattering by molecules and particles
[26]   Contrast restoration of weather degraded images [J].
Narasimhan, SG ;
Nayar, SK .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2003, 25 (06) :713-724
[27]   Vision and the atmosphere [J].
Narasimhan, SG ;
Nayar, SK .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2002, 48 (03) :233-254
[28]  
Paszke A., 2016, ARXIV PREPRINT ARXIV
[29]  
Paszke A, 2019, ADV NEUR IN, V32
[30]   U-Net: Convolutional Networks for Biomedical Image Segmentation [J].
Ronneberger, Olaf ;
Fischer, Philipp ;
Brox, Thomas .
MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION, PT III, 2015, 9351 :234-241