the Creative Commons Attribution 4.0 License.
the Creative Commons Attribution 4.0 License.
Copulabased synthetic data augmentation for machinelearning emulators
Thomas Nagler
Robin J. Hogan
Can we improve machinelearning (ML) emulators with synthetic data? If data are scarce or expensive to source and a physical model is available, statistically generated data may be useful for augmenting training sets cheaply. Here we explore the use of copulabased models for generating synthetically augmented datasets in weather and climate by testing the method on a toy physical model of downwelling longwave radiation and corresponding neural network emulator. Results show that for copulaaugmented datasets, predictions are improved by up to 62 % for the mean absolute error (from 1.17 to 0.44 W m^{−2}).
 Article
(2228 KB)  Fulltext XML
 BibTeX
 EndNote
The use of machine learning (ML) in weather and climate is becoming increasingly popular (Huntingford et al., 2019; Reichstein et al., 2019). ML approaches are being applied to an increasingly diverse range of problems for improving the modelling of radiation (e.g. Cheruy et al., 1996; Chevallier et al., 1998, 2000; Krasnopolsky et al., 2005; Meyer et al., 2021; Ukkonen et al., 2020; Veerman et al., 2021), ocean (e.g. Bolton and Zanna, 2019; Krasnopolsky et al., 2005), chemistry (e.g. Nowack et al., 2018), and convection (e.g. Krasnopolsky et al., 2013), as well as the representation of subgrid processes (e.g. Brenowitz and Bretherton, 2018; Gentine et al., 2018; O'Gorman and Dwyer, 2018; Rasp et al., 2018), and the postprocessing of model outputs (e.g. Krasnopolsky and Lin, 2012; Rasp and Lerch, 2018).
When it comes to training ML models for weather and climate applications two main strategies may be identified: one in which input and output pairs are directly provided (e.g. both come from observations) and a second in which inputs are provided but corresponding outputs are generated through a physical model (e.g. parameterization schemes or even a whole weather and climate model). Although the former may be considered the most common training strategy in use today, when the underlying physical processes are well understood (e.g. radiative transfer) and numerical codes are available, the latter may be of particular interest for developing onetoone emulators (i.e. statistical surrogates of their physical counterparts), which can be used to improve computational performance for a tradeoff in accuracy (e.g. Chevallier et al., 1998; Meyer et al., 2021; Ukkonen et al., 2020; Veerman et al., 2021). Here, for clarity, we will only be focusing on the latter case and refer to them as emulators.
In ML, the best way to make a model more generalizable is to train it on more data (Goodfellow et al., 2016). However, depending on the specific field and application, input data may be scarce, representative of only a subset of situations and domains, or, in the case of synthetically generated data, require large computational resources, bespoke infrastructures, and specific domain knowledge. For example, generating atmospheric profiles using a general circulation model (GCM) may require indepth knowledge of the GCM and large computational resources (e.g. data used in Meyer et al., 2021).
A possible solution to these issues may be found by augmenting the available input dataset with more samples. Although this may be a straightforward task for classification problems (e.g. by translating or adding noise to an image), this may not be the case for parameterizations of physical processes used in weather and climate models. In this context, it is common to work with highdimensional and strongly dependent data (e.g. between physical quantities such as air temperature, humidity, and pressure across grid points). Although this dependence may be well approximated by simple physical laws (e.g. the ideal gas law for conditions found in the Earth's atmosphere), the generation of representative data across multiple dimensions for most weather and climate applications is challenging (e.g. the nonlinear relationship between cloud properties, humidity, and temperature).
To serve a similar purpose as real data, synthetically generated data thus need to preserve the statistical properties of real data in terms of individual behaviour and (inter)dependences. Several methods may be suitable for generating synthetic data such as copulas (e.g. Patki et al., 2016), variational autoencoders (e.g. Wan et al., 2017), and, more recently, generative adversarial networks (GANs; e.g. Xu and Veeramachaneni, 2018). Although the use of GANs for data generation is becoming increasingly popular among the core ML community, these require multiple models to be trained, leading to difficulties and computational burden (Tagasovska et al., 2019). Variational approaches, on the other hand, make strong distributional assumptions that are potentially detrimental to generative models (Tagasovska et al., 2019). Compared to blackbox deeplearning models, the training of vine copulas is relatively easy and robust, while taking away a lot of guesswork in specifying hyperparameters and network architecture. Furthermore, copula models give a direct representation of statistical distributions, making them easier to interpret and tweak after training. As such, copulabased models have been shown to be effective for generating synthetic data comparable to real data in the context of privacy protection (Patki et al., 2016).
The goal of this paper is to improve ML emulators by augmenting the physical model's inputs using copulas. We give a brief overview of methods in Sect. 2.1 with specific implementation details in Sect. 2.2–2.5. Results are shown in Sect. 3, with a focus on evaluating synthetically generated data in Sect. 3.1 and ML predictions in Sect. 3.2. We conclude with a discussion and prospects for future research in Sect. 4.
2.1 Overview
The general method for training an ML emulator for a set of N samples involves the use of paired inputs $\mathit{x}=\mathit{\{}{x}_{\mathrm{1}},\mathrm{\dots},{x}_{N}\mathit{\}}$ and outputs $\mathit{y}=\mathit{\{}{x}_{\mathrm{1}},\mathrm{\dots},{x}_{N}\mathit{\}}$ to find the best function approximation for a specific architecture and configuration. For inference, the trained ML emulator is then used to predict new outputs y^{∗} from inputs x^{∗}. Outputs y are generated through a physical model from x and fed to the ML emulator for training (Fig. 1a). In this paper we introduce an additional step: augmentation through copulabased synthetic data generation (Fig. 1b). The method is demonstrated with a toy model of downwelling radiation as the physical model (Sect. 2.4) and a simple feedforward neural network (FNN) as the ML emulator (Sect. 2.5). To evaluate the impact of copulagenerated synthetic data on predictions we focus on predicting vertical profiles of longwave radiation from those of drybulb air temperature, atmospheric pressure, and cloud optical depth (other parameters affecting longwave radiative transfer, such as gas optical depth, are treated as constant in the simple model described in Sect. 2.4). This task is chosen at it allows us to (i) evaluate copulabased models for generating correlated multidimensional data (e.g. with dependence across several quantities and grid points), some of which (e.g. cloud optical depth) are highly nonGaussian; (ii) develop a simple and fast toy physical model that may be representative of other physical parameterizations such as radiation, (urban) land surface, cloud, or convection schemes; and (iii) develop a fast and simple ML emulator used to compute representative statistics. Here we define case (a) as the baseline and generate six different subcases for case (b) using (i) three levels of data augmentation factors (i.e. either 1×, 5×, or 10× the number of profiles in the real dataset) (ii) generated from three different copula types. In the following sections we give background information and specific implementation details about the general method used for setting up the source data (Sect. 2.2), data generation (Sect. 2.3), target generation (Sect. 2.4), and estimator training (Sect. 2.5) as shown in Fig. 1b.
2.2 Source data
Inputs are derived from the EUMETSAT Numerical Weather Prediction Satellite Application Facility (NWPSAF; Eresmaa and McNally, 2014) dataset. This contains a representative collection of 25 000 atmospheric profiles previously used to evaluate the performance of radiation models (e.g. Hocking et al., 2021; Hogan and Matricardi, 2020). Profiles were derived from 137verticallevel global operational shortrange ECMWF forecasts correlated in more than one dimension (between quantities and spatially across levels) and extending from the top of the atmosphere (TOA; 0.01 hPa; level 1) to the surface (bottom of the atmosphere; BOA; level 137). Inputs consist of profiles of drybulb air temperature (T in K; Fig. 2a), atmospheric pressure (p in Pa; Fig. 2b), and cloud layer optical depth (τ_{c}; Fig. 2c). τ_{c} is derived from other quantities to simplify the development of models as described in Sect. 2.4. Drybulb air temperature, atmospheric pressure, and cloud layer optical depth are then used as inputs to the physical model (Sect. 2.4) to compute outputs containing profiles of downwelling longwave radiation (L^{↓} in W m^{−2}; Fig. 2d). As both copula models and ML emulator work on twodimensional data, data are reshaped to input X and output Y matrices with each profile as row (sample) and flattened level and quantity as column (feature) and reconstructed to their original shape where required. Prior to being used, source data are shuffled at random and split into three batches of 10 000 profiles (40 %) for training (X_{train}, Y_{train}), 5000 (20 %) for validation (X_{val}, Y_{val}), and 10 000 (40 %) for testing (X_{test}, Y_{test}).
2.3 Data generation
Data generation is used to generate additional input samples (here atmospheric profiles) to be fed to the physical model (Sect. 2.4) and ML (Sect. 2.5) emulator. Optimally, synthetically generated data should resemble the observed data as closely as possible with respect to (i) the individual behaviour of variables (e.g. the drybulb air temperature at a specific level) and (ii) the dependence across variables and dimensions (e.g. the drybulb air temperature across two levels). Copulas are statistical models that allow these two aims to be disentangled (Trivedi and Zimmer, 2006; Joe, 2014) and to generate new samples that are statistically similar to the original data in terms of their individual behaviour and dependence.
2.3.1 Background on copula models
Suppose we want to generate synthetic data from a probabilistic model for n variables ${Z}_{\mathrm{1}},\mathrm{\dots},{Z}_{n}$. To achieve the first aim, we need to find appropriate marginal cumulative distributions $F,\mathrm{\dots},{F}_{n}$. A simple approach is to approximate them by the corresponding empirical distribution functions. To achieve the second aim, however, we need to build a model for the joint distribution function $F({z}_{\mathrm{1}},\mathrm{\dots},{z}_{n})$. The key result, Sklar's theorem (Sklar, 1959), states that any joint distribution function can be written as
The function C is called copula and encodes the dependence between variables.
Copulas are distribution functions themselves. More precisely, if all variables are continuous, C is the joint distribution of the variables ${U}_{\mathrm{1}}={F}_{\mathrm{1}}\left({Z}_{\mathrm{1}}\right),\mathrm{\dots},{U}_{n}={F}_{n}\left({Z}_{n}\right)$. This fact facilitates estimation and simulation from the model. To estimate the copula function C, we (i) estimate marginal distributions ${\widehat{F}}_{\mathrm{1}},\mathrm{\dots},{\widehat{F}}_{n}$, (ii) construct pseudoobservations ${\widehat{U}}_{\mathrm{1}}={\widehat{F}}_{\mathrm{1}}\left({Z}_{\mathrm{1}}\right),\mathrm{\dots},{\widehat{U}}_{n}={\widehat{F}}_{n}\left({Z}_{n}\right)$, and (iii) estimate C from the pseudoobservations. Then, given estimated models $\widehat{C}$ and ${\widehat{F}}_{\mathrm{1}},\mathrm{\dots},{\widehat{F}}_{n}$ for the copula and marginal distributions, we can generate synthetic data as follows.

Simulate random variables ${U}_{\mathrm{1}},\mathrm{\dots},{U}_{n}$ from the estimated copula $\widehat{C}$.

Define ${Z}_{\mathrm{1}}={\widehat{F}}_{\mathrm{1}}^{\mathrm{1}}\left({X}_{\mathrm{1}}\right),\mathrm{\dots},{Z}_{n}={\widehat{F}}_{n}^{\mathrm{1}}\left({X}_{n}\right)$.
2.3.2 Parametric copula families
In practice, it is common to only consider subfamilies of copulas that are conveniently parameterized. There is a variety of such parametric copula families. Such families can be derived from existing models for multivariate distributions by inverting the equation of Sklar's theorem:
For example, we can take F as the joint distribution function of a multivariate Gaussian and ${F}_{\mathrm{1}},\mathrm{\dots},{F}_{n}$ as the corresponding marginal distributions. Then Eq. (2) yields a model for the copula called the Gaussian copula, which is parameterized by a correlation matrix. The Gaussian copula model includes all possible dependence structure in a multivariate Gaussian distribution. The benefit comes from the fact that we can combine a given copula with any type of marginal distribution, not just the ones the copula was derived from. That way, we can build flexible models with arbitrary marginal distributions and Gaussianlike dependence. The same principle applies to other multivariate distributions and many copula models have been derived, most prominently the Student's t copula and Archimedean families. A comprehensive list can be found in Joe (2014).
2.3.3 Vine copula models
When there are more than two variables (n>2) the type of dependence structure these models can generate is rather limited. Gaussian and Student copulas only allow for symmetric dependencies between variables. Quite often, dependence is asymmetric, however. For example, dependence between Z_{1} and Z_{2} may be stronger when both variables take large values. Many Archimedean families allow for such asymmetries but require all pairs of variables to have the same type and strength of dependence.
Vine copula models (Aas et al., 2009; Czado, 2019) are a popular solution to this issue. The idea is to build a large dependence model from only twodimensional building blocks. We can explain this with a simple example with just three variables: Z_{1}, Z_{2}, and Z_{3}. We can model the dependence between Z_{1} and Z_{2} by a twodimensional copula C_{1,2} and the dependence between Z_{2} and Z_{3} by another, possibly different, copula C_{2,3}. These two copulas already contain some information about the dependence between Z_{1} and Z_{3}, the part of the dependence that is induced by Z_{2}. The missing piece is the dependence between Z_{1} and Z_{3} after the effect of Z_{2} has been removed. Mathematically, this is the conditional dependence between Z_{1} and Z_{3} given Z_{2} and can be modelled by yet another twodimensional copula C_{1,32}. The principle is easily extended to an arbitrary number of variables ${Z}_{\mathrm{1}},\mathrm{\dots},{Z}_{n}$. Algorithms for simulation and selection of the right conditioning order and parametric families for each (conditional) pair are given in Dißman et al. (2013).
Because all twodimensional copulas can be specified independently, such models are extremely flexible and allow for highly heterogenous dependence structures. Using parametric models for pairwise dependencies remains a limiting factor, however. If necessary, it is also possible to use nonparametric models for the twodimensional building blocks. Here, the joint distribution of pseudoobservations (${\widehat{U}}_{\mathrm{1}}$, ${\widehat{U}}_{\mathrm{2}}$) is estimated by a suitable kernel density estimator (see Nagler et al., 2017).
2.3.4 Implementation
Here we use Synthia (Meyer and Nagler, 2021) version 0.3.0 (Meyer and Nagler, 2020) with pyvinecopulib 0.5.5 (Nagler and Vatter, 2020) to fit three different copula types: Gaussian, vineparametric, and vinenonparametric. Vineparametric fits a parametric model for each pair in the model from the catalogue of Gaussian, Student, Clayton, Gumbel, Frank, Joe, BB1, BB6, BB7, and BB8 copula families and their rotations (see Joe, 2014, for details on these families) using the Akaike information criterion (AIC). Vinenonparametric uses transformation local quadratic likelihood fitting as explained in Nagler et al. (2017). Copulas are fitted to X_{train} to generate synthetic training sets ${{\mathbf{X}}^{\prime}}_{\text{train}}$ using three augmentation factors (i.e. each containing either 1×, 5×, or 10× the number of profiles in X_{train}). X_{train} plus ${{\mathbf{X}}^{\prime}}_{\text{train}}$ form augmented training sets containing 20 000 profiles (or double the amount of training data) for 1× augmentation factor and 60 000 and 110 000 profiles for 5× and 10× augmentation factors, respectively. As the generation of new profiles with copula models is random, the generation is also repeated 10 times for each case to allow meaningful statistics to be computed.
2.4 Target generation
Target generation is used to generate outputs from corresponding inputs using a physical model. Here, outputs are computed using a simple toy model based on Schwarzschild's equation (e.g. Petty, 2006) to estimate the downwelling longwave radiation under the assumption that atmospheric absorption does not vary with wavelength as
where z is the geometric height, B is the Planck function at level z (i.e. B=σ_{SB}T^{4}, where σ_{SB} is the Stefan–Boltzmann constant, giving the flux in W m^{−2} emitted from a horizontal blackbody surface), and a is the rate at which radiation is intercepted and/or emitted. A common approximation is to treat longwave radiation travelling at all angles as if it were all travelling with a zenith angle of 53^{∘} (Elsasser, 1942): in this case a=Dβ_{e}, where β_{e} is the extinction coefficient of the medium, and $D=\mathrm{1}/\mathrm{cos}\left(\mathrm{53}\right)=\mathrm{1.66}$ is the diffusivity factor, which accounts for the fact that the effective path length of radiation passing through a layer of thickness Δz is on average 1.66Δz due to the multiple different angles of propagation. In the context of ML, a(z) and B(z) are known and F(z) is to be predicted. Here we use the difference in two atmospheric pressures expressed in sigma coordinates (Δσ, where σ is the pressure p at a particular height divided by the surface pressure p_{0}) instead of z. The layer optical depth τ=β_{e}Δz is calculated from the totalcolumn gas optical depth τ_{g} and cloud layer optical depth τ_{c} as $\mathit{\tau}={\mathit{\tau}}_{\text{c}}+{\mathit{\tau}}_{g}\mathrm{\Delta}{\mathit{\sigma}}_{i}$, since Δσ is the fraction of mass of the full atmospheric column in layer i. Then, as the downwelling flux at the top of the atmosphere is 0, the equation is discretized as follows assuming B and a are constant within a layer:
where B_{i} is the Planck function of layer i, ${\mathit{\u03f5}}_{i}=\mathrm{1}{e}^{{a}_{i}\mathrm{\Delta}z}=\mathrm{1}{e}^{D\mathit{\tau}}$ is the emissivity of layer i, ${L}_{i+\mathrm{1}/\mathrm{2}}^{\downarrow}$ is the downwelling flux at the top of layer i, and ${L}_{i\mathrm{1}/\mathrm{2}}^{\downarrow}$ is the downwelling flux at the bottom of layer i. We compute L^{↓} from T, p, and τ_{c} using the real NWPSAF (X_{train}) or augmented (X_{train} plus ${{\mathbf{X}}^{\prime}}_{\text{train}}$) data. To reduce, and thus simplify, the number of quantities used in the physical model and ML emulator (Sect. 2.5), τ_{c} is precomputed and used instead of vertical profiles of liquid and ice mixing ratios (q_{l} and q_{l}) and effective radius (r_{l} and r_{l}) as $\frac{\mathrm{3}}{\mathrm{2}}\frac{\mathrm{\Delta}p}{g}\left(\frac{{q}_{l}}{{\mathit{\rho}}_{l}{r}_{l}}+\frac{{q}_{i}}{{\mathit{\rho}}_{i}{r}_{i}}\right)$, where ρ_{l} is the density of liquid water (1000 kg m^{−3}), ρ_{i} is the density of ice (917 kg m^{−3}), and g is the standard gravitational acceleration (9.81 m s^{−2}). For τ_{g} we use a constant value of 1.7 determined by minimizing the absolute error between profiles computed with this simple model and the comprehensive atmospheric radiation scheme ecRad (Hogan and Bozzo, 2018).
2.5 Estimator training
As the goal of this paper is to determine whether the use of synthetic data improves the prediction of ML emulators, here we implement a simple feedforward neural network (FNN). FNNs are one of the simplest and most common neural networks used in ML (Goodfellow et al., 2016) and have been previously used in similar weather and climate applications (e.g. Chevallier et al., 1998; Krasnopolsky et al., 2002). FNNs are composed of artificial neurons (conceptually derived from biological neurons) connected with each other; information moves forward from the input nodes through hidden nodes. The multilayer perceptron (MLP) is a type of FNN composed of at least three layers of nodes: an input layer, a hidden layer, and an output layer, with all but the input nodes using a nonlinear activation function.
Here we implement a simple MLP consisting of three hidden layers with 512 neurons each. This is implemented in TensorFlow (Abadi et al., 2015) and configured with the Exponential Linear Unit activation function, Adam optimizer, Huber loss, 1000epoch limit, and early stopping with patience of 25 epochs. The MLP is trained with profiles of drybulb air temperature (Fig. 2a), atmospheric pressure (Fig. 2b), and layer cloud optical depth (Fig. 2c) as inputs and profiles of downwelling longwave radiation (Fig. 2d) as outputs. Inputs are normalized and both inputs and outputs are flattened into twodimensional matrices as described in Sect. 2.2. The baseline case (Fig. 1a) uses 10 000 input profiles without data augmentation for training, and copulabased cases (Fig. 1b) use either 20 000, 60 000, or 110 000 profiles. The validation dataset Y_{val} of 5000 profiles is used as input for the early stopping mechanism, while the test dataset Y_{test} of 10 000 profiles is used to compute statistics (Sect. 3.2). Because of the stochastic nature of MLPs, training (and inference) is repeated 10 times for each case to allow meaningful statistics to be computed. Given that the generation of random profiles in the case of augmented datasets is also repeated 10 times (see Sect. 2.3.4), any case using data generation includes 100 iterations in total (i.e. for each data generation run, the estimator is trained 10 times).
3.1 Copula
The quality of synthetic data is assessed in terms of summary statistics (e.g. Seitola et al., 2014) between the training X_{train} and copulasimulated ${{\mathbf{X}}^{\prime}}_{\mathrm{train}}$ datasets. For each copula type we compute a vector of summary statistics s_{i}=f(p_{i}), where f is the statistic function and p_{i}=Dw_{i}, with D a matrix of flattened source or simulated data and w a vector of random numbers for the ith iteration. Summary statistics are computed for mean, variance, and quantiles, iterating 100 times to allow meaningful statistics to be computed. As we consider random linear combinations of variables in source and copulagenerated data, we expect these summaries to coincide only if both marginal distributions and dependence between variables are captured. Figure 3 shows scatterplots of summary statistics s_{i} for (a) mean, (b) variance, (c) standard deviation, and (d) 10 %, (e) 50 %, and (f) 90 % quantiles. Real NWPSAF data are shown on the x axis and copulagenerated data on the y axis, with each point corresponding to a random projection as described earlier (100 points in total). For a perfect copula model, we expect all points to fall on the main diagonal, where x=y. Figure 3 shows that for all copula models, synthetically generated data are close to the real data, with larger errors in variance and standard deviation.
Qualitatively, we can evaluate copulagenerated profiles in terms of their overall shape and smoothness across multiple levels, as well as range and density at each level. To this end we plot a sidebyside comparison of source (Fig. 4, left panel) and Gaussiancopulagenerated (Fig. 4, right panel) profiles showing the median profile and random selection of 90 profiles grouped in batches of 3 (i.e. each having 30 profiles) for the central 0 %–25 %, outer 25 %–50 %, and 50 %–100 % quantiles calculated with band depth statistics (LópezPintado and Romo, 2009). Simulated profiles of drybulb air temperature (Fig. 4b) appear less smooth than the real ones across levels (Fig. 4a); however, both density and range are simulated well at each level. Simulated profiles of atmospheric pressure (Fig. 4d) are simulated well: they are smooth across all levels with similar range and density (Fig. 4c). The highly nonGaussian and spiky profiles of cloud optical depth (Fig. 4e) make qualitative comparisons difficult; however, simulated profiles (Fig. 4f) have a similar range and density, with high density for low values, and most range between levels 80 and 120.
3.2 Machine learning
To evaluate whether ML emulators trained on augmented datasets have lower prediction errors compared to the baseline, here we use the test dataset X_{test} of 10 000 profiles defined in Sect. 2.2. Statistics are computed based on a vector of differences d between the physically predicted baseline Y_{test} and MLemulated ${{\mathbf{Y}}^{\prime}}_{\mathrm{test}}$ (i.e. $\mathit{d}={\mathbf{Y}}_{\mathrm{test}}{{\mathbf{Y}}^{\prime}}_{\mathrm{test}}$). From this, the mean bias $\left(\text{MB}=\frac{\mathrm{1}}{N}\sum _{i=\mathrm{1}}^{N}{d}_{i}\right)$ and mean absolute error $\left(\text{MAE}=\frac{\mathrm{1}}{N}\sum _{i=\mathrm{1}}^{N}\left{d}_{i}\right\right)$ for the set of N profiles are computed.
Box plots of MB and MAE are shown in Fig. 5. Summary MB and MAE for the ML emulator with the lowest MAE using an augmentation factor of 10× are reported in Table 2. A qualitative sidebyside comparison of baseline and MLpredicted profiles using Gaussiancopulagenerated profiles with an augmentation factor of 10× is shown in Fig. 6.
MBs (Fig. 5a) across all copula types and augmentation factors are generally improved, with median MBs and respective spreads decreasing with larger augmentation factors. Overall, the Gaussian copula model performs better than vineparametric and vinenonparametric models. MAEs (Fig. 5b) show a net improvement from the baseline across all copula models and augmentation factors. When using an augmentation factor of 1× (i.e. with double the amount of training data), the median MAE is reduced to approximately 1.1 W m^{−2} from a baseline of approximately 1.4 W m^{−2} and further reduced with increasing augmentation factors. In the best case, corresponding to an augmentation factor of 10× (i.e. with an additional 100 000 synthetic profiles), the copula and ML emulator combinations with the lowest MAE (Table 2) show that MBs are reduced from a baseline of 0.08 W m^{−2} to −0.02 and −0.05 W m^{−2} for Gaussian and vinenonparametric, respectively, but increased to 0.10 W m^{−2} for vineparametric. MAEs are reduced from a baseline of 1.17 W m^{−2} to 0.45, 0.56, and 0.44 W m^{−2} for Gaussian, vineparametric, and vinenonparametric copula types, respectively.
The ML training configuration with the lowest overall MB and MAE combination during inference corresponds to a Gaussian copula and augmentation factor of 10× (Table 2). Errors between the physically predicted Y_{test} and MLpredicted ${{\mathbf{Y}}^{\prime}}_{\mathrm{test}}$ are shown for the baseline (Fig. 6a) and Gaussian copula case (Fig. 6b). These are shown grouped by their central 0 %–25 %, outer 25 %–50 %, and 50 %–100 %. Qualitatively, most MLgenerated profiles show improvements. The most central 25 % profiles are within ±20 W m^{−2} for the Gaussian copula case and about ±40 W m^{−2} for the baseline case. Nearsurface errors (levels 130BOA) are reduced to approximately ±5 W m^{−2} from approximately ±10 W m^{−2}.
Results from a qualitative comparison of synthetically generated profiles (Fig. 4) show that synthetic profiles tend to be less smooth and noisier than the real NWPSAF. Nevertheless, a machinelearning evaluation shows that errors for emulators trained with augmented datasets are cut by up to 75 % for the mean bias (from 0.08 to −0.02 W m^{−2}; Table 2) and by up to 62 % for the mean absolute error (from 1.17 to 0.44 W m^{−2}; Table 2).
In this study, we show how copulabased models may be used to improve the prediction of ML emulators by generating augmented datasets containing statistically similar profiles in terms of their individual behaviour and dependence across variables (e.g. drybulb air temperature at a specific level and across several levels). Although the focus of this paper is to evaluate copulabased data generation models to improve predictions of ML emulators, we speculate that the same or similar methods of data generation have the potential to be used in several other MLrelated applications, such as to (i) test architectures (e.g. instead of crossvalidation, one may generate synthetic datasets of different size to test the effect of sample size on different ML architectures), (ii) generate data for unencountered conditions (e.g. for climate change scenarios by extending data ranges or relaxing marginal distributions), and (iii) compress data (e.g. by storing reduced parameterized versions of the data if the number of samples is much larger than the number of features).
Although so far, we have only highlighted the main benefits of copulabased models, several limiting factors should also be considered. A key factor for very highdimensional data is that both Gaussian and vine copula models scale quadratically in the number of features – in terms of both memory and computational complexity. This can be alleviated by imposing structural constraints on the model, for example using structured covariance matrix or truncating the vine after a fixed number of trees. However, this limits their flexibility and adds some arbitrariness to the modelling process. A second drawback compared to GANs is that the model architecture cannot be tailored to a specific problem, like images. For such cases, a preliminary data compression step as in Tagasovska et al. (2019) may be necessary.
As highlighted here, data augmentation for ML emulators may be of particular interest to scientists and practitioners looking to achieve a better generalization of their ML emulators (i.e. synthetic data may act as a regularizer to reduce overfitting; Shorten and Khoshgoftaar, 2019). Although a comprehensive analysis of prediction errors using different ML architectures is out of scope, our work is a first step towards further research in this area. Moreover, although we did not explore the generation of data for unencountered conditions (e.g. by extending the range of air temperature profiles while keeping a meaningful dependency across other quantities and levels), the use of copulabased synthetic data generation may prove useful to make emulators more resistant to outliers (e.g. in climate change scenario settings) and should be investigated in future research.
Software, data, and tools are archived with a Singularity (Kurtzer et al., 2017) image and deposited on Zenodo as described in the scientific reproducibility section of Meyer et al. (2020). Users may download the archive (Meyer, 2021) and optionally rerun experiments.
DM was responsible for conceptualization, data curation, investigation, software, resources, visualization, and validation. Formal analysis was conducted by DM and TN. DM, TN, and RH developed the methodology. DM and TN were responsible for original draft preparation, and DM, TN, and RH conducted review and editing of the paper.
The authors declare that they have no conflict of interest.
Publisher's note: Copernicus Publications remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
We thank three anonymous reviewers for their useful comments and feedback.
This paper was edited by Sylwester Arabas and reviewed by three anonymous referees.
Aas, K., Czado, C., Frigessi, A., and Bakken, H.: Paircopula constructions of multiple dependence, Insur. Math. Econ., 44, 182–198, https://doi.org/10.1016/j.insmatheco.2007.02.001, 2009.
Abadi, M., Barham, P., Chen, J., Chen, Z., Davis, A., Dean, J., Devin, M., Ghemawat, S., Irving, G., Isard, M., Kudlur, M., Levenberg, J., Monga, R., Moore, S., Murray, D. G., Steiner, B., Tucker, P., Vasudevan, V., Warden, P., Wicke, M., Yu, Y., and Zheng, X.: TensorFlow: A System for LargeScale Machine Learning, in: 12th USENIX Symposium on Operating Systems Design and Implementation (OSDI 16), Savannah, GA, 265–283, 2016.
Bolton, T. and Zanna, L.: Applications of Deep Learning to Ocean Data Inference and Subgrid Parameterization, J. Adv. Model. Earth Syst., 11, 376–399, https://doi.org/10.1029/2018MS001472, 2019.
Brenowitz, N. D. and Bretherton, C. S.: Prognostic Validation of a Neural Network Unified Physics Parameterization, Geophys. Res. Lett., 45, 6289–6298, https://doi.org/10.1029/2018GL078510, 2018.
Cheruy, F., Chevallier, F., Morcrette, J.J., Scott, N. A., and Chédin, A.: Une méthode utilisant les techniques neuronales pour le calcul rapide de la distribution verticale du bilan radiatif thermique terrestre, Comptes Rendus de l'Academie des Sciences Serie II, 322, 665–672, hal02954375, 1996.
Chevallier, F., Ruy, F. C., Scott, N. A., and Din, A. C.: A Neural Network Approach for a Fast and Accurate Computation of a Longwave Radiative Budget, J. Appl. Meteorol. Climatol., 37, 1385–1397, https://doi.org/10.1175/15200450(1998)037<1385:ANNAFA>2.0.CO;2, 1998.
Chevallier, F., Morcrette, J.J., Chéruy, F., and Scott, N. A.: Use of a neuralnetworkbased longwave radiativetransfer scheme in the ECMWF atmospheric model, Q. J. Roy. Meteor. Soc., 126, 761–776, https://doi.org/10.1002/qj.49712656318, 2000.
Czado, C.: Analyzing Dependent Data with Vine Copulas: A Practical Guide With R, Springer International Publishing, Cham, https://doi.org/10.1007/9783030137854, 2019.
Dißmann, J., Brechmann, E. C., Czado, C., and Kurowicka, D.: Selecting and estimating regular vine copulae and application to financial returns, Comput. Stat. Data Anal., 59, 52–69, https://doi.org/10.1016/j.csda.2012.08.010, 2013.
Elsasser, W. M.: Heat transfer by infrared radiation in the atmosphere, Blue Hill Meteorological Observatory, Harvard University, Milton, MA, USA, 1942.
Eresmaa, R. and McNally, A. P.: Diverse profile datasets from the ECMWF 137level shortrange forecasts, EUMETSAT Satellite Application Facility (NWP SAF), European Centre for Mediumrange Weather Forecasts Shinfield Park, Reading, RG2 9AX, UK, 2014.
Gentine, P., Pritchard, M., Rasp, S., Reinaudi, G., and Yacalis, G.: Could Machine Learning Break the Convection Parameterization Deadlock?, Geophys. Res. Lett., 45, 5742–5751, https://doi.org/10.1029/2018GL078202, 2018.
Goodfellow, I., Bengio, Y., and Courville, A.: Deep learning, MIT Press, Cambridge, 775 pp., 2016.
Hocking, J., Vidot, J., Brunel, P., Roquet, P., Silveira, B., Turner, E., and Lupu, C.: A new gas absorption optical depth parameterisation for RTTOV version 13, Geosci. Model Dev., 14, 2899–2915, https://doi.org/10.5194/gmd1428992021, 2021.
Hogan, R. J. and Bozzo, A.: A Flexible and Efficient Radiation Scheme for the ECMWF Model, J. Adv. Model. Earth Syst., 10, 1990–2008, https://doi.org/10.1029/2018MS001364, 2018.
Hogan, R. J. and Matricardi, M.: Evaluating and improving the treatment of gases in radiation schemes: the Correlated KDistribution Model Intercomparison Project (CKDMIP), Geosci. Model Dev., 13, 6501–6521, https://doi.org/10.5194/gmd1365012020, 2020.
Huntingford, C., Jeffers, E. S., Bonsall, M. B., Christensen, H. M., Lees, T., and Yang, H.: Machine learning and artificial intelligence to aid climate change research and preparedness, Environ. Res. Lett., 14, 124007, https://doi.org/10.1088/17489326/ab4e55, 2019.
Joe, H.: Dependence Modeling with Copulas, 1st edn., Chapman and Hall/CRC, https://doi.org/10.1201/b17116, 2014.
Krasnopolsky, V. M. and Lin, Y.: A Neural Network Nonlinear Multimodel Ensemble to Improve Precipitation Forecasts over Continental US, Adv. Meteorol., 2012, 649450 , https://doi.org/10.1155/2012/649450, 2012.
Krasnopolsky, V. M., Chalikov, D. V., and Tolman, H. L.: A neural network technique to improve computational efficiency of numerical oceanic models, Ocean Model., 21, 363–383, https://doi.org/10.1016/S14635003(02)000100, 2002.
Krasnopolsky, V. M., FoxRabinovitz, M. S., and Chalikov, D. V.: New Approach to Calculation of Atmospheric Model Physics: Accurate and Fast Neural Network Emulation of Longwave Radiation in a Climate Model, Mon. Wea. Rev., 133, 1370–1383, https://doi.org/10.1175/MWR2923.1, 2005.
Krasnopolsky, V. M., FoxRabinovitz, M. S., and Belochitski, A. A.: Using Ensemble of Neural Networks to Learn Stochastic Convection Parameterizations for Climate and Numerical Weather Prediction Models from Data Simulated by a Cloud Resolving Model, Advances in Artificial Neural Systems, 2013, 485913, https://doi.org/10.1155/2013/485913, 2013.
Kurtzer, G. M., Sochat, V., and Bauer, M. W.: Singularity: Scientific containers for mobility of compute, PLoS ONE, 12, e0177459, https://doi.org/10.1371/journal.pone.0177459, 2017.
LópezPintado, S. and Romo, J.: On the Concept of Depth for Functional Data, J. Am. Stat. Assoc., 104, 718–734, https://doi.org/10.1198/jasa.2009.0108, 2009.
Meyer, D.: Data archive for paper “Copulabased synthetic data augmentation for machine learningemulators” (Version 1.2.0) [Data set], https://doi.org/10.5281/zenodo.5150327, 2021.
Meyer, D. and Nagler, T.: Synthia: multidimensional synthetic data generation in Python (Version 0.3.0), Zenodo, https://doi.org/10.5281/zenodo.5150200, 2020.
Meyer, D. and Nagler, T.: Synthia: Multidimensional synthetic data generation in Python, Journal of Open Source Software, https://doi.org/10.21105/joss.02863, 2021.
Meyer, D., Schoetter, R., Riechert, M., Verrelle, A., Tewari, M., Dudhia, J., Masson, V., Reeuwijk, M., and Grimmond, S.: WRFTEB: Implementation and Evaluation of the Coupled Weather Research and Forecasting (WRF) and Town Energy Balance (TEB) Model, J. Adv. Model. Earth Syst., 12, e2019MS001961, https://doi.org/10.1029/2019MS001961, 2020.
Meyer, D., Hogan, R. J., Dueben, P. D., and Mason, S. L.: Machine Learning Emulation of 3D Cloud Radiative Effects, J. Adv. Model. Earth Syst., https://doi.org/10.1029/2021MS002550, 2021.
Nagler, T., Schellhase, C., and Czado, C.: Nonparametric estimation of simplified vine copula models: comparison of methods, Dependence Model., 5, 99–120, https://doi.org/10.1515/demo20170007, 2017.
Nowack, P., Braesicke, P., Haigh, J., Abraham, N. L., Pyle, J., and Voulgarakis, A.: Using machine learning to build temperaturebased ozone parameterizations for climate sensitivity simulations, Environ. Res. Lett., 13, 104016, https://doi.org/10.1088/17489326/aae2be, 2018.
O'Gorman, P. A. and Dwyer, J. G.: Using Machine Learning to Parameterize Moist Convection: Potential for Modeling of Climate, Climate Change, and Extreme Events, J. Adv. Model. Earth Syst., 10, 2548–2563, https://doi.org/10.1029/2018MS001351, 2018.
Patki, N., Wedge, R., and Veeramachaneni, K.: The Synthetic Data Vault, in: 2016 IEEE International Conference on Data Science and Advanced Analytics (DSAA), 2016 IEEE International Conference on Data Science and Advanced Analytics (DSAA), Montreal, QC, Canada, 399–410, https://doi.org/10.1109/DSAA.2016.49, 2016.
Petty, G. W.: A First Course in Atmospheric Radiation, End of Line Clearance Book, Madison, Wis, 459 pp., 2006.
Rasp, S. and Lerch, S.: Neural Networks for Postprocessing Ensemble Weather Forecasts, Mon. Weather Rev., 146, 3885–3900, https://doi.org/10.1175/MWRD180187.1, 2018.
Rasp, S., Pritchard, M. S., and Gentine, P.: Deep learning to represent subgrid processes in climate models, P. Natl. Acad. Sci. USA, 115, 9684–9689, https://doi.org/10.1073/pnas.1810286115, 2018.
Reichstein, M., CampsValls, G., Stevens, B., Jung, M., Denzler, J., Carvalhais, N., and Prabhat: Deep learning and process understanding for datadriven Earth system science, Nature, 566, 195–204, https://doi.org/10.1038/s4158601909121, 2019.
Seitola, T., Mikkola, V., Silen, J., and Järvinen, H.: Random projections in reducing the dimensionality of climate simulation data, Tellus A, 66, 25274, https://doi.org/10.3402/tellusa.v66.25274, 2014.
Shorten, C. and Khoshgoftaar, T. M.: A survey on Image Data Augmentation for Deep Learning, J. Big Data, 6, 60, https://doi.org/10.1186/s4053701901970, 2019.
Sklar, M.: Fonctions de repartition an dimensions et leurs marges, Open Journal of Statistics, 8, 229–231, 1959.
Tagasovska, N., Ackerer, D., and Vatter, T.: Copulas as highdimensional generative models: Vine copula autoencoders, in: Advances in neural information processing systems 32, edited by: Wallach, H., Larochelle, H., Beygelzimer, A., dAlchéBuc, F., Fox, E., and Garnett, R., Curran Associates, Inc., 6528–6540, 2019.
Trivedi, P. K. and Zimmer, D. M.: Copula Modeling: An Introduction for Practitioners, FNT in Econometrics, 1, 1–111, https://doi.org/10.1561/0800000005, 2006.
Ukkonen, P., Pincus, R., Hogan, R. J., Nielsen, K. P., and Kaas, E.: Accelerating radiation computations for dynamical models with targeted machine learning and code optimization, J. Adv. Model. Earth Syst., 12, e2020MS002226, https://doi.org/10.1029/2020ms002226, 2020.
Veerman, M. A., Pincus, R., Stoffer, R., van Leeuwen, C. M., Podareanu, D., and van Heerwaarden, C. C.: Predicting atmospheric optical properties for radiative transfer computations using neural networks, Phil. Trans. R. Soc. A., 379, 20200095, https://doi.org/10.1098/rsta.2020.0095, 2021.
Wan, Z., Zhang, Y., and He, H.: Variational autoencoder based synthetic data generation for imbalanced learning, in: 2017 IEEE Symposium Series on Computational Intelligence (SSCI), Honolulu, HI, 27 November–1 December 2017, https://doi.org/10.1109/SSCI.2017.8285168, 2017.
Xu, L. and Veeramachaneni, K.: Synthesizing Tabular Data using Generative Adversarial Networks, arXiv [preprint], arXiv:1811.11264, 27 November 2018.