Testing the reliability of interpretable neural networks in geoscience using the Madden–Julian oscillation
- 1Department of Atmospheric Science, Colorado State University, Fort Collins, Colorado, USA
- 2Lawrence Berkeley National Laboratory, Berkeley, California, USA
- 3Department of Agricultural and Environmental Sciences, University of California, Davis, Davis, California, USA
Correspondence: Benjamin A. Toms (firstname.lastname@example.org)
We test the reliability of two neural network interpretation techniques, backward optimization and layerwise relevance propagation, within geoscientific applications by applying them to a commonly studied geophysical phenomenon, the Madden–Julian oscillation. The Madden–Julian oscillation is a multi-scale pattern within the tropical atmosphere that has been extensively studied over the past decades, which makes it an ideal test case to ensure the interpretability methods can recover the current state of knowledge regarding its spatial structure. The neural networks can, indeed, reproduce the current state of knowledge and can also provide new insights into the seasonality of the Madden–Julian oscillation and its relationships with atmospheric state variables.
The neural network identifies the phase of the Madden–Julian oscillation twice as accurately as a linear regression approach, which means that nonlinearities used by the neural network are important to the structure of the Madden–Julian oscillation. Interpretations of the neural network show that it accurately captures the spatial structures of the Madden–Julian oscillation, suggest that the nonlinearities of the Madden–Julian oscillation are manifested through the uniqueness of each event, and offer physically meaningful insights into its relationship with atmospheric state variables. We also use the interpretations to identify the seasonality of the Madden–Julian oscillation and find that the conventionally defined extended seasons should be shifted later by 1 month. More generally, this study suggests that neural networks can be reliably interpreted for geoscientific applications and may thereby serve as a dependable method for testing geoscientific hypotheses.
Neural networks have the potential to improve our understanding of the earth system in ways that are unique from other statistical and machine learning methods. Recent research within the geosciences has shown that neural networks can be used to accelerate climate model parameterizations (Brenowitz and Bretherton, 2018; Rasp et al., 2018), discover patterns of earth-system variability (Toms et al., 2020a), and make accurate global weather predictions (Weyn et al., 2019), among numerous other applications in weather and climate (e.g., Barnes et al., 2019; Ebert-Uphoff and Hilburn, 2020). These advances have been rooted in the theory that neural networks are universal function mappers – that is, given a sufficient level of neural network complexity and quality of input data, a neural network can map any relationship between two datasets (Chen and Chen, 1995).
Neural networks may be particularly useful within the geosciences if the relationships contained within their learned parameters can be understood and interpreted. Numerous methods have been proposed for such interpretation within the computer science community and have even been shown to be applicable to improving the understanding of geoscientific phenomena such as El Niño–Southern Oscillation (ENSO), sources of seasonal predictability, and severe convective storms (Toms et al., 2020a; McGovern et al., 2019; Gagne et al., 2019; Ebert-Uphoff and Hilburn, 2020). The critical caveat of using interpretable neural networks within geoscience is that the interpretations must accurately portray the relationships captured by the neural network and not mislead the scientist toward incorrect conclusions. Therefore, any interpretability methods should first be tested on topics that are well understood so that trust can be lent to studies that use the methods to discover entirely new patterns.
The Madden–Julian oscillation (MJO; Madden and Julian, 1971; Wheeler and Hendon, 2004) has been a focus of hundreds of publications across numerous decades, and although it is not fully understood, a few of its characteristics are commonly accepted by the scientific community. For example, its core characteristic is an anomaly in deep convection and associated cloud cover within the tropics that forms within the western tropical Indian Ocean and propagates eastward toward the tropical eastern Pacific Ocean over the course of 30 to 60 d (Hendon and Liebmann, 1994; Wheeler and Hendon, 2004; Kiladis et al., 2005). While we will focus on the tropical characteristics of the MJO, it is also generally accepted that the atmospheric response to deep convective heating within the MJO can generate teleconnection patterns across the globe (e.g., Roundy et al., 2010; Tseng et al., 2019; Toms et al., 2020b). The formation and propagation of the MJO are not as well understood, although numerous theories have been put forth (Zhang et al., 2020), one of which suggests that the MJO propagates in response to gradients of tropical water vapor anomalies (Sobel and Maloney, 2013; Adames and Kim, 2016). Another theory suggests that the MJO could be a large-scale envelope of eastward- and westward-propagating gravity waves and that its eastward propagation occurs because the eastward waves travel faster than the westward waves (Yang and Ingersoll, 2011, 2013). Anomalies in atmospheric state variables that coincide with the MJO are also well documented (Kiladis et al., 2005; Adames and Wallace, 2014; Monteiro et al., 2014; Adames and Wallace, 2015), although their relationship with the seasonality of the MJO is less clear, particularly given remaining uncertainties in mechanisms driving the seasonality of the MJO itself (Zhang and Dong, 2004; Jiang et al., 2018). A more thorough discussion of the current understanding of the MJO is available in the recent review paper by Jiang et al. (2020).
We use the MJO as an opportunity to test whether interpretable neural networks can capture known patterns of variability within complex geoscientific data, and we then extend our analysis into inferring new information about the MJO itself. We also provide a new definition of MJO seasonality, for both the conventional outgoing longwave radiation definition and across atmospheric state variables. The aim of this paper is threefold: (1) to highlight the ability of neural networks to capture complex relationships within geoscientific data; (2) to test neural network interpretation methods to ensure they can reliably infer the relationships captured by neural networks; and (3) to use the interpretations to gain new insights into the MJO. This paper thereby offers a conceptual guideline for how a geoscientist might go about using a neural network to discover new patterns within geoscientific data. Those interested in the MJO itself will also find new insights into its spatial structures and seasonality.
We first discuss the data we use to define the MJO and then detail how we design a neural network to infer information about its spatial structure and seasonality.
We define the MJO according to the Outgoing Longwave Radiation MJO Index (OMI; Kiladis et al., 2014), which tracks the state of the MJO using anomalies in top-of-atmosphere outgoing longwave radiation (OLR; Liebmann and Smith, 1996). Increased cloud cover inhibits the upwards ventilation of longwave radiation to space, so outgoing longwave radiation is generally used as a proxy for cloud cover in studies of the MJO. Some of the details of OMI are listed below, and it can generally be defined as a linear representation of the MJO based on outgoing longwave radiation anomalies with periods of 20 to 96 d. An important advancement of OMI beyond other MJO indices is that the structure of the MJO is calculated for each day of the year across a 121 d rolling window and thereby accounts for seasonality. The index is constructed by calculating the two leading principal components in tropical (20∘ S to 20∘ N) outgoing longwave radiation anomalies, following the removal of the seasonal cycle and filtering the outgoing longwave radiation field to contain only eastward-propagating waves with a periodicity of 30 to 96 d. The MJO also exhibits higher-frequency modes of variability and occasional westward propagation (Roundy and Frank, 2004; Zhao et al., 2013), so outgoing longwave radiation anomalies that include both eastward- and westward-propagating waves with periods of 20 to 96 d are then projected onto the 30 to 96 d principal components. This projection results in OMI including all eastward- and westward-propagating components of the MJO with periods of 20 to 96 d, with the caveat that they must coincide with the dominant, eastward-propagating, 30 to 96 d mode of the MJO.
While the process of calculating OMI is complicated, the resultant phase-space and spatial perspectives of the MJO are relatively simple, as shown in Fig. 1. A two-dimensional phase space is commonly used to define the phase and amplitude of the MJO, with each axis representing the two OMI principal components. As the MJO progresses, it completes a circle about its two-dimensional phase space, which represents the eastward propagation of a spatially coherent dipole in outgoing longwave radiation anomalies (Fig. 1a). The phase space is conventionally separated into eight octants for convenience, so the MJO is commonly studied according to its evolution across eight discrete phases. The phase of the MJO is based on the azimuth of the linear combination of the two principal components, and its magnitude is determined based on the distance of this point from the origin. An MJO event is generally considered to be “active” once the principal component magnitude is greater than 1, which is delineated by the red dots in Fig. 1b. Because the principal components are standardized to have zero mean and unit variance, MJO events of increasing amplitude become increasingly rare, such that most events have low amplitude and are clustered about the origin.
We test whether a neural network can identify the phase of the MJO given inputs of cloud characteristics and atmospheric state variables. The inputs to the neural network are tropical (30∘ S to 30∘ N), 20 to 96 d filtered fields of outgoing longwave radiation and 850, 500, and 200 hPa zonal wind, meridional wind, temperature, water-vapor mixing ratio, and geopotential (Fig. 2a), and the outputs are the eight discrete phases of the MJO according to OMI. Only days during which the MJO was active are used (i.e., its principal component magnitude was greater than 1). We use atmospheric state variables from the NASA MERRA-2 reanalysis (Gelaro et al., 2017) and outgoing longwave radiation from the NOAA once-daily outgoing longwave radiation climate data record (Lee and NOAA CDR Program, 2014), both spanning from 1 January 1980 through 31 December 2016. We remove the seasonal cycle, defined as the annual-mean cycle from all 37 years of input data, before applying a 20 to 96 d Lanczos bandpass filter with 121 weights and interpolating each variable onto a homogeneous 2∘ grid. The training data span from 1 January 1980 through 31 December 2009, and the validation data span from 1 January 2010 through 31 December 2016. The training and validation data generally capture similar phase and amplitude distributions across each MJO phase (Fig. 2b).
2.2 Neural network design
We design a neural network to be as simple as possible while still ensuring it can capture any relationships between the input atmospheric state variables and the phase of the MJO. We use fully connected networks, which can be thought of as a chain of nonlinear regression functions that map the relationships between input and outputs datasets. The neural network has one input layer, two subsequent “hidden” layers with 64 and 128 nodes each, and one output layer with eight nodes, each of which represents a phase of the MJO (Fig. 3). The hidden nodes all use the ReLu activation function, which applies the operator to the output of each node. This activation function ensures the network can use nonlinearity if it is helpful for connecting the input and output. A softmax operator is applied to the output layer, which normalizes the output of the neural network such that the sum across all output nodes is equal to 1. The outputs can therefore be thought of as a likelihood, with higher values for each node corresponding to a higher likelihood that the input sample belongs in that particular phase of the MJO. During labeling, each MJO event is labeled using an eight-unit vector, and each unit represents one phase of the MJO. An input associated with phase 3 of the MJO would therefore have an output label of , which in the perspective of the neural network implies a 100 % likelihood that the sample is associated with phase 3 of the MJO. We leave the discussion of neural networks brief to focus on this work's scientific outcomes, but there is an abundance of additional resources for the reader through publications, textbooks, and other free online resources (e.g., Goodfellow et al., 2016; Gagne et al., 2019; Toms et al., 2020a).
We train separate neural networks on data from 121 d bins centered on each calendar week of the year in order to study the seasonality of the MJO. Each neural network is therefore tasked with identifying the phase of the MJO according to the outgoing longwave radiation and state variable patterns during the period of the year to which it is assigned. Comparisons between interpretations of each neural network offer insights into the seasonality of the MJO, as discussed in subsequent sections.
Neural network interpretability generally becomes more challenging with increasing network complexity (Montavon et al., 2018). The neural network design we use is simple enough to enable robust interpretations but complex enough to capture useful relationships between the input state variables and MJO phase. We find that decreasing the number of internal nodes reduces the accuracy, presumably because the network is then not complex enough to model the relationships between the atmospheric state variables and MJO. On the other hand, increasing the number of nodes also reduces the accuracy of the neural network on the validation dataset, because it is able to overfit on meaningless noise within the inputs using the additional weights and biases. We address any overfitting by applying L2 regularization to the weights connecting the input layer to the first layer of hidden nodes, which forces the network to focus its attention on broader spatial patterns within the inputs. We thoroughly tested the accuracy of convolutional neural networks (CNNs) for our particular problem and found that the fully connected networks were both more accurate and interpretable than their CNN counterparts when L2 regularization is applied to the first layer of hidden nodes.
2.3 Neural network interpretability
The novelty of this paper is the demonstrated ability to interpret what the neural networks have learned and to then gather scientific value from the interpretations. We use two interpretation methods which we briefly discuss here and are explained in more extensive detail in the context of geoscience within Toms et al. (2020a). The two methods we use are called backward optimization and layerwise relevance propagation, both of which map the decision-making process of the neural network onto the original input dimensions.
Backward optimization uses the same method that is used to train a neural network (i.e., backpropagation) to instead interpret what a trained network has learned (Simonyan et al., 2013; Yosinski et al., 2015; Olah et al., 2017). Rather than updating the weights and biases of the network, the input itself is updated to minimize the difference between the network's associated output and a user-defined output. This process generates a single optimized pattern associated with a particular output and thereby offers a composite interpretation of patterns contained within a neural network. In our case, we input blank (i.e., all-zero) maps and optimize them to be most closely associated with a particular phase of the MJO. In doing so, we can identify the optimal patterns in each state variable for each phase of the MJO.
Layerwise relevance propagation (LRP) interprets the neural network's decision-making process for each individual input sample (Bach et al., 2015; Montavon et al., 2017, 2018). Given a trained neural network, an input sample is passed forward, the associated output is collected, and the unique pathways through which information flows from the input to the output for that specific sample are analyzed. The pathways are traced by propagating information backwards from the output layer to the input layer using rules specific to LRP. By tracing these pathways, the “relevance” of each input variable to the network's associated output can be quantified for each individual input example. The resultant relevance is unique to each input sample, because the pathways through which information flows through a neural network are similarly unique for each sample. A particularly important aspect of LRP is that the formulation of neural network that we use (i.e., fully connected networks with ReLu activation functions) conserves the relevance from the output layer to the input layer, meaning that all information important to the network's decision is included within the final LRP interpretation. LRP traces the information that positively contributes to the output of the neural network, so it is well suited to categorical output. So, in our case, LRP shows which regions of atmospheric state variables are most relevant to increases in the neural network's confidence that the sample belongs to a particular phase of the MJO.
3.1 Neural network accuracy
We first ensure the neural networks are accurate enough to offer scientifically valuable interpretations. As a reminder, we train separate neural networks on data from 121 d windows centered on each calendar week of the year. The accuracy of the neural network for the window centered on 10 January is presented from both a deterministic and probabilistic perspective in Fig. 4. The deterministic accuracy is assessed by counting the number of input samples the neural network assigns to the correct phase of the MJO. The most common error of the neural network is to assign an input sample to a phase that is one phase prior to or after the correct phase, which is likely caused by the MJO being a continuous phenomenon that we have discretized for the sake of interpretation. So, another useful accuracy metric is how often the neural network correctly assigns the input samples into either the correct phase or one phase before or after the correct phase. For the neural network centered on 10 January, the deterministic accuracy without a one-phase buffer is 74 % and the accuracy with a one-phase buffer is 92 %. The deterministic accuracy for each phase from one through eight, respectively, is as follows: 78 %, 72 %, 64 %, 88 %, 51 %, 77 %, 65 %, and 82 %. From the composite probabilistic perspective (Fig. 4b), the neural network assigns the highest likelihoods to the correct phase, although the phases immediately before and after the correct phase also have appreciably high likelihoods.
An important question regarding the usage of neural networks is whether they out-perform conventional methods, such as regression. If regression performs similarly to a neural network, then the increased complexity and nonlinearity of a neural network is not required. We therefore similarly use a form of linear regression to identify the phase of the MJO using the input state variables and outgoing longwave radiation across 121 d windows centered on each calendar week. The multi-output linear regression models have no hidden nodes and no nonlinearities but are otherwise identical to the neural networks (i.e., as in Fig. 3, but without the blue hidden layers). These models therefore receive atmospheric state variables as inputs, which are then connected to eight output nodes, after which a softmax operator is applied to transform the output into a normalized likelihood. This method therefore does not allow nonlinearities but does still permit the model to identify patterns unique to each phase of the MJO. We regularize these multi-output linear regression models using L2 regularization to ensure they are not overfit to the training data, similar to the neural networks. The accuracies of the neural network and multi-output linear regression approaches are compared in Fig. 5. The neural networks are nearly twice as accurate as linear regression for all weeks of the year, which means the nonlinearities and increased number of pathways for information to flow through the neural network are essential to modeling the spatial structures of the MJO. We therefore conclude that interpretations of the neural networks can offer insights into relationships between the MJO and atmospheric state variables that conventional linear methods can not. It is worth noting that the neural network is less capable of identifying the MJO during boreal summer than boreal winter. This may be caused by numerous factors, including a reduced fraction of OLR variability attributable to the MJO during boreal summer compared to boreal winter (e.g., Kiladis et al., 2014), or the possibility that the nonlinearity of our neural network architecture is more useful for identifying the boreal winter MJO than boreal summer MJO. Because of this uncertainty, we opt to not assess the seasonality of the accuracy, although it is an interesting point for future study.
3.2 Interpreting the neural network
3.2.1 Identifying the spatial structures of the MJO
We use backward optimization and layerwise relevance propagation (LRP) to infer the spatial structure of the MJO and its seasonality according to the neural networks. Examples of LRP applied to inputs for the neural network trained on the 121 d window centered on 10 January are shown in Fig. 6. We use four examples of MJO phase 7 for which the neural network correctly identifies the phase of the MJO, and for simplicity we only show the LRP maps for outgoing longwave radiation, although similar maps are generated for each input variable. The LRP maps show that the neural network focuses its attention on outgoing longwave radiation anomalies across the Maritime Continent, particularly within its eastern extent, which is consistent with previous research on the regions of convection associated with phase 7 of the MJO (Wheeler and Hendon, 2004; Kiladis et al., 2014). The LRP heatmaps also highlight the spatial uniqueness of each phase-7 MJO event, which can not be inferred by a linear regression model. It is likely that the increased accuracy of the neural networks compared to the multi-output linear regression models is caused by this ability of the neural network to capture the spatial uniqueness of each event.
We next test the neural networks more rigorously and challenge them to identify the most common spatial structures of the MJO across its eight phases. To do so, we use backward optimization and optimize inputs such that the spatial patterns within the inputs make the neural networks most confident that the inputs are associated with a particular phase of the MJO. Numerically, this means that the outputs associated with the optimized inputs have a likelihood of approximately 1 in the phase for which they are optimized, and likelihoods of 0 for all other phases. We again only show the optimized outgoing longwave radiation fields for simplicity, although the optimization also identifies the characteristic patterns in the 15 other state variables.
The spatial pattern of the MJO during boreal winter (10 January) and boreal summer (1 August) according to both OMI and the neural networks is shown in Fig. 7. The neural networks capture similar features to OMI during both seasons, in particular the prominent eastward propagation during boreal winter and the transition to northeastward propagation during boreal summer. The neural network focuses on a core of outgoing longwave radiation anomalies across the Indo-Pacific region and within the eastern Pacific, while OMI includes a greater magnitude of anomalies within the central Pacific. Given the similarities between OMI and the composite neural network interpretations, we conclude that nonlinearities of the MJO are primarily manifested through the uniqueness of each event as highlighted in Fig. 6. The composites of LRP relevance similarly capture the dominant structures of the MJO during both seasons and agree rather well with the optimized inputs (Fig. 7).
3.2.2 Testing the seasonality of the MJO
Because the neural network so accurately captures the seasonal evolution of the MJO within the outgoing longwave radiation composites, we now extend the interpretations to study the seasonality of the MJO. We first test how the spatial structure of the MJO changes across seasons using LRP. To do so, we calculate the composite relevance for each variable for each calendar week of the year and present the annual evolution of the relevance in Fig. 8. The relevances of each variable exhibit unique seasonal cycles, aside from outgoing longwave radiation, which is similarly important for the identification of the MJO throughout all periods of the year. For example, the seasonal cycle of lower-tropospheric zonal wind (U850) reaches a maximum in relevance during boreal summer, whereas upper-tropospheric zonal wind (U200) is most important during the spring and fall. Some variables exhibit a uni-modal seasonal cycle (e.g., U850, T200), whereas other variables exhibit a bimodal seasonal cycle (e.g., U200, V200, Z850). In general, upper-tropospheric anomalies are most important for identifying the MJO during boreal winter, while lower-tropospheric anomalies are most important during boreal summer.
The fact that upper-tropospheric anomalies are most important for identifying the MJO during boreal winter may explain the seasonality in coupling between the MJO and the stratosphere (Son et al., 2017; Densmore et al., 2019; Toms et al., 2020b). Previous research has hypothesized that the MJO can be modulated by sources of stratospheric variability such as the quasi-biennial oscillation through a downward influence of upper-tropospheric temperature anomalies (Abhik and Hendon, 2019; Martin et al., 2020). So, because upper-tropospheric thermodynamic anomalies are particularly important during boreal winter (Fig. 8), any influences on the thermodynamic structure of the upper troposphere by the stratosphere may have an increased impact on the MJO. This discussion highlights the capability of neural network interpretations to guide and test proposed hypotheses, although a direct test of this hypothesis is beyond the scope of this paper.
We now examine the optimal spatial patterns of the MJO throughout the year to provide some spatial context to the seasonality of the relevances shown in Fig. 8. Figure 9 shows the optimal spatial patterns associated with phase 6 of the MJO at four different times of the year, the times of which are denoted by the dashed white lines in Fig. 8. In general, the spring structure of the MJO is more similar to the winter structure than the summer structure. The 15 April optimal patterns are nearly identical to the 15 January optimal patterns, aside from lower-tropospheric moisture anomalies which are more similar between 15 April and 1 August. The upper-tropospheric anomalies during boreal winter are more representative of a Matsuno–Gill type response to convective heating (Matsuno, 1966), whereas during boreal summer the signature is more diffuse and elongated across the Equator. Figure 9 is generally supportive of the idea that lower-tropospheric anomalies are most important for identifying the MJO during boreal summer, whereas upper-tropospheric anomalies are most important during boreal winter.
Mechanistic studies of the MJO commonly depend on accurate definitions of when each MJO seasonal mode occurs, since the spatial structures of the winter and summer modes differ so substantially (Fig. 9). Should the seasonal definitions of the MJO be inaccurate, then there is a risk that the mechanistic studies themselves are not targeting processes specific to each season. We therefore use the backward optimization interpretations of the neural networks to define the MJO seasonal modes. To do so, we spatially correlate the optimal patterns for each state variable to the optimal patterns on 10 January and 1 August, which are generally considered to be the peak of the boreal winter and boreal summer modes. We then define the boreal winter mode to exist during periods for which the optimized MJO patterns have a correlation of greater than 0.75 with the optimized pattern for 10 January and similarly define the boreal summer mode to exist when the correlation is greater than 0.75 with the optimized pattern for 1 August. Using our definition, the seasonality differs across atmospheric state variables, although the boreal winter and summer modes generally span from late November through early March and early June through early October, respectively (dark colors in Fig. 10). Lower-tropospheric variables generally lead the transition from the boreal winter mode to the equinoctial transition toward the boreal summer mode, although a less clear relationship exists during the transition back to the boreal winter mode.
Finally, we define extended boreal winter as the period during which the correlation between each weekly optimal pattern and the 10 January optimal pattern is greater than that between the weekly optimal patterns and the 1 August optimal pattern. Extended boreal summer spans the rest of the year. Using this definition, extended boreal winter MJO extends from early November through late April across most state variables and from mid-November through late April for outgoing longwave radiation in particular (dark and light colors in Fig. 10). Many studies of the MJO have previously used extended winter and summer seasons which span the months of October through March and April through September, respectively (Yoo and Son, 2016; Zhang and Zhang, 2018). Our results suggest that these extended seasons should, at a minimum, be shifted 1 month later in the year.
We have tested the ability of interpretable neural networks to identify complex, multi-scale geophysical phenomena via their application to the Madden–Julian oscillation (MJO). We first evaluated whether neural networks can identify the MJO and then used neural network interpretability methods to study the seasonality and spatial structure of the MJO and its relationship to atmospheric state variables. Our study therefore contributes both to the general usage of neural networks within geoscience and to knowledge of the MJO itself, so we separate our discussion of the implications for both communities below.
4.1 Implications for neural networks in earth science
We have shown that neural networks are highly interpretable, even for complex, multi-scale geophysical phenomena. Two methods proposed by the computer science community – backward optimization and layerwise relevance propagation – provide particularly useful interpretations of neural networks (Toms et al., 2020a). Namely, backward optimization offers composite interpretations, while layerwise relevance propagation enables interpretations on either a composite or case-by-case basis. Both methods project the decision-making process of a neural network back onto the original dimensions of the input, which is particularly useful for geoscientific applications where each input variable may have unique physical importance to the problem being studied.
The capability of neural networks to include nonlinearities and simultaneously model different input patterns that lead to similar outputs proved useful for studying the seasonality of the MJO. The neural networks identified the phase of the MJO twice as accurately as the multi-output linear regression, which implies that interpretations of the neural network characterize the MJO more accurately than the linear regression approach. We hypothesized that the increase in accuracy was caused by the neural networks' ability to model the uniqueness of each MJO event, which is not feasible using conventional linear approaches such as regression. The amount of neural network complexity required for tasks across the geosciences will vary greatly, so the benefits of interpretable neural networks are also likely to vary across sub-disciplines. We have found that a baseline approach of comparing the accuracy of neural networks to more simple methods such as linear regression is useful in determining the necessity of a neural network.
Based on this study and other supporting work (Toms et al., 2020a), the interpretations of what a neural network learns can be used to advance geoscientific knowledge. Even for cases where interpretability is not the main objective, neural network interpretations can offer insights into how and why neural networks are making their decisions and can be used to ensure that neural networks are making decisions using reasoning consistent with physics. While we use a relatively simple type of neural network, the proposed methods are applicable to other types of neural networks as well, such as convolutional neural networks and long short-term memory (LSTM) networks. We found fully connected networks to be particularly useful for our application and more accurate than convolutional neural networks, which, in light of the surging popularity of convolutional neural networks within geoscience, suggests that fully connected networks also have utility for geospatial problems.
4.2 Implications for the Madden–Julian oscillation
We also used neural networks as an approach to better understand the spatial structure and seasonality of the MJO. Our results are generally consistent with the thorough body of literature on the MJO, which supports the reliability and robustness of interpretable neural networks within geoscience.
Consistent with previous studies, we find that the spatial structure of the MJO generally exhibits two dominant modes of variability distinguished between the boreal summer and winter. We find that the extended boreal winter mode of the MJO occurs between early November and late April, with the boreal summer mode occurring throughout the remainder of the year. This definition of the extended seasons is delayed 1 month compared to conventional definitions, which use an extended boreal winter of October through March. Furthermore, the seasonality of the relationship between the MJO and atmospheric state variables is more complex, with each variable exhibiting a unique seasonality. Some state variables such as lower-tropospheric zonal winds exhibit a uni-modal seasonality, whereas others such as upper-tropospheric zonal winds exhibit a bi-modal seasonality. We also find that upper-tropospheric thermodynamic anomalies are particularly useful in identifying the MJO during boreal winter, which may relate to the enhanced coupling between the MJO and stratospheric processes during this season.
Consistent with previous studies, we find that the spatial structure of the MJO generally exhibits two dominant modes of variability distinguished between the boreal summer and winter. We also extend our analysis to test numerous aspects of the MJO, from its nonlinearities to its relationships with atmospheric state variables. The key points of this analysis are as follows:
The neural networks identify the phase of the MJO twice as accurately as the multi-output linear regression approach, which suggests that nonlinearities are important to the structure of the MJO. These nonlinearities are reflected in the spatial uniqueness of each MJO event, given that the composite structure of the MJO identified by the neural networks and linear methods are remarkably similar (Figs. 5, 6).
Each state variable exhibits a unique seasonality in its relationship with the MJO. For example, some state variables such as lower-tropospheric zonal winds exhibit a uni-modal seasonality, whereas others such as upper-tropospheric zonal winds exhibit a bi-modal seasonality (Figs. 8, 9).
Upper-tropospheric thermodynamic anomalies are particularly important for identifying the MJO during boreal winter, which may relate to the enhanced coupling between the MJO and stratospheric processes during this season (Fig. 8).
We find that the extended boreal winter mode occurs between early November and late April, while the boreal summer mode occurs throughout the remainder of the year. This definition of the extended seasons is delayed 1 month compared to the conventional definition, which uses an extended boreal winter of October through March (Fig. 10).
Our results show that neural networks are highly interpretable, even for spatially complex geoscientific applications. Because of the high reliability of the interpretations, neural networks are viable tools for testing hypotheses related to the MJO and other spatially complex geophysical phenomena. More complex hypotheses can now be tested: for example, does horizontal advection of the lower-tropospheric mean moisture by the MJO circulation govern the propagation of the MJO (e.g., Jiang et al., 2018)? Or a neural network can be used to identify whether an MJO event will initiate given spatial inputs of atmospheric variables, from which interpretability methods can identify important patterns for MJO initiation. An example method for such a project would be to input maps of atmospheric variables and output whether an MJO event initiates at some point in the future, similar to the categorical output in this study. A critical requirement for using neural networks in such studies is the proven ability to reliably interpret what the networks have learned, which is now possible.
All data used in this study and an example script for training a neural network and generating the LRP heatmaps and optimal input fields are available at the following DOI: https://doi.org/10.5281/zenodo.3968896 (Toms, 2020). The unprocessed versions of the MERRA-2 reanalysis data used in this study are available for download at the National Aeronautics and Space Administration (NASA) Goddard Space Flight Center (GSFC) Global Modeling and Assimilation Office website: https://gmao.gsfc.nasa.gov/reanalysis/MERRA-2/ (Gelaro et al., 2017). The unprocessed versions of the NOAA once-daily outgoing longwave radiation climate data record are available for download from the following website: https://www.ncdc.noaa.gov/cdr/atmospheric/outgoing-longwave-radiation-daily (Lee and NOAA CDR Program, 2014). The unprocessed MJO OMI data are available from the National Oceanic and Atmospheric Administration Physical Sciences Laboratory (NOAA PSL) at the following website: https://www.psl.noaa.gov/mjo/mjoindex/ (NOAA, 2020).
BAT, KK, P, and DY conceived the idea and designed the experiment. KK and P provided expertise on neural networks throughout the project. BAT performed the analysis and wrote the paper.
The authors declare that they have no conflict of interest.
This research has been supported by the U.S. Department of Energy (grant no. DE-FG02-97ER25308), the U.S. Department of Energy, Office of Science (grant no. DE-AC02-05CH11231), the U.S. Department of Energy, Office of Science (grant no. DE-AC02-05CH11231), the National Institute of Food and Agriculture (grant no. CA-D-LAW-2462-RR), and the Packard Foundation for Science and Engineering.
This paper was edited by Richard Neale and reviewed by two anonymous referees.
Abhik, S. and Hendon, H. H.: Influence of the QBO on the MJO during coupled model multiweek forecasts, Geophys. Res. Lett., 46, 9213–9221, 2019. a
Adames, Á. F. and Kim, D.: The MJO as a dispersive, convectively coupled moisture wave: Theory and observations, J. Atmos. Sci., 73, 913–941, 2016. a
Adames, Á. F. and Wallace, J. M.: Three-dimensional structure and evolution of the MJO and its relation to the mean flow, J. Atmos. Sci., 71, 2007–2026, 2014. a
Adames, Á. F. and Wallace, J. M.: Three-dimensional structure and evolution of the moisture field in the MJO, J. Atmos. Sci., 72, 3733–3754, 2015. a
Bach, S., Binder, A., Montavon, G., Klauschen, F., Müller, K.-R., and Samek, W.: On pixel-wise explanations for non-linear classifier decisions by layer-wise relevance propagation, PloS one, 10, e0130140, https://doi.org/10.1371/journal.pone.0130140, 2015. a
Barnes, E. A., Hurrell, J. W., Ebert-Uphoff, I., Anderson, C., and Anderson, D.: Viewing forced climate patterns through an AI Lens, Geophys. Res. Lett., 46, 13389–13398, 2019. a
Brenowitz, N. D. and Bretherton, C. S.: Prognostic validation of a neural network unified physics parameterization, Geophys. Res. Lett., 45, 6289–6298, 2018. a
Chen, T. and Chen, H.: Universal approximation to nonlinear operators by neural networks with arbitrary activation functions and its application to dynamical systems, IEEE T. Neural Netw., 6, 911–917, 1995. a
Densmore, C. R., Sanabia, E. R., and Barrett, B. S.: QBO influence on MJO amplitude over the Maritime Continent: Physical mechanisms and seasonality, Mon. Weather Rev., 147, 389–406, 2019. a
Gelaro, R., McCarty, W., Suárez, M. J., Todling, R., Molod, A., Takacs, L., Randles, C. A., Darmenov, A., Bosilovich, M. G., Reichle, R., Wargan, K., Coy, L., Cullather, R., Draper, C., Akella, S., Buchard, V., Conaty, A., da Silva, A. M., Gu, W., Kim, G., Koster, R., Lucchesi, R., Merkova, D., Nielsen, J. E., Partyka, G., Pawson, S., Putman, W., Rienecker, M., Schubert, S. D., Sienkiewicz, M., and Zhao, B.: The modern-era retrospective analysis for research and applications, version 2 (MERRA-2), J. Climate, 30, 5419–5454, 2017. a, b
Goodfellow, I., Bengio, Y., and Courville, A.: Deep learning, MIT press, available at: https://books.google.com/books?hl=en&lr=&id=omivDQAAQBAJ&oi=fnd&pg=PR5&dq=Goodfellow,+I.,+Bengio,+Y.,+and+Courville,+A.:+Deep+learning&ots=MNO3arszWQ&sig=NLK3dUkGYpMXR4-kZ1z7yD9O-7g#v=onepage&q=Goodfellow%2C I.%2C Bengio%2C Y.%2C and Courville%2C A.%3A Deep learning&f=false (last access: 31 December 2020), 2016. a
Hendon, H. H. and Liebmann, B.: Organization of convection within the Madden-Julian oscillation, J. Geophys. Res.-Atmos., 99, 8073–8083, 1994. a
Jiang, X., Adames, Á. F., Kim, D., Maloney, E. D., Lin, H., Kim, H., Zhang, C., DeMott, C. A., and Klingaman, N. P.: Fifty Years of Research on the Madden-Julian Oscillation: Recent Progress, Challenges, and Perspectives, J. Geophys. Res.-Atmos., 125, e2019JD030911, https://doi.org/10.1029/2019JD030911, 2020. a
Kiladis, G. N., Dias, J., Straub, K. H., Wheeler, M. C., Tulich, S. N., Kikuchi, K., Weickmann, K. M., and Ventrice, M. J.: A comparison of OLR and circulation-based indices for tracking the MJO, Mon. Weather Rev., 142, 1697–1715, 2014. a, b, c
Lee, H.-T. and NOAA CDR Program: NOAA Climate Data Record (CDR) of Daily Outgoing Longwave Radiation (OLR), Version 1.2, NOAA National Climatic Data Center, https://doi.org/10.7289/V5SJ1HH2, 2011. a, b
Liebmann, B. and Smith, C. A.: Description of a complete (interpolated) outgoing longwave radiation dataset, B. Am. Meteorol. Soc., 77, 1275–1277, 1996. a
Madden, R. A. and Julian, P. R.: Detection of a 40–50 day oscillation in the zonal wind in the tropical Pacific, J. Atmos. Sci., 28, 702–708, 1971. a
Martin, Z., Vitart, F., Wang, S., and Sobel, A.: The impact of the stratosphere on the MJO in a forecast model, J. Geophys. Res.-Atmos., 125, e2019JD032106, https://doi.org/10.1029/2019JD032106, 2020. a
Matsuno, T.: Quasi-geostrophic motions in the equatorial area, J. Meteorol. Soc. Jpn. Ser. II, 44, 25–43, 1966. a
McGovern, A., Lagerquist, R., John Gagne, D., Jergensen, G. E., Elmore, K. L., Homeyer, C. R., and Smith, T.: Making the black box more transparent: Understanding the physical implications of machine learning, B. Am. Meteorol. Soc., 100, 2175–2199, 2019. a
Montavon, G., Lapuschkin, S., Binder, A., Samek, W., and Müller, K.-R.: Explaining nonlinear classification decisions with deep taylor decomposition, Pattern Recogn., 65, 211–222, 2017. a
Monteiro, J. M., Adames, Á. F., Wallace, J. M., and Sukhatme, J. S.: Interpreting the upper level structure of the Madden-Julian oscillation, Geophys. Res. Lett., 41, 9158–9165, 2014. a
Olah, C., Mordvintsev, A., and Schubert, L.: Feature visualization, Distill, 2, e7, 2017. a
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, 2018. a
Roundy, P. E. and Frank, W. M.: Applications of a multiple linear regression model to the analysis of relationships between eastward-and westward-moving intraseasonal modes, J. Atmos. Sci., 61, 3041–3048, 2004. a
Roundy, P. E., MacRitchie, K., Asuma, J., and Melino, T.: Modulation of the global atmospheric circulation by combined activity in the Madden–Julian oscillation and the El Niño–Southern Oscillation during boreal winter, J. Climate, 23, 4045–4059, 2010. a
Sobel, A. and Maloney, E.: Moisture modes and the eastward propagation of the MJO, J. Atmos. Sci., 70, 187–192, 2013. a
Son, S.-W., Lim, Y., Yoo, C., Hendon, H. H., and Kim, J.: Stratospheric control of the Madden–Julian oscillation, J. Climate, 30, 1909–1922, 2017. a
Toms, B. A., Barnes, E. A., and Ebert-Uphoff, I.: Physically interpretable neural networks for the geosciences: Applications to earth system variability, J. Adv. Model. Earth Sy., 12, e2019MS002002, https://doi.org/10.1029/2019MS002002, 2020a. a, b, c, d, e, f
Toms, B. A., Barnes, E. A., Maloney, E. D., and van den Heever, S. C.: The Global Teleconnection Signature of the Madden-Julian Oscillation and its Modulation by the Quasi-Biennial Oscillation, J. Geophys. Res.-Atmos., 25, e2020JD032653, https://doi.org/10.1029/2020JD032653, 2020b. a, b
Tseng, K.-C., Maloney, E., and Barnes, E.: The consistency of MJO teleconnection patterns: an explanation using linear Rossby wave theory, J. Climate, 32, 531–548, 2019. a
Weyn, J. A., Durran, D. R., and Caruana, R.: Can Machines Learn to Predict Weather? Using Deep Learning to Predict Gridded 500-hPa Geopotential Height From Historical Weather Data, J. Adv. Model. Earth Sy., 11, 2680–2693, 2019. a
Yang, D. and Ingersoll, A. P.: Testing the hypothesis that the MJO is a mixed Rossby–gravity wave packet, J. Atmos. Sci., 68, 226–239, 2011. a
Yang, D. and Ingersoll, A. P.: Triggered convection, gravity waves, and the MJO: A shallow-water model, J. Atmos. Sci., 70, 2476–2486, 2013. a
Yoo, C. and Son, S.-W.: Modulation of the boreal wintertime Madden-Julian oscillation by the stratospheric quasi-biennial oscillation, Geophys. Res. Lett., 43, 1392–1398, 2016. a
Zhang, C. and Dong, M.: Seasonality in the Madden–Julian Oscillation, J. Climate, 17, 3169–3180, https://doi.org/10.1175/1520-0442(2004)017<3169:SITMO>2.0.CO;2, 2004. a
Zhang, C. and Zhang, B.: QBO-MJO Connection, J. Geophys. Res.-Atmos., 123, 2957–2967, 2018. a
Zhao, C., Li, T., and Zhou, T.: Precursor signals and processes associated with MJO initiation over the tropical Indian Ocean, J. Climate, 26, 291–307, 2013. a