DSU-Net: A Dynamic Stage Unfolding Network for high-noise image compressive sensing denoising

被引:1
作者
Zhang, Jie [1 ]
Lu, Miaoxin [1 ]
Huang, Wenxiao [1 ]
Shi, Xiaoping [2 ]
Wang, Yanfeng [1 ]
机构
[1] Zhengzhou Univ Light Ind, Coll Elect Informat Engn, Zhengzhou 450002, Peoples R China
[2] Harbin Inst Technol, Sch Astronaut, Harbin 150001, Peoples R China
基金
中国国家自然科学基金;
关键词
Compressive sensing; Image denoising; Transformer; Deep unfolding network; TRANSFORMER; ALGORITHM; SPARSE; MRI;
D O I
10.1016/j.neucom.2024.129071
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep unfolding networks (DUNs) have demonstrated considerable efficacy in the domain of compressive sensing (CS), attributed to their superior performance and interpretability. Nonetheless, many existing CS methodologies fail to account for the influence of noise prior to image sampling, resulting in blurring and distortion in the reconstructed images. To address these issues, this paper proposes a dynamic stage unfolding network (DSUNet). Firstly, a novel dynamic stage unfolding mechanism is proposed to achieve feature refinement by dynamically optimizing various noisy images in a sequential manner. Secondly, a step inertia fusion module (SIFM) is developed to execute multi-tiered information fusion from adjacent stages, thereby promoting feature reuse and minimizing the loss of detailed information. Finally, a step cross transformer denoiser (SCTD) is designed to capture correlations between distant pixels, effectively addressing the limitations associated with local attributes and significantly improving image reconstruction performance. Comprehensive experimental evaluations indicate that the proposed DSUNet achieves outstanding performance under both low CS ratios and high noise levels, successfully addressing the challenges of denoising and reconstruction in high-resolution and high-noise images. Codes are available at https://github.com/zzuli407/DSU-Net.
引用
收藏
页数:11
相关论文
共 69 条
  • [1] NTIRE 2017 Challenge on Single Image Super-Resolution: Dataset and Study
    Agustsson, Eirikur
    Timofte, Radu
    [J]. 2017 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2017, : 1122 - 1131
  • [2] Contour Detection and Hierarchical Image Segmentation
    Arbelaez, Pablo
    Maire, Michael
    Fowlkes, Charless
    Malik, Jitendra
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2011, 33 (05) : 898 - 916
  • [3] A Fast Iterative Shrinkage-Thresholding Algorithm for Linear Inverse Problems
    Beck, Amir
    Teboulle, Marc
    [J]. SIAM JOURNAL ON IMAGING SCIENCES, 2009, 2 (01): : 183 - 202
  • [4] Low-Complexity Single-Image Super-Resolution based on Nonnegative Neighbor Embedding
    Bevilacqua, Marco
    Roumy, Aline
    Guillemot, Christine
    Morel, Marie-Line Alberi
    [J]. PROCEEDINGS OF THE BRITISH MACHINE VISION CONFERENCE 2012, 2012,
  • [5] A First-Order Primal-Dual Algorithm for Convex Problems with Applications to Imaging
    Chambolle, Antonin
    Pock, Thomas
    [J]. JOURNAL OF MATHEMATICAL IMAGING AND VISION, 2011, 40 (01) : 120 - 145
  • [6] FSOINET: FEATURE-SPACE OPTIMIZATION-INSPIRED NETWORK FOR IMAGE COMPRESSIVE SENSING
    Chen, Wenjun
    Yang, Chunling
    Yang, Xin
    [J]. 2022 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2022, : 2460 - 2464
  • [7] Deep Unfolding Network for Image Compressed Sensing by Content-Adaptive Gradient Updating and Deformation-Invariant Non-Local Modeling
    Cui, Wenxue
    Fan, Xiaopeng
    Zhang, Jian
    Zhao, Debin
    [J]. IEEE TRANSACTIONS ON MULTIMEDIA, 2024, 26 : 4012 - 4027
  • [8] Cui WX, 2018, 2018 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), P1748, DOI 10.1109/ICASSP.2018.8461766
  • [9] Dong SH, 2023, Arxiv, DOI arXiv:2312.00360
  • [10] Dong WS, 2011, PROC CVPR IEEE, P457, DOI 10.1109/CVPR.2011.5995478