首页|MAUN:Memory-Augmented Deep Unfolding Network for Hyperspectral Image Reconstruction

MAUN:Memory-Augmented Deep Unfolding Network for Hyperspectral Image Reconstruction

扫码查看
Spectral compressive imaging has emerged as a powerful technique to collect the 3D spectral information as 2D measurements.The algorithm for restoring the original 3D hyperspectral images(HSIs)from compressive measurements is pivotal in the imaging process.Early approaches painstakingly designed networks to directly map compressive measurements to HSIs,resulting in the lack of interpretability without ex-ploiting the imaging priors.While some recent works have introduced the deep unfolding framework for explainable recon-struction,the performance of these methods is still limited by the weak information transmission between iterative stages.In this paper,we propose a Memory-Augmented deep Unfolding Network,termed MAUN,for explainable and accurate HSI reconstruction.Specifically,MAUN implements a novel CNN scheme to facilitate a better extrapolation step of the fast iterative shrinkage-thresholding algorithm,introducing an extra momentum incorporation step for each iteration to alleviate the information loss.Moreover,to exploit the high correlation of intermediate images from neighboring iterations,we customize a cross-stage transformer(CSFormer)as the deep denoiser to simultaneously capture self-similarity from both in-stage and cross-stage features,which is the first attempt to model the long-distance dependencies between iteration stages.Extensive experiments demonstrate that the proposed MAUN is superior to other state-of-the-art methods both visually and metrically.Our code is publicly available at https://github.com/HuQ1an/MAUN.

Qian Hu、Jiayi Ma、Yuan Gao、Junjun Jiang、Yixuan Yuan

展开 >

Electronic Information School,Wuhan University,Wuhan 430072,China

School of Computer Science and Technology,Harbin In-stitute of Technology,Harbin 150001,China

Department of Electronic Engineering,Chi-nese University of Hong Kong,Hong Kong 999077,China

国家自然科学基金

62276192

2024

相关文献
同作者其他文献同项目成果
自动化学报(英文版)
中国自动化学会,中国科学院自动化研究所,中国科技出版传媒股份有限公司

自动化学报(英文版)

CSTPCDEI
ISSN:2329-9266
年,卷(期):2024.11(5)
Qian Hu,Jiayi Ma,Yuan Gao,et al.MAUN:Memory-Augmented Deep Unfolding Network for Hyperspectral Image Reconstruction[J].自动化学报(英文版),2024,11(5):1139-1150.DOI:10.1109/JAS.2024.124362.
  • 47