From Pixels to Patches: Pooling Strategies for Earth Embeddings
Organizations: Wherobots · Microsoft AI for Good Research Lab
Abstract
Geospatial foundation models increasingly expose pixel-level embedding products that can be downloaded and reused without access to the underlying encoder. In this setting, downstream tasks with patch- or region-level labels require a post-hoc aggregation step that maps dense pixel embeddings to a single representation. The default choice, mean pooling, discards within-patch variability and can underperform under spatial distribution shift. To study this setting, we introduce EuroSAT-Embed: 81,000 embedding GeoTIFFs derived from three foundation models: AlphaEarth, OlmoEarth, and Tessera. Using these fixed embedding products, we benchmark 11 training-free pooling methods and 2 train-set-fitted baselines under both random and geographically disjoint test splits. Richer pooling schemes reduce the geographic generalization gap by over 50% relative to mean pooling and improve accuracy by up to 6% on spatial splits. We recommend a three-tier strategy: (1) mean as a baseline, (2) stats pooling (min/max/mean/std) as the default at 4x the embedding dimension, and (3) covariance pooling for peak accuracy. Across all three embedding products, simple distributional statistics improve spatial-split performance over mean pooling.
Figures & tables
| Spatial split | Random split | ||||||||
| Pooling | AEF | OlmoEarth ∗ | Tessera | Avg | AEF | OlmoEarth ∗ | Tessera | Avg | Gap |
| Std | 78.6 | 90.6 | 87.8 | 85.7 | 90.7 | 94.8 | 95.5 | 93.6 | 8.0 |
| Mean | 84.0 | 92.3 | 85.5 | 87.3 | 95.5 | 96.5 | 96.1 | 96.0 | 8.8 |
| GeM | 85.0 | 91.4 | 86.9 | 87.8 | 95.6 | 95.9 | 96.5 | 96.0 | 8.2 |
| Center-Weighted | 84.9 | 92.4 | 87.0 | 88.1 | 96.2 | 96.7 | 96.3 | 96.4 | 8.3 |
| Max | 86.0 | 91.1 | 89.7 | 88.9 | 92.9 | 93.0 | 93.8 | 93.2 | 4.3 |
| Spatial split | Random split | ||||||||
| Pooling | AEF | OlmoEarth ∗ | Tessera | Avg | AEF | OlmoEarth ∗ | Tessera | Avg | Gap |
| Std | 78.1 | 81.7 | 84.0 | 81.3 | 93.7 | 93.3 | 94.1 | 93.7 | 12.4 |
| Max | 80.6 | 85.5 | 80.8 | 82.3 | 90.4 | 91.4 | 92.5 | 91.4 | 9.1 |
| GeM | 85.8 | 86.6 | 85.4 | 85.9 | 94.6 | 93.3 | 95.1 | 94.3 | 8.4 |
| Median+IQR | 86.1 | 86.0 | 85.7 | 85.9 | 94.1 | 94.0 | 94.4 | 94.1 | 8.2 |
| Mean | 85.6 | 87.8 | 84.7 | 86.0 | 94.6 | 94.8 | 94.9 | 94.8 | 8.7 |
Spatial Generalization
Spatial Generalization studies whether models and spatial representations remain reliable across unseen locations, geographic scales, and changing environments.
See all 38 papers