4.3. Training Procedure
We employed a training period spanning 21 years of data with the dataset divided as follows: training data from 2003 to 2019 (17 years), validation data from 2020 to 2021 (2 years), and test data from 2022 to 2023 (2 years). This corresponds to approximately 81% of the data for training and 9.5% for validation and testing.
The model was trained using the AdamW optimizer with parameters , , and weight decay . The initial learning rate was set to , with five warm-up epochs and a cosine decay schedule to progressively adjust the learning rate.
Network hyperparameters were selected as follows: hidden layers of 128 neurons, resulting in 128-dimensional vector representations within the GNNs and associated MLPs; four processing layers in the model processor; and a hierarchical mesh with three uniform levels at resolutions of 81, 27, and 9, respectively. Training was conducted for 150 epochs without autoregressive steps due to computational constraints.
The training environment consisted of a desktop workstation with an Intel Core i9-9900K processor (3.60 GHz, 8 cores, 16 threads), 32 GB of RAM, and an NVIDIA GeForce RTX 3060 GPU with 12 GB of VRAM, which enabled efficient handling of large datasets and accelerated computation.
Before training, static features, such as bathymetry, coordinates, and normalization statistics, were computed, and differences between consecutive states were normalized to facilitate residual learning. The structure of the hierarchical mesh and bipartite connections were generated and stored for efficient access.
During training, the model learns to predict one-step-ahead oceanographic states. For evaluation, ensemble predictions are generated by introducing diversity in the initial states through noise perturbations (Gaussian, Perlin, and Perlin fractal). The ensemble predictions are then aggregated via daily averages, enabling both probabilistic and deterministic assessment of model performance.
4.4. Test Configurations
A preliminary parameter calibration phase was conducted before the ensemble experiments to define suitable noise configurations and to ensure a structured and reproducible exploration of the parameter space. Rather than performing a dense grid search, we adopt a controlled sensitivity analysis strategy aimed at sampling representative regions of each noise family.
In this phase, candidate parameter values for each noise type were evaluated based on their impact on forecast accuracy. Specifically, a one-factor-at-a-time approach was employed, in which each parameter was varied while the remaining ones were kept fixed, allowing us to isolate its individual effect. Each configuration was evaluated by computing the RMSE at each forecast lead time for 29 randomly selected initialization dates, drawn using a fixed random seed to ensure reproducibility, and then averaging across initialization dates to obtain a representative RMSE curve as a function of lead time.
This procedure defines a parameter screening stage, in which an initial set of candidate values is explored and a reduced number of representative configurations is selected for the subsequent ensemble experiments. The selection is guided by RMSE performance and aims to retain configurations that capture distinct behavioral regimes while avoiding redundancy.
The goal of this study was not to perform an exhaustive exploration of the full parameter space, but to compare qualitatively different types of perturbations under practical computational constraints. Therefore, the selected configurations should be understood as representative cases designed to illustrate differences in noise structure and behavior.
In addition, the parameter values were chosen such that the overall noise intensity remained roughly comparable across Gaussian and Perlin-based perturbations. This allows the comparison to focus on structural differences rather than differences in magnitude. For the Gaussian noise model, the standard deviation of the noise distribution was the only parameter considered for exploration, as it directly controls the amplitude of the perturbations.
The standard deviations used ( = 0.01, 0.03, 0.05, 0.1, 0.2, 0.5) correspond to approximately 0.5%, 1.4%, 2.3%, 4.5%, 9.1%, and 22.7% of the observed SST variability in the study region (standard deviation ≈ 2.2 K). These values were selected as part of an initial systematic exploration of the Gaussian noise amplitude, spanning a wide range of perturbation regimes from weak to strongly forced conditions.
As shown in
Figure 8, the best RMSE performance is consistently obtained for the lowest noise amplitudes. A gradual degradation of forecast skill is observed as the standard deviation increases, with a noticeable change in slope around
, beyond which performance deteriorates more rapidly, and a stronger increase in error is observed for
.
Based on this behavior, the selected configurations (
= 0.01, 0.05, 0.1) are chosen to represent low to intermediate noise amplitudes within the explored range. In particular,
and
correspond to weak or near-deterministic perturbations, while
lies at the upper end of the low-noise regime. The selected Gaussian configurations are summarized in
Table 3.
For the Perlin noise model, two parameters are considered: the spatial scale of the noise and the tileability constraint. The spatial scale is controlled by the number of grid subdivisions used to generate the noise field, which determines the size of the resulting spatial structures. The tileability parameter is set to enforce spatial continuity along the latitude dimension, ensuring coherent perturbations across domain boundaries. Spatial coherence is enforced by construction through the tileability constraint.
The resolutions of the Perlin noise were chosen based on the spatial autocorrelation structure and characteristic spatial scales of the SST field in the study region. Analysis of SST anomalies derived from the CMEMS product at 0.05° × 0.05° resolution indicates a dominant spatial scale of approximately 15 pixels (≈83 km), as shown in
Figure 9.
A configuration with 15×15 resolution on the 300 × 300 pixel domain generates perturbations of roughly 20 pixels (≈111 km), comparable in magnitude to the observed SST structures. This ensures that the noise captures physically meaningful spatial patterns. The classic Perlin noise configuration with 12 × 12 resolution produces slightly larger and coarser patterns, allowing evaluation of the effect of pattern size and density on forecast performance. A 3 × 3 base Perlin noise was also included to test very coarse, low-detail patterns.
Based on this physical constraint, the explored configurations are defined by varying the number of spatial subdivisions (3, 5, 6, 12, 15, and 25), which correspond to progressively finer spatial scales. This range spans scales both larger and smaller than the dominant SST variability (15 pixels), enabling a structured exploration of how the noise field interacts with physically relevant spatial scales.
The tileability parameter was also evaluated under a fixed-resolution setting (2 × 6 × 6) to assess its potential influence. However, no significant differences were observed in forecast performance between tileable and non-tileable configurations, suggesting a limited impact within the considered experimental setup. As a result, tileability is not further considered in the remainder of the study.
The RMSE analysis, as shown in
Figure 10, indicates that the coarsest configuration (3 subdivisions) yields slightly better performance at early forecast lead times. However, no monotonic relationship is observed between spatial scale and forecast error. In particular, the finest-scale configuration does not consistently exhibit the worst performance, and intermediate and fine-scale configurations produce comparable RMSE values. As lead time increases, differences across configurations further diminish, indicating a weak sensitivity of forecast skill to the spatial scale of the perturbations.
Based on this behavior, two representative configurations (3 × 3 and 12 × 12 resolutions) are selected for the subsequent evaluation stage. These configurations are chosen to represent contrasting levels of spatial detail: a coarse-scale regime characterized by large spatial structures, and an intermediate-scale regime with finer variability. In addition, they differ from the resolution configurations selected for the fractal Perlin noise model, enabling a clearer comparison between noise types. The selected Perlin configurations are summarized in
Table 4, where we use the following systematic naming scheme:
Classic Perlin noise uses P_res_XxYxZ, where XxYxZ indicates the spatial resolution along each dimension.
Fractal Perlin noise uses PF_res_XxY, where XxY indicates the 2D spatial resolution. Optional modifications are added as suffixes: (=value) indicates a modified scale factor applied to the noise.
For the fractal Perlin noise configurations, three parameters are explored: spatial resolution, tileability, and noise scale (). As in the previous configurations, the tileability constraint does not lead to noticeable differences in the spatial structure of the perturbations under the considered experimental setup and is therefore not further explored in detail.
The baseline configuration is derived from the setup proposed in Pangu-Weather [
15], which uses a fractal Perlin noise with 15 × 15 spatial resolution, three octaves, persistence of 0.5, lacunarity of 2, and a noise scale of 0.2. In this study, octaves, persistence, and lacunarity are kept fixed to the recommended values from Pangu-Weather to preserve the multi-scale structure of the noise and isolate the effect of the selected control parameters.
Spatial resolutions are chosen to ensure compatibility with the iterative octave construction, where each successive octave modifies the effective spatial scale according to the lacunarity factor. Given the 300 × 300 domain, the studied configurations (5 × 5, 15 × 15 and 25 × 25) are designed to be compatible with the domain discretization and the previous spatial analysis, while providing a comparable setting to the base Perlin configurations.
Unlike standard Perlin noise, which can be formulated in a fully spatio-temporal (3D) framework, the implementation used here is purely two-dimensional in space. Due to the octave-based construction, the noise field is generated independently for each temporal state, and therefore, no explicit coherence is enforced between noise realizations at consecutive time steps. However, as shown in
Figure 11, the RMSE analysis does not reveal a consistent sensitivity to resolution within this range, indicating that performance differences are not strongly driven by spatial resolution, as was also observed for the base Perlin configuration. Accordingly, the final analysis focuses on two representative resolutions: 5 × 5 and 15 × 15, corresponding respectively to higher and lower levels of spatial subdivision within the tested range.
The noise scale parameter is analyzed because it directly controls how strong the random perturbations are. Three values are tested: 0.05, 0.2, and 0.4. The smallest value (0.05) represents a weak-perturbation case that is close to deterministic behavior, while 0.2 corresponds to the baseline used in previous work (Pangu-Weather), and 0.4 represents a strongly perturbed case where random effects become dominant. As shown in
Figure 12, the RMSE clearly depends on this parameter. Lower noise values consistently lead to better forecast accuracy. Because of this, all three values are kept for the final evaluation to cover the full range of model behavior under different noise strengths. The selected fractal Perlin configurations are summarized in
Table 5.
All experiments were conducted using an ensemble-based approach. Unless otherwise specified, the default ensemble size was 5 members, each corresponding to a different random realization of the noise. Preliminary tests were first performed using 29 randomly selected initialization dates from the test dataset, matching those used in the noise analysis stage. This stage was used to assess the relative performance of different noise configurations under multiple initial conditions and to discard clearly underperforming candidates, including probabilistic metrics.
To characterize variability, results are reported using empirical 95% confidence intervals computed over evaluation metrics across initialization dates and ensemble members, providing an estimate of performance dispersion under stochastic initialization. No formal statistical significance testing is performed. Therefore, comparisons between configurations are interpreted in terms of relative performance trends and observed variability rather than strict statistical significance.
The configurations selected from this preliminary analysis were then evaluated in a full test setting spanning two years, starting on 1 January 2022. This final evaluation uses continuous forecasts initialized throughout the entire test period, ensuring that results reflect performance across a wide range of meteorological conditions rather than a limited set of initial states.