Earth Observation Pretraining and Transfer
Pretrain Earth observation representations and evaluate their transfer.
Task & evaluation
- Objective
- Learn satellite-image features useful for both land cover and vegetation moisture.
- Input
- Optical and radar observations, including information about missing measurements.
- Output
- Shared features used by simple land-cover and moisture predictors.
- Data
- About 20,000 geographically separated pretraining samples, plus labeled evaluation datasets.
- Evaluation
- Land-cover accuracy and moisture prediction error.
- Research challenge
- Preserve information about both surface types and changing vegetation conditions.
Understanding the task
Follow satellite observations into a shared representation of land cover and vegetation moisture.
Can one view of the land answer two questions?
A satellite image can reveal what grows at a location and how its condition changes. This task tests both: classifying land cover and estimating how much water living vegetation contains. Recognizing a forest does not necessarily mean recognizing when its vegetation becomes dry.
Live fuel moisture is water mass relative to dry vegetation mass. It matters for vegetation and fire research, although this task does not predict fires. The goal is a shared representation that preserves useful information about both surface type and changing moisture.
What survives when a sensor goes missing?
Mask observations in a synthetic landscape, then follow their aggregate features into two fixed toy readouts. All colors and values below are illustrative, not satellite images or benchmark predictions.
1 · Observations → shared features
2 · Freeze → two kinds of readout
Twelve time slots and both modalities are available.
In the benchmark: The encoder combines observations from twelve time slots and different sensor bands. Both readouts use features near the labeled location. Even an example with no observed sensor groups stays in the evaluation; it receives zero features.
Different sensors, incomplete observations
The model learns from about 20,000 public samples with optical Sentinel-2 imagery, Sentinel-1 radar and WorldCover land maps. Optical bands measure reflected light; radar provides a different response to the surface. Observations arrive over time, and some sensor groups are missing.
Training excludes the land-cover evaluation region and neighborhoods around moisture-measurement sites. Downstream labels cannot teach the encoder. They are reserved for fitting simple readouts and selecting models. Missing observations remain part of evaluation rather than being discarded.
The land-cover development set has 1,115 training and 344 validation examples. Moisture has about 31,000 training and 8,300 validation examples. These are separate datasets, not two labels attached to every location.
Technical details
| Input / target | Representation | Meaning |
|---|---|---|
| Sentinel-2 L2A | [N,H,W,12,12], uint16 | 12 temporal slots × 12 optical bands |
| Sentinel-1 RTC (optional) | [N,H,W,12,2], float32 | 2 radar bands; mask unavailable groups |
| AWF | 16 × 16 input; labeled center | 1,115 train / 344 validation |
| LFMC | 32 × 32 input; center patch | 31,175 train / 8,262 validation |
| Validity mask | [N,12,num_band_sets] | All-missing rows return zero features and still count |
| Pretraining crop cache | 39,674 crops of 16 × 16 | Two deterministic crops per allowed sample |
What does a useful representation preserve?
Land-cover accuracy counts correct classes. Moisture error is the average absolute difference between predicted and measured values. Both are compared with raw spectral features and an untrained encoder of the same size, helping distinguish learned information from architecture alone.
The final reward averages land-cover validation accuracy and moisture test skill. Skill measures the reduction in error relative to always predicting the training-set mean, with negative skill set to zero. For example, reducing error from 25 to 15 gives skill 0.4; combined with accuracy 0.8, reward is 0.6.
There is no independent land-cover test split, and public moisture test data were exposed during earlier development. The reported score therefore cannot establish untouched-test generalization. Validation moisture error alone also cannot determine the final reward.
Technical details
| Quantity | Illustrative calculation | Interpretation |
|---|---|---|
| AWF accuracy | 8 correct / 10 = 0.80 | Use a fraction, not the number 80 |
| LFMC test MAE | (|60−50| + |80−100|) / 2 = 15 | Errors retain target units |
| Constant test MAE | Assume training mean = 75: (25 + 25) / 2 = 25 | Mean comes from training labels only |
| LFMC skill | max(0, 1 − 15 / 25) = 0.40 | 40% error reduction against constant |
| Final reward | 0.5 × (0.80 + 0.40) = 0.60 | Equal weight for accuracy and regression skill |
| Worse than constant | MAE 30 → max(0, 1 − 30 / 25) = 0 | Reward is 0.40 if accuracy remains 0.80 |
Learn the landscape, then freeze its features
The reference encoder first learns to recognize mixtures of land-cover classes within image patches. It then also learns an image-derived moisture indicator, NDMI, which compares near-infrared and short-wave infrared reflectance. Its average and variation over time encourage sensitivity to changing vegetation conditions.
NDMI is a proxy, not a field measurement of live fuel moisture. Whether learning this proxy improves prediction of measured vegetation water content is a question for the downstream experiment.
After training, the encoder stays fixed. A small readout, called a probe, learns to combine its features: logistic regression chooses a land-cover class, while ridge regression predicts a moisture value. Keeping these readouts simple tests what information the shared features already contain.
The animation illustrates this separation using synthetic signals. Its colors, features and predictions are teaching examples, not real satellite observations or measured model outputs.
Technical details
| Stage | Recorded configuration | Role |
|---|---|---|
| Patch inputs | 4 × 4 patches; 12 slots; observed-band masks | Spatial, temporal and band-group structure |
| Encoder | 24 layers; width 1,024; 16 heads; MLP ratio 4 | 307,770,368 parameters trained from scratch |
| Auxiliary heads | Linear WorldCover; linear 1,024 → 2 moisture | Training only; removed on export |
| Frozen pooling | Average observed temporal and band-group tokens | AWF upsampled center / LFMC center patch |
| Probe | 1,024 features → logistic or ridge regression | Only probe weights see downstream labels |
A larger gain for land cover than moisture
The trained representation reaches about 77% land-cover accuracy, versus 67% from raw spectra: an improvement of about ten percentage points. Moisture error falls from about 22.3 to 21.5, roughly a 4% reduction.
These validation results come from one adaptively developed training run. Resampling locations helps assess uncertainty across sites, but does not account for trying different recipes or variability across training runs. Without a comparison that omits the moisture objective, the moisture improvement cannot be attributed specifically to NDMI training.
| Representation | Land-cover accuracy | Moisture error |
|---|---|---|
| Untrained encoder | 63.4% | 22.65 |
| Raw spectral features | 66.9% | 22.32 |
| Trained encoder | 77.0% | 21.46 |
Compute and resources
Final adaptation and validation
Reported measurement
- Time
- ≈38.5 min of timed stages
- Hardware
- 1 × H200
- GPU allocation
- 0.644 GPU-h
The final moisture-learning phase, model export and full validation.
Elapsed time by stage
- Moisture adaptation
- 11.77 min
- CPU export
- 0.42 min
- Validation
- 26.27 min
Partial pipeline only: earlier land-cover training, data preparation and baselines are excluded. Allocation includes startup.
Resource details
- Minimum GPU memory
- Not establishedThe recorded H200 run does not establish performance or memory requirements on smaller cards.
What makes transfer work?
A direct next comparison is land-cover-only training versus the combined objectives. Another is optical imagery alone versus optical plus radar, especially when observations are missing. Repeating complete training runs would show whether the small moisture gain persists.
Temporal alignment is another challenge: pretraining and moisture observations use different sampling cadences. Better handling of when a measurement was made could matter as much as adding model capacity.
References and provenance
OlmoEarth: Stable Latent Image Modeling for Multimodal Earth Observation
Introduces the Earth-observation model family that motivates learning reusable representations from satellite measurements.
Task definition and experimental evidence
Documents the training objectives, frozen probes and data from ESA Sentinel and WorldCover, African Wildlife Foundation and Globe-LFMC. The paper’s decoder results use a different protocol.
Code links require access to the project’s GitHub repository.
Leaderboard
SimulatedDesign preview · leaderboard values and research histories are simulated.
| # | Method | Type | AWF validation accuracy (simulated) ↑ | Research time | Run |
|---|---|---|---|---|---|
| 1 | Agent search | 0.6252 | 76 h | ||
| 2 | Agent search | 0.6229 | 60 h | ||
| 3 | Illustrative trained-encoder baseline | Reference | 0.6200 | — | — |
| 4 | Illustrative random-encoder baseline | Reference | 0.5084 | — | — |
How to read these leaderboard metrics
These leaderboard values and research histories are simulated design examples. The leaderboard is ranked by AWF validation accuracy (simulated). The research curves show relative improvement in this metric over Illustrative trained-encoder baseline; only accepted new-best results update the retained trajectory. A positive gain means a better result, with the sign adjusted for metrics where lower is better. The task protocol is described in Task & evaluation above. These preview columns and trajectories are not measured benchmark results and must not be compared with the recorded experiments. This simulated table plots AWF validation accuracy, not final reward. Real final reward averages AWF validation accuracy and LFMC test skill. Measured Large-starter validation evidence is in the task guide; simulated runs are interface fixtures.
- AWF validation accuracy (simulated)Primary metric · Higher is better
- Synthetic illustration of AWF fraction correct. Real final scoring also requires LFMC test skill; see the worked example and recorded validation evidence.
AI4AI research progress
Research trajectory
Runtime–performance frontier
How improvement is measured
Simulated relative AWF accuracy improvement over the illustrative trained-encoder baseline. Only retained results update the curve. These are interface fixtures, not measured experiments.
Accepted new-best experiments raise the trajectory. The frontier includes evaluated candidates; trial, partial and stopped experiments are excluded.