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 条
[51]   Uformer: A General U-Shaped Transformer for Image Restoration [J].
Wang, Zhendong ;
Cun, Xiaodong ;
Bao, Jianmin ;
Zhou, Wengang ;
Liu, Jianzhuang ;
Li, Houqiang .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :17662-17672
[52]   FISTA-CSNet: a deep compressed sensing network by unrolling iterative optimization algorithm [J].
Xin, Liqi ;
Wang, Dingwen ;
Shi, Wenxuan .
VISUAL COMPUTER, 2023, 39 (09) :4177-4193
[53]   Image compressive sensing recovery via group residual based nonlocal low-rank regularization [J].
Xu, Jin ;
Fu, Zhizhong ;
Zhu, Xiaodong .
NEUROCOMPUTING, 2021, 449 :315-329
[54]   ADMM-CSNet: A Deep Learning Approach for Image Compressive Sensing [J].
Yang, Yan ;
Sun, Jian ;
Li, Huibin ;
Xu, Zongben .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2020, 42 (03) :521-538
[55]   Dynamic proximal unrolling network for compressive imaging [J].
Yang, Yixiao ;
Tao, Ran ;
Wei, Kaixuan ;
Fu, Ying .
NEUROCOMPUTING, 2022, 510 :203-217
[56]   DR2-Net: Deep Residual Reconstruction Network for image compressive sensing [J].
Yao, Hantao ;
Dai, Feng ;
Zhang, Shiliang ;
Zhang, Yongdong ;
Tian, Qi ;
Xu, Changsheng .
NEUROCOMPUTING, 2019, 359 :483-493
[57]   CSformer: Bridging Convolution and Transformer for Compressive Sensing [J].
Ye, Dongjie ;
Ni, Zhangkai ;
Wang, Hanli ;
Zhang, Jian ;
Wang, Shiqi ;
Kwong, Sam .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2023, 32 :2827-2842
[58]   COAST: COntrollable Arbitrary-Sampling NeTwork for Compressive Sensing [J].
You, Di ;
Zhang, Jian ;
Xie, Jingfen ;
Chen, Bin ;
Ma, Siwei .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :6066-6080
[59]   Restormer: Efficient Transformer for High-Resolution Image Restoration [J].
Zamir, Syed Waqas ;
Arora, Aditya ;
Khan, Salman ;
Hayat, Munawar ;
Khan, Fahad Shahbaz ;
Yang, Ming-Hsuan .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :5718-5729
[60]  
Zeng RJ, 2024, Arxiv, DOI arXiv:2411.01327