arXiv:2604.15828cs.CV2026-04中稿 · IEEE/CVF Conferenc…

轻量级模型高效完成高光谱分类,参数少于2%却性能领先

SSFT: A Lightweight Spectral-Spatial Fusion Transformer for Generic Hyperspectral Classification

论文配图:SSFT: A Lightweight Spectral-Spatial Fusion Transformer for Generic Hyperspectral Classification
图 1 · 摘自论文原文
  • 分谱段与空间路径学习,用交叉注意力融合信息
  • 在多场景数据集上超越现有方法,参数仅用前人1/50
  • 无需数据增强仍稳定,适合资源受限的通用分类任务

高光谱成像通过捕捉丰富的光谱特征实现材料的细粒度识别,但因维度高、谱段冗余、标注数据少及域偏移强,建模困难。尤其在地球观测之外,标注的高光谱数据常稀缺且不平衡,亟需紧凑模型实现跨采集场景的通用分类。本文提出轻量级谱-空融合变压器SSFT,将表征学习分解为谱段与空间两条路径,通过交叉注意力融合互补的波长依赖与结构信息。在挑战性多源异构的HSI-Benchmark上,SSFT达到当前最优性能,参数量不足此前领先方法的2%。进一步在更大规模的SpectralEarth基准上验证,其紧凑架构仍具竞争力。消融实验表明,谱段与空间路径均关键,其中空间建模贡献最大,且模型无需数据增强仍保持鲁棒。

原文摘要 · Abstract (English)

Hyperspectral imaging enables fine-grained recognition of materials by capturing rich spectral signatures, but learning robust classifiers is challenging due to high dimensionality, spectral redundancy, limited labeled data, and strong domain shifts. Beyond earth observation, labeled HSI data is often scarce and imbalanced, motivating compact models for generic hyperspectral classification across diverse acquisition regimes. We propose the lightweight Spectral-Spatial Fusion Transformer (SSFT), which factorizes representation learning into spectral and spatial pathways and integrates them via cross-attention to capture complementary wavelength-dependent and structural information. We evaluate our SSFT on the challenging HSI-Benchmark, a heterogeneous multi-dataset benchmark covering earth observation, fruit condition assessment, and fine-grained material recognition. SSFT achieves state-of-the-art overall performance, ranking first while using less than 2% of the parameters of the previous leading method. We further evaluate transfer to the substantially larger SpectralEarth benchmark under the official protocol, where SSFT remains competitive despite its compact size. Ablation studies show that both spectral and spatial pathways are crucial, with spatial modeling contributing most, and that SSFT remains robust without data augmentation.

高光谱分类轻量模型Transformer跨域适应

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。