用特征图像桥接预训练RGB模型,提升高光谱图像超分辨率效果
EigenSR: Eigenimage-Bridged Pre-Trained RGB Learners for Single Hyperspectral Image Super-Resolution
- 先用特征图像微调预训练RGB模型,再通过迭代光谱正则化恢复光谱相关性
- 在多个数据集上优于现有方法,空间和光谱指标均达新高
- 适合缺乏高光谱数据的科研与遥感应用
单张高光谱图像超分辨率(single-HSI-SR)旨在提升单个低分辨率高光谱图像的分辨率。由于数据稀缺,该技术发展远落后于自然彩色图像超分辨率。近年来,基于大规模基准数据集预训练的模型在未见数据上表现优异,或可缓解高光谱数据不足问题。但预训练的RGB模型与高光谱图像在通道数上差异显著,难以捕捉光谱维度的相关性,限制了其在高光谱图像上的应用。受高光谱图像空间-光谱解耦启发,本文提出EigenSR框架:首先利用空间成分(即特征图像)对预训练模型进行微调,随后在迭代光谱正则化(ISR)下对未见高光谱图像进行推理,以保持光谱一致性。该方法优势在于:1)有效将预训练RGB模型的空间纹理处理能力引入高光谱图像,同时保留光谱保真度;2)在光谱去相关域中学习,提升对光谱无关数据的泛化能力;3)在特征图像域中推理天然利用高光谱图像的光谱低秩特性,降低计算复杂度。本工作通过特征图像实现预训练RGB模型与高光谱图像的桥梁连接,解决了高光谱训练数据有限的问题,因此命名为EigenSR。大量实验表明,EigenSR在空间与光谱指标上均超越现有最先进方法。
原文摘要 · Abstract (English)
Single hyperspectral image super-resolution (single-HSI-SR) aims to improve the resolution of a single input low-resolution HSI. Due to the bottleneck of data scarcity, the development of single-HSI-SR lags far behind that of RGB natural images. In recent years, research on RGB SR has shown that models pre-trained on large-scale benchmark datasets can greatly improve performance on unseen data, which may stand as a remedy for HSI. But how can we transfer the pre-trained RGB model to HSI, to overcome the data-scarcity bottleneck? Because of the significant difference in the channels between the pre-trained RGB model and the HSI, the model cannot focus on the correlation along the spectral dimension, thus limiting its ability to utilize on HSI. Inspired by the HSI spatial-spectral decoupling, we propose a new framework that first fine-tunes the pre-trained model with the spatial components (known as eigenimages), and then infers on unseen HSI using an iterative spectral regularization (ISR) to maintain the spectral correlation. The advantages of our method lie in: 1) we effectively inject the spatial texture processing capabilities of the pre-trained RGB model into HSI while keeping spectral fidelity, 2) learning in the spectral-decorrelated domain can improve the generalizability to spectral-agnostic data, and 3) our inference in the eigenimage domain naturally exploits the spectral low-rank property of HSI, thereby reducing the complexity. This work bridges the gap between pre-trained RGB models and HSI via eigenimages, addressing the issue of limited HSI training data, hence the name EigenSR. Extensive experiments show that EigenSR outperforms the state-of-the-art (SOTA) methods in both spatial and spectral metrics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。