用预训练模型提升小数据下的图像融合效果,性能领先。
PanAdapter: Two-Stage Fine-Tuning with Spatial-Spectral Priors Injecting for Pansharpening

- 分两阶段微调:先提取多尺度先验,再注入到ViT中
- 仅训练少量参数,即在多个基准上达到顶尖性能
- 适合资源有限但需高质量融合的遥感图像应用
pansharpening 是一项挑战性的图像融合任务,涉及利用低分辨率多光谱图像(LRMS)和高分辨率全色图像(PAN)恢复高质量图像。尽管已有许多基于深度学习的端到端专用模型提出,但其规模和性能受限于数据集大小。鉴于预训练模型在参数量和特征表示上的优势,当用于小数据下游任务时表现出色。为此,我们提出一种高效的微调方法 PanAdapter,利用预训练模型中的额外高级语义信息,缓解 pansharpening 任务中数据集小的问题。针对图像修复与 pansharpening 之间存在较大领域差异的问题,采用两阶段训练策略逐步适应下游任务。第一阶段,微调预训练的 CNN 模型,并通过提出的局部先验提取(LPE)模块在两个尺度上提取任务特定先验;第二阶段,将提取的双尺度先验分别输入级联适配器的两个分支,在每个适配器中设计两个参数高效模块,实现两分支交互并注入冻结的预训练 Vision Transformer (ViT) 块。实验证明,仅需训练少量参数的 LPE 模块和适配器,该方法即可受益于预训练图像修复模型,并在多个基准 pansharpening 数据集上达到当前最优性能。代码即将发布。
原文摘要 · Abstract (English)
Pansharpening is a challenging image fusion task that involves restoring images using two different modalities: low-resolution multispectral images (LRMS) and high-resolution panchromatic (PAN). Many end-to-end specialized models based on deep learning (DL) have been proposed, yet the scale and performance of these models are limited by the size of dataset. Given the superior parameter scales and feature representations of pre-trained models, they exhibit outstanding performance when transferred to downstream tasks with small datasets. Therefore, we propose an efficient fine-tuning method, namely PanAdapter, which utilizes additional advanced semantic information from pre-trained models to alleviate the issue of small-scale datasets in pansharpening tasks. Specifically, targeting the large domain discrepancy between image restoration and pansharpening tasks, the PanAdapter adopts a two-stage training strategy for progressively adapting to the downstream task. In the first stage, we fine-tune the pre-trained CNN model and extract task-specific priors at two scales by proposed Local Prior Extraction (LPE) module. In the second stage, we feed the extracted two-scale priors into two branches of cascaded adapters respectively. At each adapter, we design two parameter-efficient modules for allowing the two branches to interact and be injected into the frozen pre-trained VisionTransformer (ViT) blocks. We demonstrate that by only training the proposed LPE modules and adapters with a small number of parameters, our approach can benefit from pre-trained image restoration models and achieve state-of-the-art performance in several benchmark pansharpening datasets. The code will be available soon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。