通过低秩分解提升光谱压缩成像的效率与质量
LRDUN: A Low-Rank Deep Unfolding Network for Efficient Spectral Compressive Imaging
- 将低秩分解融入感知模型,分两步重构光谱基与子空间图像
- 相比全量重建,计算成本大幅降低且图像质量达到顶尖水平
- 适合关注高效重建与物理先验融合的研究者
深度展开网络(DUNs)在光谱压缩成像(SCI)重建中取得显著进展,成为主流范式。现有DUNs基于全高光谱图像(HSI)成像模型,每阶段直接处理高维HSI数据,仅凭单一2D编码测量逐步优化整个数据立方体。但该方法存在计算冗余,并面临将2D残差映射回3D HSI空间的病态问题。本文提出两种新成像模型:分别对应光谱基和子空间图像,显式将低秩(LR)分解融入传感模型。相较于恢复完整HSI,估计这些紧凑的低维成分可显著缓解病态性。基于此,我们构建了低秩深度展开网络(LRDUN),在展开的近端梯度下降(PGD)框架下联合求解两个子问题。此外,引入广义特征展开机制(GFUM),使数据保真项中的物理秩与先验模块的特征维度解耦,增强网络表征能力与灵活性。在模拟与真实数据集上的大量实验表明,所提LRDUN在保持最先进(SOTA)重建质量的同时,显著降低了计算成本。
原文摘要 · Abstract (English)
Deep unfolding networks (DUNs) have achieved remarkable success and become the mainstream paradigm for spectral compressive imaging (SCI) reconstruction. Existing DUNs are derived from full-HSI imaging models, where each stage operates directly on the high-dimensional HSI, refining the entire data cube based on the single 2D coded measurement. However, this paradigm leads to computational redundancy and suffers from the ill-posed nature of mapping 2D residuals back to 3D space of HSI. In this paper, we propose two novel imaging models corresponding to the spectral basis and subspace image by explicitly integrating low-rank (LR) decomposition with the sensing model. Compared to recovering the full HSI, estimating these compact low-dimensional components significantly mitigates the ill-posedness. Building upon these novel models, we develop the Low-Rank Deep Unfolding Network (LRDUN), which jointly solves the two subproblems within an unfolded proximal gradient descent (PGD) framework. Furthermore, we introduce a Generalized Feature Unfolding Mechanism (GFUM) that decouples the physical rank in the data-fidelity term from the feature dimensionality in the prior module, enhancing the representational capacity and flexibility of the network. Extensive experiments on simulated and real datasets demonstrate that the proposed LRDUN achieves state-of-the-art (SOTA) reconstruction quality with significantly reduced computational cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。