arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2605.28331cs.CV
收藏

使用可训练张量分解将RGB模型迁移到高光谱图像

Transfer learning RGB models to hyperspectral images with trainable tensor decompositions

KU Leuven, Dept. of Computer Science(库勒芬大学计算机科学系);Leuven.AI - KU Leuven Institute for AI(莱文人工智能 - 库勒芬大学人工智能研究所)

Mariette Schönfeld, Laurens Devos, Wannes Meert, Hendrik Blockeel

AI总结:

提出一种通过可训练张量分解将预训练RGB模型的卷积滤波器分解为空间和光谱成分,并替换光谱成分以适应高光谱图像通道数的方法,实现高光谱图像迁移学习,实验表明该方法比其他方法更准确和鲁棒。

摘要:

迁移学习使得大型视觉网络能够通过将模型的通用滤波器专门化到新任务,从而应用于各种领域。然而,这些网络假设输入图像具有3个输入通道,使其与多光谱或高光谱图像不兼容。当前缓解这种不兼容性的方法要么牺牲图像信息,要么牺牲模型信息。本文提出了一种新颖的方法,通过使用部分可训练的张量分解来保留图像和模型中的空间信息。我们创建预训练卷积滤波器的这种分解,将滤波器分离为空间和光谱成分。然后,将光谱成分替换为具有更高通道维度的可训练成分。这创建了能够专门化到新数据集的高光谱滤波器,同时保留原始滤波器的空间模式。在各种高光谱数据集上的实验表明,我们的方法比其他高光谱迁移学习方法更准确和鲁棒。

英文摘要:

Transfer learning makes it possible to use large vision networks on a variety of domains, by specializing their models' general filters to new tasks. However, these networks assume the input images to have 3 input channels, making them incompatible with multi- or hyperspectral images. Current approaches that mitigate this incompatibility sacrifice information in either the image, or the model. This work proposes a novel approach that preserves the image and spatial information present in the model by using partially trainable tensor decompositions. We create such decompositions of pretrained convolutional filters, separating the filters into spatial and spectral components. The spectral components are then replaced with trainable components of higher channel dimensionality. This creates hyperspectral filters that can specialize to new datasets, while retaining the spatial patterns of the original filter. Experiments on a variety of hyperspectral datasets show that our approach is more accurate and robust than other hyperspectral transfer learning methods.

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

↑