arXiv:2511.16084cs.CVcs.AI2025-11被引 1

通过谱降维与课程学习,显著加速高光谱图像分类训练。

SpectralTrain: A Universal Framework for Hyperspectral Image Classification

  • 结合主成分分析降维与渐进式复杂度训练,提升学习效率。
  • 在3个数据集上实现2-7倍加速,精度损失小。
  • 通用框架适配各类模型,对气候遥感有应用潜力。

高光谱图像(HSI)分类通常涉及大规模数据和高计算成本,限制了深度学习模型在真实遥感任务中的部署。本文提出SpectralTrain,一种架构无关的通用训练框架,通过将课程学习(CL)与基于主成分分析(PCA)的谱降维相结合,逐步引入光谱复杂度的同时保留关键信息,从而以显著降低的计算成本高效学习光谱-空间模式。该框架独立于特定网络结构、优化器或损失函数,兼容经典与前沿模型。在Indian Pines、Salinas-A及新提出的CloudPatch-7三个基准数据集上的实验表明,其在不同空间尺度、光谱特性与应用领域均具强泛化能力。结果表明,训练时间缩短2-7倍,精度略有下降,具体取决于主干模型。该框架在云分类任务中的应用进一步揭示其在气候相关遥感中的潜力,强调训练策略优化是架构设计的有效补充。代码已开源:https://github.com/mh-zhou/SpectralTrain。

原文摘要 · Abstract (English)

Hyperspectral image (HSI) classification typically involves large-scale data and computationally intensive training, which limits the practical deployment of deep learning models in real-world remote sensing tasks. This study introduces SpectralTrain, a universal, architecture-agnostic training framework that enhances learning efficiency by integrating curriculum learning (CL) with principal component analysis (PCA)-based spectral downsampling. By gradually introducing spectral complexity while preserving essential information, SpectralTrain enables efficient learning of spectral -- spatial patterns at significantly reduced computational costs. The framework is independent of specific architectures, optimizers, or loss functions and is compatible with both classical and state-of-the-art (SOTA) models. Extensive experiments on three benchmark datasets -- Indian Pines, Salinas-A, and the newly introduced CloudPatch-7 -- demonstrate strong generalization across spatial scales, spectral characteristics, and application domains. The results indicate consistent reductions in training time by 2-7x speedups with small-to-moderate accuracy deltas depending on backbone. Its application to cloud classification further reveals potential in climate-related remote sensing, emphasizing training strategy optimization as an effective complement to architectural design in HSI models. Code is available at https://github.com/mh-zhou/SpectralTrain.

高光谱训练加速遥感框架

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