Search papers, labs, and topics across Lattice.
This paper addresses the challenge of applying RGB-pretrained convolutional networks to hyperspectral images by introducing trainable tensor decompositions of convolutional filters. The method decomposes pre-trained filters into spatial and spectral components, replacing the spectral components with trainable high-dimensional counterparts, thus enabling specialization to hyperspectral data while preserving spatial patterns. Experiments on multiple hyperspectral datasets demonstrate improved accuracy and robustness compared to existing hyperspectral transfer learning techniques.
Unlock the power of RGB-pretrained vision models for hyperspectral images without sacrificing crucial spatial information, thanks to a novel trainable tensor decomposition approach.
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.
