|
|
基于邻近梯度算法展开的压缩感知双域学习网络
|
Abstract:
算法展开网络在压缩感知图像重建应用中取得了巨大成功，但这些网络中未充分挖掘和利用图像的通道及空间信息的深层学习潜力，同时在重建精度和成本控制等方面都有待进一步深入探索和改进。为了实现对图像的迅速采样并从有限采样数据中准确重建图像，本文提出了一种基于邻近梯度算法展开的双域学习网络PGD-DDLN。该网络将邻近梯度算法的两步更新迭代分别展开到深度网络架构中，并在网络中加入了对图像通道和空间信息的双域学习过程。大量实验表明，我们的PGD-DDLN网络在定量指标和视觉质量方面都达到较为先进的结果。
Algorithm unfolding networks have achieved great success in compressive sensing image reconstruc-tion applications, yet these networks have not fully exploited and leveraged the deep learning potential of image channel and spatial information. Additionally, there is still a need for further exploration and improvement in areas such as reconstruction accuracy and cost control. To achieve rapid image sampling and accurate image reconstruction from limited sampled data, this paper proposes a Dual-Domain Learning Network based on the Unfolded Proximal Gradient Algorithm, termed PGD-DDLN. This network unfolds the two-step update iterations of the proximal gradient algorithm into a deep network architecture and incorporates a dual-domain learning process for image channel and spatial information. Extensive experiments demonstrate that our PGD-DDLN network achieves state-of-the-art results in both quantitative metrics and visual quality.
| [1] | Kulkarni, K., Lohit, S., Turaga, P., Kerviche, R. and Ashok, A. (2016). ReconNet: Non-Iterative Reconstruction of Images from Compressively Sensed Measurements. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, 27-30 June 2016, 449-458. https://doi.org/10.1109/cvpr.2016.55 |
| [2] | Chen, Z., Cong, R., Xu, Q. and Huang, Q. (2021) DpaNet: Depth Potentiality-Aware Gated Attention Network for RGB-D Salient Object Detection. IEEE Transactions on Image Processing, 30, 7012-7024. https://doi.org/10.1109/tip.2020.3028289 |
| [3] | Gregor, K. and Le Cun, Y. (2010) Learning Fast Approximations of Sparse Coding. Proceedings of the 27th International Conference on International Conference on Machine Learning, Haifa, 21-24 June 2010, 399-406. |
| [4] | Yang, Y., Sun, J., Li, H. and Xu, Z. (2020) ADMM-CSNet: A Deep Learning Approach for Image Compressive Sensing. IEEE Transactions on Pattern Analysis and Machine Intelligence, 42, 521-538. https://doi.org/10.1109/tpami.2018.2883941 |
| [5] | Zhang, J. and Ghanem, B. (2018) ISTA-Net: Interpretable Optimization-Inspired Deep Network for Image Compressive Sensing. 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, 18-23 June 2018, 1828-1837. https://doi.org/10.1109/cvpr.2018.00196 |
| [6] | Song, J., Chen, B. and Zhang, J. (2023) Dynamic Path-Controllable Deep Unfolding Network for Compressive Sensing. IEEE Transactions on Image Processing, 32, 2202-2214. https://doi.org/10.1109/tip.2023.3263100 |
| [7] | You, D., Xie, J. and Zhang, J. (2021) ISTA-Net++: Flexible Deep Unfolding Network for Compressive Sensing. 2021 IEEE International Conference on Multimedia and Expo (ICME), Shenzhen, 5-9 July 2021, 1-6. https://doi.org/10.1109/icme51207.2021.9428249 |
| [8] | Kulkarni, K., Lohit, S., Turaga, P., Kerviche, R. and Ashok, A. (2016) ReconNet: Non-Iterative Reconstruction of Images from Compressively Sensed Measurements. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, 27-30 June 2016, 449-458. https://doi.org/10.1109/cvpr.2016.55 |
| [9] | Martin, D., Fowlkes, C., Tal, D. and Malik, J. (2001) A Database of Human Segmented Natural Images and Its Application to Evaluating Segmentation Algorithms and Measuring Ecological Statistics. Proceedings Eighth IEEE International Conference on Computer Vision, Vancouver, 7-14 July 2001, 416-423. https://doi.org/10.1109/iccv.2001.937655 |
| [10] | Song, J., Chen, B. and Zhang, J. (2021) Memory-Augmented Deep Unfolding Network for Compressive Sensing. Proceedings of the 29th ACM International Conference on Multimedia, New York, 20-24 October 2021, 4249-4258. |
