Model architecture used for training in the case of ANN and GRU models. The yellow blocks represent either GRU layers or ANN layers, depending on the chosen architecture. The red blocks correspond to dropout layers modified to remain active during inference, enabling posterior distribution sampling for uncertainty quantification. The final blue block is a dense layer that maps the learned features to the output parameters, such as spectral model components.