HD-Net: Hierarchical Distillation Network for High-Efficiency Single Image Deraining

Kejian Hu, Zhichen Zhang, Xiaowen Cai, Xiang Chen, Nanfeng Jiang, Yu Zhou, Tiesong Zhao*

*Corresponding author for this work

Research output: Chapters, Conference Papers, Creative and Literary WorksRGC 32 - Refereed conference paper (with host publication)peer-review

2 Citations (Scopus)

Abstract

Rain streaks usually result in severe image visual degradation and foreground occlusion, affecting the quality of computer tasks in outdoor scenes. Currently, the mainstream methods in single-image deraining are based on data-driven. However, the deep learning network could be imperfect, with limited power for learning the global information from rain streaks all over the map. In order to solve this problem, we proposed a novel Hierarchical Distillation Network (HDNet). In this network, Hierarchical Feature Extraction Block (HFEB) can fully utilize the Transformer's learning ability in high-level features, integrate local detail extraction and global structure representation, and compensate for the weakness of the Convolutional Neural Network (CNN), which is overattentive to the underlying image features. Furthermore, the Distillation-Calibration Block (DCB) are adopted to avoid feature redundancy during model training and calibrate the channel and spatial information through the feature transmission, which could significantly improve the learning efficiency. Finally, the experiment results show that our model performs better than traditional CNN models and state-of-the-art methods. © 2022 IEEE.
Original languageEnglish
Title of host publication2022 IEEE 24th International Workshop on Multimedia Signal Processing (MMSP)
PublisherIEEE
ISBN (Electronic)978-1-6654-7189-3
DOIs
Publication statusPublished - 2022
EventIEEE 24th International Workshop on Multimedia Signal Processing (MMSP) -
Duration: 26 Sept 202228 Sept 2022

Publication series

NameIEEE International Workshop on Multimedia Signal Processing
ISSN (Print)2163-3517

Conference

ConferenceIEEE 24th International Workshop on Multimedia Signal Processing (MMSP)
Period26/09/2228/09/22

Research Keywords

  • Image deraining
  • Transfomer
  • Feature distillation
  • Coordinate calibration

Fingerprint

Dive into the research topics of 'HD-Net: Hierarchical Distillation Network for High-Efficiency Single Image Deraining'. Together they form a unique fingerprint.

Cite this