压缩感知重构算法的两步深度展开策略研究  

Two-step deep unfolding strategy for compressed sensing reconstruction algorithms

作　　者：邵凯[1,2,3] 闫力力 王光宇[1,2,3] SHAO Kai;YAN Lili;WANG Guangyu(School of Communication and Information Engineering,Chongqing University of Posts and Telecommunications,Chongqing 400065,China;Chongqing Key Laboratory of Mobile Communications Technology,Chongqing 400065,China;Engineering Research Center of Mobile Communications,Ministry of Education,Chongqing 400065,China)

机构地区：[1]重庆邮电大学通信与信息工程学院,重庆400065 [2]重庆邮电大学移动通信技术重庆市重点实验室,重庆400065 [3]重庆邮电大学移动通信教育部工程研究中心,重庆400065

出　　处：《智能系统学报》2023年第5期1117-1126,共10页CAAI Transactions on Intelligent Systems

摘　　要：针对压缩感知中重构算法的深度展开问题,提出了一种两步深度展开策略(two-step deep unfolding,TwDU)。已有深度展开重构算法通常依赖前一步估计值估计当前值,TwDU对已有深度展开重构算法增加估计深度,依赖于前两步估计值估计当前展开值。TwDU对已有深度展开算法前两步估计值增加了两个训练权重。训练权重优化利用了信号估计值之间的相关特性,可以随着数据的特性自我学习和调整,所提TwDU策略应用于可学习迭代软阈值算法(learned iterative soft thresholding algorithm,LISTA)、可训练迭代软阈值算法(trainable iterative soft thresholding algorithm,TISTA)、可学习近似消息传递算法(learned approximate message passing,LAMP)等已有深度展开算法。通过在一维和二维稀疏信号的仿真验证,TwDU策略在重构精度和收敛速度上都更具有明显优势。A two-step deep unfolding(TwDU)strategy is put forward for the deep unfolding of reconstruction al-gorithms in compressed sensing.The existing deep unfolding reconstruction algorithms usually estimate the current value based on the previous one-step estimated value.TwDU increases the estimation depth for the existing deep unfold-ing reconstruction algorithms and estimates the current unfolding value based on the previous two-step estimation value.TwDU increases two training weights for the previous two-step estimation value in the existing deep unfolding recon-struction algorithm.The training weights are self-adaptive,which can learn and adjust following the changes in data characteristics by themselves and optimize and utilize the correlation among the estimated signal values.The proposed TwDU strategy is applied to the existing deep unfolding reconstruction algorithms,such as the learned iterative soft thresholding algorithm,learned approximate message passing algorithm,and trainable iterative soft thresholding al-gorithm.The simulation results in one-dimensional and two-dimensional sparse signals confirm that the TwDU strategy has obvious advantages regarding reconstruction accuracy and convergence speed.

关 键 词：压缩感知 稀疏信号 信号重构 深度学习 深度展开 模型驱动 迭代软阈值 近似消息传递算法 图像处理 

分 类 号：TP399[自动化与计算机技术—计算机应用技术] TN911[自动化与计算机技术—计算机科学与技术]