国内刊号:43-1258/TP
国际刊号:1007-130X
发布日期:
作者:徐雯, 于瓅
单位:(安徽理工大学计算机科学与工程学院,安徽 淮南 232001)
关键词:压缩感知,深度学习,注意力机制,特征增强,
基金:安徽省重点研究与开发计划(202104d07020010)
针对基于深度学习的压缩感知重构算法中存在的图像重构细化程度低、网络泛化能力弱的问题,提出基于迭代收缩阈值与深度学习的压缩感知图像重构网络模型EH-ISTANet。该模型由采样子网、初始化子网和增强重构子网3部分组成,添加注意力机制并配合邻近映射模块将得到的特征送入增强模块中对重构图像的边缘和纹理进行增强。重构阶段模仿传统迭代收缩阈值算法的展开过程,每个阶段可以灵活地模拟测量矩阵,并在梯度下降步骤中动态地调整步长。经验证,该模型在不同数据集、不同采样率下的峰值信噪比均有所提升,表明其在提高泛化能力和重构精度方面优于其他模型。压缩感知比率为10%时,该模型在5种测试集上的平均信噪比比CSNet、AMP-Net和AMP-Net-BM模型平均提高了1.69 dB、4.36 dB和1.93 dB。
来源:2025年第3期
《计算机工程与科学》期刊编辑部