Abstract:Deep unfolding networks have gained increasing attention in the field of compressed sensing (CS) owing to their theoretical interpretability and superior reconstruction performance. However, most existing deep unfolding methods often face the following issues: 1) they learn directly from single-channel images, leading to a simple feature representation that does not fully capture complex features; and 2) they treat various image components uniformly, ignoring the characteristics of different components. To address these issues, we propose a novel wavelet-domain deep unfolding framework named WTDUN, which operates directly on the multi-scale wavelet subbands. Our method utilizes the intrinsic sparsity and multi-scale structure of wavelet coefficients to achieve a tree-structured sampling and reconstruction, effectively capturing and highlighting the most important features within images. Specifically, the design of tree-structured reconstruction aims to capture the inter-dependencies among the multi-scale subbands, enabling the identification of both fine and coarse features, which can lead to a marked improvement in reconstruction quality. Furthermore, a wavelet domain adaptive sampling method is proposed to greatly improve the sampling capability, which is realized by assigning measurements to each wavelet subband based on its importance. Unlike pure deep learning methods that treat all components uniformly, our method introduces a targeted focus on important subbands, considering their energy and sparsity. This targeted strategy lets us capture key information more efficiently while discarding less important information, resulting in a more effective and detailed reconstruction. Extensive experimental results on various datasets validate the superior performance of our proposed method.

Deep Unfolding for Compressed Sensing with Denoiser

DSU-Net: A Dynamic Stage Unfolding Network for High-Noise Image Compressive Sensing Denoising

CPP-Net: Embracing Multi-Scale Feature Fusion into Deep Unfolding CP-PPA Network for Compressive Sensing

Deep Unfolding Network for Image Compressed Sensing by Content-adaptive Gradient Updating and Deformation-invariant Non-local Modeling

D 3 U-Net: Dual-Domain Collaborative Optimization Deep Unfolding Network for Image Compressive Sensing

UFC-Net: Unrolling Fixed-point Continuous Network for Deep Compressive Sensing

WDU-Net: Wavelet-Guided Deep Unfolding Network for Image Compressed Sensing Reconstruction.

D3C2-Net: Dual-Domain Deep Convolutional Coding Network for Compressive Sensing

Dynamic Path-Controllable Deep Unfolding Network for Compressive Sensing

Fast Hierarchical Deep Unfolding Network for Image Compressed Sensing

High-Throughput Deep Unfolding Network for Compressive Sensing MRI

WTDUN: Wavelet Tree-Structured Sampling and Deep Unfolding Network for Image Compressed Sensing

ASAUN-CS: Adaptive Stage Activated Unfolding Network for Compressive Sensing

A Dictionary Learning Based Unsupervised Neural Network for Single Image Compressed Sensing

G2-DUN: Gradient Guided Deep Unfolding Network for Image Compressive Sensing

LR-CSNet: Low-Rank Deep Unfolding Network for Image Compressive Sensing

AMP-Net: Denoising based Deep Unfolding for Compressive Image Sensing

Optimization-inspired Cumulative Transmission Network for image compressive sensing

Deep Parameterized Neural Networks for Hyperspectral Image Denoising

Multi-Channel Representation Learning Enhanced Unfolding Multi-Scale Compressed Sensing Network for High Quality Image Reconstruction

DECS-NET: CONVOLUTIONAL SELF-ENCODING NETWORK FOR HYPERSPECTRAL IMAGE DENOISING