SIGMAE: A Spectral-Index-Guided Foundation Model for Multispectral Remote Sensing
Organizations: School of Artificial Intelligence, Wuhan University, Wuhan, 430072, China · School of Artificial Intelligence and Automation, Wuhan University of Science and Technology, Wuhan, 430081, China · School of Electronic Information, Wuhan University of Science and Technology, Wuhan, 430081, China · Helmholtz-Zentrum Dresden-Rossendorf, Freiberg, 09599, Germany ·
Abstract
Pretraining and fine-tuning have emerged as a new paradigm in remote sensing image interpretation. Among them, Masked Autoencoder (MAE)-based pretraining stands out for its strong capability to learn general feature representations via reconstructing masked image regions. However, applying MAE to multispectral remote sensing images remains challenging due to complex backgrounds, indistinct targets, and the lack of semantic guidance during masking, which hinders the learning of underlying structures and meaningful spatial-spectral features. To address this, we propose a simple yet effective approach, Spectral Index-Guided MAE (SIGMAE), for multispectral image pretraining. The core idea is to incorporate domain-specific spectral indices as prior knowledge to guide dynamic token masking toward informative regions. SIGMAE introduces Semantic Saliency-Guided Dynamic Token Masking (SSDTM), a curriculum-style strategy that quantifies each patch's semantic richness and internal heterogeneity to adaptively select the most informative tokens during training. By prioritizing semantically salient regions and progressively increasing sample difficulty, SSDTM enhances spectrally rich and structurally aware representation learning, mitigates overfitting, and reduces redundant computation compared with random masking. Extensive experiments on five widely used datasets covering various downstream tasks, including scene classification, semantic segmentation, object extraction and change detection, demonstrate that SIGMAE outperforms other pretrained geospatial foundation models. Moreover, it exhibits strong spatial-spectral reconstruction capability, even with a 90% mask ratio, and improves complex target recognition under limited labeled data. The source codes and model weights will be released at https://github.com/zxk688/SIGMAE.
Figures & tables
| Dataset | Task | Image Size | Fine-tuning | Testing |
| FOD | Seg. | 6,087 | 3,704 | |
| Wildfire Detection | Seg. | 2,263 | 4,053 | |
| EuroSAT | Cls. | 21,600 | 5,400 | |
| SegMunich | Seg. | 39,402 | 9,846 | |
| OSCD | CD | 7,160 | 3600 |
| Model | Arch. | Pretrained EO Data | Learning Strategy | Params (M) | Patch Volume |
| CROMA (NIPS’24) | ViT | SSL4EO-S12 ( Wang et al., 2023 ) | Contrastive | 396.13 | 3M |
| SatlasNet (ICCV’23) | Swin-T | SatlasPretrain ( Bastani et al., 2023 ) | Supervised | 128.57 | 856K |
| S12-MAE (GRSM’23) | ViT | SSL4EO-S12 ( Wang et al., 2023 ) | MIM | 61.99 | 3M |
| ScaleMAE (ICCV’23) | ViT | fMoW-RGB ( Christie et al., 2018 ) | MIM | 396.21 | 363.6K |
| SpectralGPT (TPAMI’24) | ViT | fMoW-S2, BigEarthNet ( Sumbul et al., 2019 ) | MIM | 614.75 | 1.47M |
| SoftCon (TGRS’24) | ViT | SSL4EO-S12 ( Wang et al., 2023 ) | Contrastive | 242.19 | 3M |
| Hyperparameters | Pretraining | Fine-tuning |
| Optimizer | AdamW | AdamW |
| Base learning rate | 1e-4 | 1e-4 |
| Weight decay | 0.05 | 0.001 |
| Adam | (0.9, 0.95) | (0.9, 0.999) |
| Layer-wise lr decay | – | 0.75 |
| Batch size | 900 | 20 |
| Model | FOD | Wildfires Detection | OSCD | EuroSAT | |||||||||||
| mIoU | F1 | Recall | Precision | mIoU | F1 | Recall | Precision | mIoU | F1 | Recall | Precision | mIoU | OA | mF1 | |
| CROMA | 57.10 | 62.87 | 72.68 | 59.05 | 89.52 | 89.52 | 95.59 | 84.18 | 58.92 | 66.98 | 76.53 | 62.80 | 95.13 | 97.61 | 97.49 |
| S12-MAE | 56.59 | 62.09 | 60.58 | 64.15 | 89.31 | 89.28 | 84.59 | 94.53 | 57.54 | 65.09 | 76.72 | 60.90 | 95.51 | 97.85 | 97.73 |
| SatlasNet | 57.00 | 62.79 | 62.62 | 62.98 | 90.70 | 90.78 | 94.96 | 86.96 | 55.22 | 61.58 | 87.67 | 57.39 | 95.09 | 97.59 | 97.44 |
| ScaleMAE | 56.16 | 61.49 | 67.24 | 58.66 | 85.79 | 85.25 | 82.15 | 88.60 | 52.26 | 57.09 | 72.60 | 54.70 | 94.55 | 97.25 | 97.14 |
| SpectralGPT | 55.69 | 60.68 | 74.28 | 56.93 | 89.97 | 90.00 | 85.44 | 95.07 | 61.66 | 70.50 | 79.49 | 63.30 | 95.27 | 97.94 | 97.72 |
| Method | Background | Arable land | Perm. Crops | Pastures | Forests | Surface water | Shrub | Open spaces | Wetlands | Mine, dump | Artificial veg. | Urban fabric | Buildings | mF1 |
| CROMA | 90.64 | 82.77 | 21.42 | 64.07 | 89.12 | 70.51 | 20.62 | 33.32 | 44.72 | 40.57 | 27.79 | 78.33 | 67.25 | 56.24 |
| S12-MAE | 89.48 | 81.89 | 22.25 | 62.24 | 88.51 | 71.25 | 16.73 | 34.80 | 44.08 | 40.94 | 25.51 | 76.32 | 64.25 | 55.25 |
| SatlasNet | 90.81 | 81.69 | 17.98 | 61.32 | 88.43 | 81.30 | 16.45 | 43.81 | 50.44 | 47.08 | 22.44 | 75.93 | 63.69 | 57.03 |
| ScaleMAE | 85.12 | 81.14 | 20.04 | 60.82 | 87.62 | 70.49 | 15.77 | 29.45 | 44.29 | 39.47 | 25.52 | 77.32 | 65.57 | 54.05 |
| SpectralGPT | 87.80 | 83.34 | 31.24 | 65.53 | 89.11 | 85.84 | 21.31 | 46.05 | 46.41 | 52.59 | 23.98 | 78.97 | 65.18 | 59.80 |
| DOFA | 89.91 | 81.97 | 21.51 | 62.41 | 88.65 | 71.09 | 18.71 | 33.04 | 43.11 | 39.78 | 27.99 | 77.57 | 66.21 | 55.86 |
| Model Variants | FOD | Wildfire | EuroSAT | SegMunich | OSCD | |||||
| mIoU | F1 | mIoU | F1 | mIoU | mF1 | mIoU | mF1 | mIoU | F1 | |
| SIGMAE w/o SSDTM | 59.19 | 65.96 | 89.92 | 89.96 | 94.54 | 97.15 | 46.16 | 59.98 | 65.04 | 74.51 |
| MAE w/ Index Fusion | 59.70 | 67.03 | 89.06 | 89.04 | 95.16 | 97.50 | 46.31 | 59.96 | 64.96 | 74.44 |
| SIGMAE w/o | 61.23 | 68.78 | 90.67 | 90.73 | 94.58 | 97.18 | 46.44 | 60.16 | 66.07 | 75.70 |
| SIGMAE (Full) | 61.21 | 68.87 | 91.12 | 91.20 | 95.53 | 97.70 | 47.45 | 60.90 | 65.45 | 75.03 |
Remote Sensing Image Understanding
Remote sensing image understanding uses AI to interpret satellite and aerial imagery for objects, land cover, change, hazards, and spatial relationships.
See all 207 papers