arXiv:2510.22697cs.CV2025-10被引 1

WaveMAE用小波分解提升遥感图像自监督学习效果

WaveMAE: Wavelet decomposition Masked Auto-Encoder for Remote Sensing

  • 用多级小波变换分离频域成分,引导模型学习尺度感知的高频特征
  • 在PANGAEA多个任务上超越现有方法,分割与回归任务提升显著
  • 引入地理先验位置编码,适合需要空间结构理解的遥感应用

自监督学习(SSL)已成为遥感领域构建基础模型的关键策略,因标注数据稀缺,全监督方法难以应用。本文提出WaveMAE,一种针对多光谱卫星影像的掩码自编码框架。不同于传统像素重建,WaveMAE利用多级离散小波变换(DWT)解耦频率成分,引导编码器学习具有尺度感知能力的高频表征。我们进一步提出地理条件位置编码(GPE),通过球谐函数引入地理先验,促使嵌入同时遵循语义与地理空间结构。为确保评估公平性,所有方法均在相同数据集fMoW-S2上预训练,并在涵盖语义分割、回归、变化检测和多标签分类的PANGAEA基准上系统评估。大量实验表明,WaveMAE在多个下游任务上持续优于现有最优方法,尤其在分割与回归任务中表现突出。更值得注意的是,仅含26.4%参数的轻量级变体也达到了顶尖性能。结果证明WaveMAE是一种强大且具备地理感知能力的多光谱遥感图像基础模型。

原文摘要 · Abstract (English)

Self-supervised learning (SSL) has recently emerged as a key strategy for building foundation models in remote sensing, where the scarcity of annotated data limits the applicability of fully supervised approaches. In this work, we introduce WaveMAE, a masked autoencoding framework tailored for multispectral satellite imagery. Unlike conventional pixel-based reconstruction, WaveMAE leverages a multi-level Discrete Wavelet Transform (DWT) to disentangle frequency components and guide the encoder toward learning scale-aware high-frequency representations. We further propose a Geo-conditioned Positional Encoding (GPE), which incorporates geographical priors via Spherical Harmonics, encouraging embeddings that respect both semantic and geospatial structure. To ensure fairness in evaluation, all methods are pretrained on the same dataset (fMoW-S2) and systematically evaluated on the diverse downstream tasks of the PANGAEA benchmark, spanning semantic segmentation, regression, change detection, and multilabel classification. Extensive experiments demonstrate that WaveMAE achieves consistent improvements over prior state-of-the-art approaches, with substantial gains on segmentation and regression benchmarks. The effectiveness of WaveMAE pretraining is further demonstrated by showing that even a lightweight variant, containing only 26.4% of the parameters, achieves state-of-the-art performance. Our results establish WaveMAE as a strong and geographically informed foundation model for multispectral remote sensing imagery.

自监督学习遥感图像小波变换基础模型

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