Mutual learning for domain adaptation: Self-distillation image dehazing network with sample-cycle

被引:0
作者
Chen, Erkang [1 ]
Tong, Lihan [1 ]
Ye, Tian [3 ]
Chen, Sixiang [3 ]
Zhang, Yunchen [4 ]
Liu, Yun [2 ,5 ,6 ]
机构
[1] Jimei Univ, Sch Ocean Informat Engn, Xiamen 361021, Peoples R China
[2] Chongqing Coll Int Business & Econ, Big Data & Intelligence Engn Sch, Chongqing 401520, Peoples R China
[3] Hong Kong Univ Sci & Technol Guangzhou, Guangzhou 511458, Peoples R China
[4] Fujian Normal Univ, Coll Photon & Elect Engn, Fuzhou 350007, Peoples R China
[5] Southwest Univ, Coll Artificial Intelligence, Chongqing 400715, Peoples R China
[6] Nanyang Technol Univ, Coll Comp & Data Sci, Singapore 639798, Singapore
基金
中国国家自然科学基金;
关键词
Image dehazing; Domain adaptation; Self-distillation; Sample-cycle; QUALITY ASSESSMENT; VISION;
D O I
10.1016/j.displa.2024.102904
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Deep learning-based methods have made significant achievements for image dehazing. However, most of existing dehazing networks focus on training models using synthetic hazy images, resulting in generalization performance degradation when applying on real-world hazy images because of the domain shift problem. In this paper, we propose a mutual learning dehazing framework for domain adaption. Specifically, we first devise two siamese networks: a teacher network in the synthetic domain and a student network in the real domain, and then optimize them in a mutual learning manner by leveraging the exponential moving average (EMA) and joint loss. Moreover, we design a sample-cycle strategy based on haze density augmentation (HDA) module to introduce pseudo real-world image pairs provided by the student network into training for further improving the generalization performance. Extensive experiments demonstrate that the proposed framework outperforms state-of-the-art dehazing techniques in terms of subjective and objective evaluation.
引用
收藏
页数:8
相关论文
共 49 条
  • [1] I-HAZE: A Dehazing Benchmark with Real Hazy and Haze-Free Indoor Images
    Ancuti, Cosmin
    Ancuti, Codruta O.
    Timofte, Radu
    De Vleeschouwer, Christophe
    [J]. ADVANCED CONCEPTS FOR INTELLIGENT VISION SYSTEMS, ACIVS 2018, 2018, 11182 : 620 - 631
  • [2] Single Image Dehazing Using Haze-Lines
    Berman, Dana
    Treibitz, Tali
    Avidan, Shai
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2020, 42 (03) : 720 - 734
  • [3] Non-Local Image Dehazing
    Berman, Dana
    Treibitz, Tali
    Avidan, Shai
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 1674 - 1682
  • [4] DehazeNet: An End-to-End System for Single Image Haze Removal
    Cai, Bolun
    Xu, Xiangmin
    Jia, Kui
    Qing, Chunmei
    Tao, Dacheng
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2016, 25 (11) : 5187 - 5198
  • [5] CHARBONNIER P, 1994, IEEE IMAGE PROC, P168
  • [6] Degradation-adaptive neural network for jointly single image dehazing and desnowing
    Chen, Erkang
    Chen, Sixiang
    Ye, Tian
    Liu, Yun
    [J]. FRONTIERS OF COMPUTER SCIENCE, 2024, 18 (02)
  • [7] PSD: Principled Synthetic-to-Real Dehazing Guided by Physical Priors
    Chen, Zeyuan
    Wang, Yangchao
    Yang, Yang
    Liu, Dong
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 7176 - 7185
  • [8] Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
  • [9] Multi-Scale Boosted Dehazing Network with Dense Feature Fusion
    Dong, Hang
    Pan, Jinshan
    Xiang, Lei
    Hu, Zhe
    Zhang, Xinyi
    Wang, Fei
    Yang, Ming-Hsuan
    [J]. 2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, : 2154 - 2164
  • [10] Image Dehazing Transformer with Transmission-Aware 3D Position Embedding
    Guo, Chunle
    Yan, Qixin
    Anwar, Saeed
    Cong, Runmin
    Ren, Wenqi
    Li, Chongyi
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, : 5802 - 5810