扩散模型与双注意力交替优化的图像压缩感知
DOI:
CSTR:
作者:
作者单位:

1.长江大学计算机科学学院 荆州 434023; 2.长江大学人工智能科研平台 荆州 434023

作者简介:

通讯作者:

中图分类号:

TP391.41; TN911.73

基金项目:

湖北省自然科学基金青年项目(2024AFB345)、湖北省自然科学基金(2024AFB851)项目资助


Diffusion model and dual-attention alternating optimization for image compressed sensing reconstruction
Author:
Affiliation:

1.School of Computer Science,Yangtze University, Jingzhou 434023, China; 2.Artificial Intelligence Research Platform,Yangtze University, Jingzhou 434023, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对当前基于深度展开网络的图像压缩感知重建方法存在低采样率下细节恢复困难、历史特征信息利用不足以及块效应的问题,提出了一种基于扩散模型与双注意力交替优化的图像压缩感知重建网络(diffusion and dual-attention alternating optimization network,DDANet)。该网络将扩散模型与深度展开框架相结合,通过设计扩散去噪与双注意力交替迭代机制,实现数据分布先验与跨阶段特征融合的协同增强,改善低采样率下的恢复质量。双注意力机制包括跨阶段通道注意力(cross-stage channel attention,CSCA)模块和窗口稀疏注意力(window sparse attention,WSA)模块,通过融合跨阶段特征与挖掘局部自相似性提升图像重建细节。针对块效应提出了坐标空间自校正(coordinate spatial self-correction,CSSC)模块,通过空间、坐标和自校准三分支协同作用,抑制块边界不连续伪影,保证重建图像的全局一致性。实验结果表明,在Set11测试集上,当采样率为1%时,所提方法在峰值信噪比(peak signal-to-noise ratio,PSNR)指标上相比同为扩散模型的扩散消息传递深度展开网络(diffusion message passing deep unfolding networks,DMP-DUN)提升了0.61 dB,相比基于优化思想的交叉注意力 Transformer 展开框架(optimization-inspired cross-attention Transformer unfolding framework,OTCUF)提升了1.31 dB;当采样率为10%时,PSNR相比OTCUF和不动点延拓展开网络(unrolling fixed-point continuous network,UFC-Net)分别提升了1.22和1.77 dB,表明该方法在低采样率条件下具有显著优势。

    Abstract:

    Aiming at the problems of difficulty in detail recovery at low sampling rates, insufficient utilization of historical feature information and blocking artifacts in current image compressed sensing reconstruction methods based on deep unfolding networks, this paper proposes a diffusion and dual-attention alternating optimization network (DDANet) for image compressed sensing reconstruction. The network combines the diffusion model with the deep unfolding framework by designing an alternating iteration mechanism between the diffusion denoising sub-stage and dual-attention sub-stage, which achieves collaborative enhancement of data distribution priors and cross-stage feature fusion, improving the recovery quality at low sampling rates. The dual-attention sub-stage includes a cross-stage channel attention (CSCA) module and a window sparse attention (WSA) module, which enhance image reconstruction details by fusing cross-stage features and exploiting local self-similarity. To address blocking artifacts, this paper proposes a coordinate spatial self-correction (CSSC) module, which suppresses block boundary discontinuity artifacts through the collaborative action of spatial, coordinate, and self-calibration branches, ensuring global consistency of reconstructed images. Experimental results show that on the Set11 test set, when the sampling rate is 1%, the proposed method improves peak signal-to-noise ratio (PSNR) by 0.61 dB compared with diffusion message passing deep unfolding networks (DMP-DUN) which also uses diffusion model, and by 1.31 dB compared with optimization-inspired cross-attention transformer unfolding framework(OTCUF); when the sampling rate is 10%, PSNR improves by 1.22 and 1.77 dB compared with OTCUF and unrolling fixed-point continuous network(UFC-Net) respectively, indicating that the method has significant advantages at low sampling rates.

    参考文献
    相似文献
    引证文献
引用本文

黄家琪,王少芳.扩散模型与双注意力交替优化的图像压缩感知[J].电子测量技术,2026,49(15):218-228

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-10-23
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告