arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.02149eess.IVcs.CV
收藏

HDNet:用于光谱压缩成像的高分辨率双域学习

HDNet: High-resolution Dual-domain Learning for Spectral Compressive Imaging

Shenzhen International Graduate School, Tsinghua University(清华大学深圳国际研究生院);Shenzhen Institute of Future Media Technology(深圳未来媒体技术研究院);Westlake University(西湖大学);ETH Zurich(苏黎世联邦理工学院)

Xiaowan Hu, Yuanhao Cai, Jing Lin, Haoqian Wang, Xin Yuan, Yulun Zhang, Radu Timofte, Luc Van Gool

摘要:

深度学习的快速发展为高光谱图像的端到端重建提供了更好的解决方案。然而,现有的基于学习的方法有两个主要缺陷。首先,具有自我注意力的网络通常会牺牲内部分辨率来平衡模型性能和复杂性,从而失去细粒度的高分辨率(HR)特征。其次,即使聚焦于空间-谱域学习(SDL)的优化收敛到理想解,重构的HSI与真实值之间仍然存在显著的视觉差异。因此,我们提出了一个高分辨率的双域学习网络(HDNet)的HSI重建。一方面,所提出的HR空间-光谱注意模块以其高效的特征融合提供了连续和精细的像素级特征。另一方面,频域学习(FDL)被引入HSI重建,以缩小频域差异。动态FDL监督迫使模型重建细粒度的频率,并补偿像素级损失造成的过度平滑和失真。在我们的HDNet中,HR像素级注意力和频率级细化相互促进HSI感知质量。大量的定量和定性评估实验表明,我们的方法实现SOTA性能模拟和真实的HSI数据集。代码和模型将在https://github.com/caiyuanhao1998/MST上发布

英文摘要:

The rapid development of deep learning provides a better solution for the end-to-end reconstruction of hyperspectral image (HSI). However, existing learning-based methods have two major defects. Firstly, networks with self-attention usually sacrifice internal resolution to balance model performance against complexity, losing fine-grained high-resolution (HR) features. Secondly, even if the optimization focusing on spatial-spectral domain learning (SDL) converges to the ideal solution, there is still a significant visual difference between the reconstructed HSI and the truth. Therefore, we propose a high-resolution dual-domain learning network (HDNet) for HSI reconstruction. On the one hand, the proposed HR spatial-spectral attention module with its efficient feature fusion provides continuous and fine pixel-level features. On the other hand, frequency domain learning (FDL) is introduced for HSI reconstruction to narrow the frequency domain discrepancy. Dynamic FDL supervision forces the model to reconstruct fine-grained frequencies and compensate for excessive smoothing and distortion caused by pixel-level losses. The HR pixel-level attention and frequency-level refinement in our HDNet mutually promote HSI perceptual quality. Extensive quantitative and qualitative evaluation experiments show that our method achieves SOTA performance on simulated and real HSI datasets. Code and models will be released at https://github.com/caiyuanhao1998/MST

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

↑