HIDFlowNet: A Flow-Based Deep Network for Hyperspectral Image Denoising

TL;DR

HIDFlowNet introduces a flow-based deep learning model for hyperspectral image denoising that explicitly models the distribution of clean images conditioned on noisy inputs, effectively addressing the ill-posed nature of the task.

Contribution

The paper proposes a novel flow-based network that decouples low- and high-frequency information and samples diverse clean HSIs, improving over deterministic methods.

Findings

Outperforms state-of-the-art methods on simulated datasets.

Achieves comparable results on real datasets.

Effectively models the distribution of clean HSIs.

Abstract

Hyperspectral image (HSI) denoising is essentially ill-posed since a noisy HSI can be degraded from multiple clean HSIs. However, existing deep learning (DL)-based approaches only restore one clean HSI from the given noisy HSI with a deterministic mapping, thus ignoring the ill-posed issue and always resulting in an over-smoothing problem. Additionally, these DL-based methods often neglect that noise is part of the high-frequency component and their network architectures fail to decouple the learning of low-frequency and high-frequency. To alleviate these issues, this paper proposes a flow-based HSI denoising network (HIDFlowNet) to directly learn the conditional distribution of the clean HSI given the noisy HSI and thus diverse clean HSIs can be sampled from the conditional distribution. Overall, our HIDFlowNet is induced from the generative flow model and is comprised of an invertible decoder and a conditional encoder, which can explicitly decouple the learning of low-frequency and high-frequency information of HSI. Specifically, the invertible decoder is built by staking a succession of invertible conditional blocks (ICBs) to capture the local high-frequency details. The conditional encoder utilizes down-sampling operations to obtain low-resolution images and uses transformers to capture correlations over a long distance so that global low-frequency information can be effectively extracted. Extensive experiments on simulated and real HSI datasets verify that our proposed HIDFlowNet can obtain better or comparable results compared with other state-of-the-art methods.