Skip to main content

Model design and training

Model design must balance predictive accuracy against inference throughput at continental or global scale, and accommodate the heterogeneity of EO data. Effective model training must examine the specificities of large EO datasets to guide hyperparameter tuning and data augmentation.

Trade-off

Developing ML models for large-scale EO applications requires balancing model complexity, predictive performance, and inference throughput. This trade-off becomes critical in operational settings, where maps must be produced on fixed schedules (quarterly to yearly monitoring) or on demand (real-time alerts, post-disaster response) over wide areas. It is largely shaped by two choices: the model family, and the use of spatial context. Model families range widely in cost. At the simple end, tree ensembles such as gradient boosting and random forests remain common in operational products [1], [2], [3], being cheap to train and deploy without GPUs. At the complex end, Vision Transformer architectures have attracted growing interest, including for large-area mapping [4], [5], [6], [7]. Simpler convolutional encoder-decoder architectures remain widely adopted in operational pipelines, where the latency and memory constraints often favor their lower computational footprint [8], [9], [10]. In particular, UNet architectures [11] with EfficientNet backbones [12] offer a well-documented trade-off between accuracy and throughput for such settings. Most of these architectures operate on image patches, exploiting the spatial context around each pixel. Many operational products instead predict each pixel independently, from its own time series [1], [4], [13], giving up spatial information in exchange for smaller inputs and simpler inference. Knowledge distillation [14] offers a further avenue to improve inference efficiency without notable performance degradation [15].

A recent benchmark of inference throughput across 45 vision backbones and geospatial foundation-model encoders illustrates the stakes: on a single H100 GPU, forward-pass throughput varies by roughly 205× between the fastest and slowest models, whereas moving the same model from a V100 to an H100 yields only a 3–5× speedup. For planetary-scale inference, model choice therefore dominates hardware choice, and combining half precision with torch.compile gives the largest gains [16].

NoteDecision · Model family and spatial context

Alternatives. Tree ensembles, convolutional encoder-decoders or vision transformers; each pixel predicted from its own time series, or within its spatial neighborhood.

Trade-off. Larger models can improve predictions but raise the cost of inference, which recurs with every update of a global map; per-pixel prediction keeps inputs small and inference simple, at the price of the spatial information that patches provide.

Constrains. The data layout (see Data formats), the inference tiling and its artifacts (see Map generation), and, in the other direction, the compute budget available for deployment.

Handling input heterogeneity

EO applications frequently combine inputs that differ in spatial resolution, number of channels, or temporal cadence. Rather than resampling all modalities to a common grid before ingestion (as described in From grid to patches), an alternative approach consists in delegating this alignment to the model itself. Typical designs employ modality-specific encoder branches that operate at each source’s native resolution and fuse representations at a shared spatial scale, either through learned upsampling, cross-attention, or feature concatenation after spatial alignment layers [17]. This strategy avoids the interpolation artifacts and information loss inherent in a priori resampling. Recent geospatial foundation models have pushed this flexibility further, using dynamic patch embeddings and scale-adaptive encoders to train a single model across sensors with arbitrary channel counts, resolutions, and tile extents, while also gracefully handling missing or variable modalities at inference time [18], [19], [20]. The added architectural complexity must, however, be weighed against the throughput and simplicity advantages of a single pre-aligned input tensor, particularly in operational pipelines where inference latency is a binding constraint.

Hyperparameter tuning

Hyperparameter optimization (HPO) is an important step in the ML pipeline, and we refer the reader to [21] for a general review of HPO good practices. We focus on two key aspects that are relevant for large EO datasets: which parameters to optimize, and how much data to use for it. Which parameters matter depends on the model family. Hyperparameter tuning for deep learning models is inherently challenging: the number of configurable hyperparameters is large, and their relative influence on model performance varies across tasks, architectures, and datasets, making it difficult to establish universally valid tuning priorities. Nevertheless, a broad consensus has emerged in the literature that the learning rate and its scheduler, batch size, and regularization mechanisms such as dropout and weight decay tend to be the most influential hyperparameters for deep neural networks [22]. Classical ML models have fewer hyperparameters: for gradient boosting trees, for example, the learning rate is the main one to tune [23]. Regarding how much data should be used for HPO, the common practice is to use the entire validation set, as in [4], [24]. However, given the computational cost of training on large datasets, a strategy encountered in the literature (e.g., [25]) is to perform hyperparameter optimization on a subset of the data, and subsequently train the final model on the full dataset using the selected configuration. This approach is a special case of multi-fidelity hyperparameter optimization, and many methods have been developed for it [26], [27], [28], ranging from simple random subsampling to gradient-based subset selection. When complex subset selection methods are not warranted, stratified random sampling should be preferred over naive uniform sampling, particularly in the presence of class imbalance or long-tailed distributions, to ensure that the subset adequately reflects the class distribution of the original dataset. When using the full dataset is prohibitively expensive, we suggest adopting a two-stage strategy: first using the subset for efficient hyperparameter search to narrow the configuration space, then validation of the top-performing configurations on the full dataset before final model training. We encourage researchers to document their hyperparameter optimization or to report when none was performed, as in [10].

Data augmentation

In the SatML field, less attention is brought to data augmentation than in the general ML field. This is partly due to the fact that “augmentation techniques designed for natural images should not be applied to satellite imagery without careful consideration” [29]. Their findings suggest that while standard natural image techniques (particularly color-based augmentations) do not translate effectively to the satellite domain, geometric operations remain a beneficial approach. Furthermore, Muhawenayo et al. [30] have identified that without specific data augmentation techniques to make them resilient, ML models often lack the necessary robustness to handle domain shifts (e.g., in terms of brightness) which can lead to decreased performance and visual artifacts. Various data augmentation techniques have been established in the literature to improve model generalization. Geometric transformations (such as rotations, flips, and random cropping) are being applied to EO data [8], [9], [31], [32], although particular care must be taken for SAR, since the look direction fixes which side of a slope shows layover or shadow. Recent research has introduced domain-specific augmentations tailored to EO data. To enhance spectral robustness, noise is frequently added to Sentinel-2 surface reflectance values [4], [31], while Muhawenayo et al. [30] advocate for varying Sentinel-2 processing strategies to mitigate sensitivity to shifting processing conventions. For models incorporating spatial metadata, Van Tricht et al. [1] apply random perturbations to latitude and longitude coordinates. Furthermore, temporal augmentations, such as dropping timesteps, are utilized to simulate cloud-induced data gaps [4] or missing modalities. Finally, several approaches specifically target sensor-induced artifacts to ensure resilience during inference; these include simulating narrow no-data seams in Sentinel-1 GRD mosaics [33], as well as modeling swath boundaries, misregistration, and detector failures [34].

A model trained with the practices described above will produce a prediction for every pixel on Earth. However, a prediction should come with a measure of its confidence. The model cannot perform equally well everywhere: it will be more reliable in regions well-represented in the training data than in under-sampled biomes, more confident under clear-sky conditions than under persistent cloud cover, and more accurate for common classes than for rare ones. Characterizing where and how much to trust the predictions is the subject of the next section, which surveys ML uncertainty quantification methods and argues that uncertainty is not a supplementary layer to be added as an afterthought, but a necessary component of any credible map product.

[1]
K. Van Tricht et al., “WorldCereal: a dynamic open-source system for global-scale, seasonal, and reproducible crop and irrigation mapping,” Earth System Science Data, vol. 15, no. 12, pp. 5491–5515, Dec. 2023, doi: 10.5194/essd-15-5491-2023.
[2]
European Space Agency, “ESA WorldCover 2021 v200 Map User Manual and Validation Report.” 2021. Available: https://esa-worldcover.org/
[3]
P. Potapov et al., “Mapping global forest canopy height through integration of GEDI and Landsat data,” Remote Sensing of Environment, vol. 253, p. 112165, 2021, doi: 10.1016/j.rse.2020.112165.
[4]
C. Mosig et al., “Sub-pixel mapping of disturbance and tree mortality dynamics from Sentinel-2 time series around the globe.” EarthArXiv preprint, Feb. 2026. doi: 10.31223/X5B18W.
[5]
M. Neumann et al., “Natural forests of the world – a 2020 baseline for deforestation and degradation monitoring,” Scientific Data, vol. 12, no. 1, Nov. 2025, doi: 10.1038/s41597-025-06097-z.
[6]
H. Herzog et al., “OlmoEarth: Stable Latent Image Modeling for Multimodal Earth Observation,” arXiv preprint arXiv:2511.13655, 2025, Available: https://arxiv.org/abs/2511.13655
[7]
J. Pauls et al., “ECHOSAT: Estimating Canopy Height Over Space and Time,” arXiv preprint arXiv:2602.21421, 2026, Available: https://arxiv.org/abs/2602.21421
[8]
S. Liu et al., “The overlooked contribution of trees outside forests to tree cover and woody biomass across Europe,” Science Advances, vol. 9, no. 37, Sep. 2023, doi: 10.1126/sciadv.adh4097.
[9]
T. Glazer et al., “TEMPO: Global Temporal Building Density and Height Estimation from Satellite Imagery,” arXiv preprint arXiv:2511.12104, 2025, Available: https://arxiv.org/abs/2511.12104
[10]
H. Kerner et al., “Fields of the World: A Machine Learning Benchmark Dataset for Global Agricultural Field Boundary Segmentation,” in Proceedings of the AAAI conference on artificial intelligence, 2025, pp. 28151–28159. doi: 10.1609/aaai.v39i27.35034.
[11]
O. Ronneberger, P. Fischer, and T. Brox, “U-Net: Convolutional Networks for Biomedical Image Segmentation,” in Medical image computing and computer-assisted intervention – MICCAI 2015, Springer International Publishing, 2015, pp. 234–241. doi: 10.1007/978-3-319-24574-4_28.
[12]
M. Tan and Q. V. Le, “EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks,” 2019, doi: 10.48550/ARXIV.1905.11946.
[13]
Z. Feng et al., “TESSERA: Temporal Embeddings of Surface Spectra for Earth Representation and Analysis,” arXiv preprint arXiv:2506.20380, 2025, Available: https://arxiv.org/abs/2506.20380
[14]
G. Hinton, O. Vinyals, and J. Dean, “Distilling the Knowledge in a Neural Network.” arXiv, 2015. doi: 10.48550/ARXIV.1503.02531.
[15]
Y. Himeur et al., “Applications of knowledge distillation in remote sensing: A survey,” Information Fusion, vol. 115, p. 102742, 2025, doi: https://doi.org/10.1016/j.inffus.2024.102742.
[16]
C. Robinson and I. Corley, “ThroughputBench: How fast can a deep learning model map the Earth?” [Online]. Available: https://geospatialml.com/posts/throughput-bench/
[17]
J. Li et al., “Deep learning in multimodal remote sensing data fusion: A comprehensive review,” International Journal of Applied Earth Observation and Geoinformation, vol. 112, p. 102926, Aug. 2022, doi: 10.1016/j.jag.2022.102926.
[18]
Z. Xiong et al., “Neural Plasticity-Inspired Foundation Model for Observing the Earth Crossing Modalities,” arXiv preprint arXiv:2403.15356, 2024.
[19]
G. Astruc, N. Gonthier, C. Mallet, and L. Landrieu, “AnySat: An Earth Observation Model for Any Resolutions, Scales, and Modalities,” arXiv preprint arXiv:2412.14123, 2024.
[20]
A. Fuller, K. Millard, and J. R. Green, “CROMA: Remote Sensing Representations with Contrastive Radar-Optical Masked Autoencoders,” in Thirty-seventh conference on neural information processing systems, 2023.
[21]
B. Bischl et al., “Hyperparameter Optimization: Foundations, Algorithms, Best Practices and Open Challenges.” arXiv, 2021. doi: 10.48550/ARXIV.2107.05847.
[22]
L. N. Smith, “A disciplined approach to neural network hyper-parameters: Part 1 – learning rate, batch size, momentum, and weight decay.” arXiv, 2018. doi: 10.48550/ARXIV.1803.09820.
[23]
P. Probst, A.-L. Boulesteix, and B. Bischl, “Tunability: Importance of hyperparameters of machine learning algorithms,” Journal of Machine Learning Research, vol. 20, no. 53, pp. 1–32, 2019, Available: http://jmlr.org/papers/v20/18-444.html
[24]
J. Pauls et al., “Estimating Canopy Height at Scale,” arXiv preprint arXiv:2406.01076, 2024, Available: https://arxiv.org/abs/2406.01076
[25]
J. Pauls et al., “Capturing Temporal Dynamics in Large-Scale Canopy Tree Height Estimation,” arXiv preprint arXiv:2501.19328, 2025, Available: https://arxiv.org/abs/2501.19328
[26]
T. Nickson, M. A. Osborne, S. Reece, and S. J. Roberts, “Automated Machine Learning on Big Data using Stochastic Algorithm Tuning.” arXiv, 2014. doi: 10.48550/ARXIV.1407.7969.
[27]
A. Klein, S. Falkner, S. Bartels, P. Hennig, and F. Hutter, “Fast Bayesian Optimization of Machine Learning Hyperparameters on Large Datasets.” arXiv, 2016. doi: 10.48550/ARXIV.1605.07079.
[28]
K. Killamsetty et al., “AUTOMATA: Gradient Based Data Subset Selection for Compute-Efficient Hyper-parameter Tuning.” arXiv, 2022. doi: 10.48550/ARXIV.2203.08212.
[29]
L. M. Hopkins, W.-K. Wong, H. Kerner, F. Li, and R. A. Hutchinson, “Data Augmentation Approaches for Satellite Imagery,” Proceedings of the AAAI Conference on Artificial Intelligence, vol. 39, no. 27, pp. 28097–28105, Apr. 2025, doi: 10.1609/aaai.v39i27.35028.
[30]
G. Muhawenayo et al., “PRUE: A Practical Recipe for Field Boundary Segmentation at Scale.” 2026. Available: https://arxiv.org/abs/2603.27101
[31]
C. F. Brown et al., “Dynamic World, Near real-time global 10 m land use land cover mapping,” Scientific Data, vol. 9, no. 1, Jun. 2022, doi: 10.1038/s41597-022-01307-4.
[32]
Microsoft, “Global Renewables Watch.” 2024. Available: https://github.com/microsoft/global-renewables-watch
[33]
C. F. Brown et al., “AlphaEarth Foundations: An embedding field model for accurate and efficient global mapping from sparse label data,” arXiv preprint arXiv:2507.22291, 2025, Available: https://arxiv.org/abs/2507.22291
[34]
V. J. Pasquarella, C. F. Brown, W. Czerwinski, and W. J. Rucklidge, “Comprehensive quality assessment of optical satellite imagery using weakly supervised video learning,” in 2023 IEEE/CVF conference on computer vision and pattern recognition workshops (CVPRW), IEEE, Jun. 2023, pp. 2125–2135. doi: 10.1109/CVPRW59228.2023.00206.