arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.04024cs.CV

面向高光谱图像重建的自适应步长感知展开网络与非局部混合注意力

Adaptive Step-size Perception Unfolding Network with Non-local Hybrid Attention for Hyperspectral Image Reconstruction

School of Mathematical Sciences, Nanjing Normal University(南京师范大学数学科学学院)

Yanan Yang, Like Xin

AI总结:

本文提出基于FISTA的自适应步长感知展开网络ASPUN,通过逐通道步长估计和非局部混合注意力Transformer,在高光谱图像重建中超越现有SOTA方法。

摘要:

深度展开方法和Transformer架构近年来在高光谱图像(HSI)重建中展现出良好效果,但仍存在两个问题:(1)在数据子问题中,大多数方法用可学习参数表示步长,然而不同光谱通道的特征与真实值之间的误差并不相等。(2)Transformer难以在感受野大小与像素级细节信息之间取得平衡。为克服上述不足,我们提出了一种自适应步长感知展开网络(ASPUN),这是一种基于FISTA算法的深度展开网络,利用自适应步长感知模块估计每个光谱通道的更新步长。此外,我们设计了一个非局部混合注意力Transformer(NHAT)模块,以充分利用Transformer的感受野优势。通过将NLHA嵌入非局部信息聚合(NLIA)模块,展开网络能够获得更好的重建结果。实验结果表明,我们的ASPUN优于现有SOTA算法,并取得了最佳性能。

英文摘要:

Deep unfolding methods and transformer architecture have recently shown promising results in hyperspectral image (HSI) reconstruction. However, there still exist two issues: (1) in the data subproblem, most methods represents the stepsize utilizing a learnable parameter. Nevertheless, for different spectral channel, error between features and ground truth is unequal. (2) Transformer struggles to balance receptive field size with pixel-wise detail information. To overcome the aforementioned drawbacks, We proposed an adaptive step-size perception unfolding network (ASPUN), a deep unfolding network based on FISTA algorithm, which uses an adaptive step-size perception module to estimate the update step-size of each spectral channel. In addition, we design a Non-local Hybrid Attention Transformer(NHAT) module for fully leveraging the receptive field advantage of transformer. By plugging the NLHA into the Non-local Information Aggregation (NLIA) module, the unfolding network can achieve better reconstruction results. Experimental results show that our ASPUN is superior to the existing SOTA algorithms and achieves the best performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑