Time, material resources and human labor required to administer experiments are often not scalable for a deep understanding of the possible results based on varying input combinations – hence, the need for models such as neural networks to learn these relationships. An ideal model would serve to predict the outcomes of physical experiments for all if not most possible conditions. However, it may be inconceivable to collect thousands or millions of training instances for the model as performing these many experiments may in fact leave the idea of building a model infeasible and thus, pointless for actual physical application. What occurs often is that training datasets are riddled with redundancies that do not provide any new information to these models and in fact impair generalization strength [1, 2]. If there existed a procedure to select training instances based on current knowledge of the parameter space – thereby dynamically building a training dataset – this would possibly lessen costs of experimentation by choosing well-crafted inputs that provide unknown information of the results and train an accurate model from a small yet sufficiently informative training dataset.
This paper introduces an effective methodology to achieve this goal and is tested in the context of two diagnostics in plasma physics (background provided in Section 5) – inelastic X-ray Thomson scattering (XRTS) [3–5] and X-ray emission spectroscopy (XES) [6, 7] – by training multiple convolutional neural networks simultaneously and using their impressions on the outputs as a form of uncertainty to carefully select instances to train on next. This approach constructs a feedback loop on training models and selecting training instances based on what the models are unsure of. We note that the algorithm presented has produced effective training datasets for deep learning models that are approximately 90% smaller in size than those of current models – with XRTS a 3 dimensional problem and XES a 10 dimensional.
The overarching goal of the algorithm is to produce a suitable training dataset from a continuous parameter space by selecting new points to add based on the information provided by instances already collected. Orthogonal sampling is often a statistical method used in generating a pseudo-random sample of a multidimensional space, popular in numerical integration for Monte-carlo applications and for computer experiments [8]. The basic premise is that the parameter space is divided into blocks and points are generated within each block to create a representative sample across the input space. We use this idea to construct a systematic scheme to represent different regions of space to which we associate uncertainty with our predictions. Suppose our parameter space is the n-unit cube of (inputs can be scaled if necessary). We divide each of the standard intervals of the axes of our input space into finer, disjoint intervals and together, a block representation is formed. The algorithm should store instances to the blocks they are contained in (i.e. the dataset) with their corresponding uncertainties. The uncertainty figures aggregated in each block provide a general idea on how confident the models are on the instances within the block, which the algorithm can use to navigate the input space; in our study, we calculate the average uncertainty as a heuristic for this aggregation with a decaying factor attributed to the uncertainties of older instances collected.
In short, the knowledge of uncertainty is gathered by multiple deep learning models that are concurrently trained by the algorithm with the key idea being that points where the models disagree on the most are points of most uncertainty in predictions. We note that the inspiration for this approach was from Section 2.2.4 of [9] to select instances based on highest variance across ensemble members of reward predictors in Deep Reinforcement Learning. With l models employed, the standard variances along the coordinates of the l outputs of an input x in the n-unit cube of is computed. The uncertainty attributed to x across all models used in this paper is simply the sum of these aforementioned coordinate variances. This allows us to select instances to add to the dataset at each epoch of the algorithm from some set of points.
Now it remains to define some criteria to select blocks for an epoch of the algorithm as it may be infeasible to always consider all of them (e.g. a simple 2 block division across each parameter of XES results in 1024 blocks). The whole basis of the algorithm is to choose blocks that the models identify as uncertain regions. However, the issue by always following this protocol is that other blocks that haven’t had adequate attention could reflect regions of greater uncertainty than where the models believe they are most uncertain in. In this paper, we select random blocks or blocks with the lowest number of points for XRTS and XES, respectively, at the start of the algorithm and every so often as a solution. In addition, with uncertainty solely based on the perspective of the models of the algorithm, there may be instances that the models are ostensibly confident on, but it may be the case that the true output is very different from what all the models believe. To combat this possible bias, we use a decaying -schedule of probabilities across the epochs of the algorithm to either select random or most uncertain instances from a set of points in the parameter space to add to the dataset. The decaying aspect arises because the models are ideally less biased with regards to this criteria as the algorithm progresses and are appropriately choosing points of highest uncertainty.
Towards the middle stages of the algorithm, training is performed in every epoch with respect to a fixed number of the most recently added instances to the dataset and periodically on the entire dataset. The reason for this is to introduce robustness to the models’ uncertainty calculations because as we include more training instances, the hope would be that the most recently added points are a more accurate representation of regions of uncertainty in space and therefore, reveal most informatively where the models have a lack of confidence on inputs. The pseudo-code is displayed in Algorithm 1.
Figure 1 outlines the model structure implemented in this paper. The convolutional layers are all created with a kernel size of 3, stride of 1, and zero padding of 1. Similarly, the transposed convolutional layers are associated with a kernel size of 2, stride of 1 and no zero padding for the first 4 ResConv modules (defined in Figure 1) followed by zero padding of 1. The attention layer is inspired from [10] in order to capture global dependencies of the output signal [11, 12]. The function h in Figure 1 takes the first of the 15 input channels and f and g splits the remaining channels evenly. We see that h acts as the convolutional information aspect of the attention layer whereas f and g create the attention map that acts on h. The attention layer has two learnable parameters: one for the
Figure 1: The deep learning model architecture consists of fully connected layers, transposed convolutional layers, convolutional layers, and an attention layer. The and
denote skip connections and matrix multiplication, respectively. Note that (n x) m means there are n channels, each m entries long and n x m denotes the standard n-by-m matrix. The ResConv module repeats 6 times beginning with an input of 32 channels of length 4.
skip connection from , and the other from the attention map applied to
. The neural network training was performed with the ADAM optimizer [13] to minimize the mean squared error.
The block division for both XRTS and XES was of 2 partitions across each parameter, resulting in 8 and 1024 blocks, respectively. Three models were used in both tests. To assess the degree of success of the algorithm, validation and test datasets of 3000 and 4000 instances, respectively, were generated for each test case with random parameter values over a predefined range that are valid in each context. The algorithm ran for about 9 and 24 hours for XRTS and XES, respectively, on a Titan X GPU card.
The final validation and test MSE losses for the XRTS case corresponding to the model of the lowest validation loss were 1.1836 x and 1.9273 x
, respectively. For XES, the corresponding losses were 7.273 x
and 7.674 x
, respectively. First and foremost, we note that in Figure 2(a) and (b), the models are seemingly overfitting their respective training datasets. An understandable factor for this occurrence could be the extensive training on these models with respect to the small training datasets. A critical indicator of the success of the uncertainty usage is that for both the XRTS and XES test case, the models repeatedly get “surprised" when we analyze Figures 2(a) and (b). The frequent increases in training losses of the models signify that the newly added instances provide possibly unfamiliar information to these models of the parameter space.
Figure 2: XRTS and XES MSE losses during each epoch. The three models used for each test case in the algorithm are the three labeled models. The final retraining in XES is performed with the model weights of the lowest validation loss over the dataset created by the algorithm.
Figure 3: Predictions from XRTS and XES models compared to true outputs. We can see that the model predictions are well in agreement with the outputs.
Alternatively, the general trends of validation losses in Figure 2(c) and (d) are that they decrease as the algorithm progresses but the rates of decay are quite different. By epoch 50, XRTS models explore a large portion of the uncertain regions in its parameter space reaching an MSE validation loss of approximately 6.25 x across all models; the remaining 190 epochs slowly but generally reduce the validation loss until a final value of 1.3 x
. A reason for this trajectory of validation losses might be because of the relatively small number of compact blocks to investigate, which allows the models to obtain an intuitive idea of the variation of the outputs associated with the entire parameter space early in the algorithm. In contrast, the many more blocks in XES causes the algorithm to possibly require more time to find which part of space (i.e. which blocks) the models are most uncertain in. This is most visible with the drastic decrease of the validation losses by 37.9% across all 3 models to 1.678 x
over the course of epochs 160 to 200. Overall, it may have been the case that if the algorithm was allowed more time to survey the parameter space, another rich source of information may have been located. In addition, the algorithm may also be punishing the XES models for their incorrect perceptions of uncertainties with the slight increase of validation loss before the aforementioned drop. In XES, the final 200 epochs train the model weights of the lowest validation loss of 1.208 x
on the entire the training dataset generated in the first 360 epochs, decreasing the MSE validation by approximately 39.8%. In summary, the XES models spend significantly more time exploring than the XRTS models before they finally recognize where to focus their attention with regards to uncertain inputs. The final XRTS training dataset contained 774 points, which is roughly 11.06% of the current model’s. Similarly, the XES dataset included 912 instances or 13.03% of its current model’s.
In conclusion, the algorithm provided a methodical collection of a training dataset based on uncertainty in predictions and trains deep learning models to learn from this information. This has reduced the amount of training data by an order of magnitude without severely impacting performance.
XRTS and XES were selected as case studies because these methodologies are applied frequently in plasma physics experiments [3–5, 14, 15].
X-ray Thomson scattering (XRTS) XRTS is a procedure to probe the temperature and density of plasma by emitting an X-ray pulse into the plasma and measuring the spectrum of the scattered pulse with respect to an angle. This spectrum can be identified once the angle and the condition of the plasma is known [16]. This paper deals with the case of only 3 parameters – temperature, ionization, and density – are retrieved from the scattering of an X-ray at 90 degrees on a Beryllium sample. A single simulation takes approximately 10-20 seconds for completion.
X-ray emission spectroscopy (XES) XES functions as another method to probe plasma conditions by measuring the emitted spectrum of X-ray and comparing results with simulations or theoretical models. We consider the case of an implosion condition [6] from a Germanium emission spectrum with the model outlined in [7]. A simulation takes approximately 10-20 minutes to execute.
[1] Lucila Ohno-Machado, Hamish S Fraser, and A Ohrn. Improving machine learning performance by removing redundant cases in medical data sets. In Proceedings of the AMIA Symposium, page 523. American Medical Informatics Association, 1998.
[2] Kashyap Chitta, Jose M Alvarez, Elmar Haussmann, and Clement Farabet. Less is more: An exploration of data redundancy with active dataset subsampling. arXiv preprint arXiv:1905.12737, 2019.
[3] HJ Lee, P Neumayer, J Castor, T Döppner, RW Falcone, C Fortmann, BA Hammel, AL Kritcher, OL Landen, RW Lee, et al. X-ray thomson-scattering measurements of density and temperature in shock-compressed beryllium. Physical review letters, 102(11):115001, 2009.
[4] Andrea L Kritcher, Paul Neumayer, John Castor, Tilo Döppner, Roger W Falcone, Otto L Landen, Hae Ja Lee, Richard W Lee, Edward C Morse, Andrew Ng, et al. Ultrafast x-ray thomson scattering of shock-compressed matter. Science, 322(5898):69–71, 2008.
[5] JC Valenzuela, C Krauland, D Mariscal, I Krasheninnikov, C Niemann, T Ma, P Mabey, G Gregori, P Wiewior, AM Covington, et al. Measurement of temperature and density using non-collective x-ray thomson scattering in pulsed power produced warm dense plasmas. Scientific reports, 8(1):8432, 2018.
[6] SP Regan, R Epstein, BA Hammel, LJ Suter, HA Scott, MA Barrios, DK Bradley, DA Callahan, C Cerjan, GW Collins, et al. Hot-spot mix in ignition-scale inertial confinement fusion targets. Physical review letters, 111(4):045001, 2013.
[7] Orlando Ciricosta, H Scott, P Durey, BA Hammel, R Epstein, TR Preston, SP Regan, SM Vinko, NC Woolsey, and JS Wark. Simultaneous diagnosis of radial profiles and mix in nif ignition-scale implosions via x-ray spectroscopy. Physics of Plasmas, 24(11):112703, 2017.
[8] Boxin Tang. Orthogonal array-based latin hypercubes. Journal of the American statistical association, 88(424):1392–1397, 1993.
[9] Paul F Christiano, Jan Leike, Tom Brown, Miljan Martic, Shane Legg, and Dario Amodei. Deep reinforcement learning from human preferences. In Advances in Neural Information Processing Systems, pages 4299–4307, 2017.
[10] Han Zhang, Ian Goodfellow, Dimitris Metaxas, and Augustus Odena. Self-attention generative adversarial networks. arXiv preprint arXiv:1805.08318, 2018.
[11] Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N Gomez, Łukasz Kaiser, and Illia Polosukhin. Attention is all you need. In Advances in neural information processing systems, pages 5998–6008, 2017.
[12] Xiaolong Wang, Ross Girshick, Abhinav Gupta, and Kaiming He. Non-local neural networks. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pages 7794–7803, 2018.
[13] Diederik P Kingma and Jimmy Ba. Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980, 2014.
[14] SM Vinko, O Ciricosta, BI Cho, K Engelhorn, H-K Chung, CRD Brown, T Burian, J Chalupsk`y, RW Falcone, C Graves, et al. Creation and diagnosis of a solid-density plasma with an x-ray free-electron laser. nature, 482(7383):59, 2012.
[15] QY van den Berg, EV Fernandez-Tello, T Burian, J Chalupsk`y, H-K Chung, O Ciricosta, GL Dakovski, V Hájková, P Hollebon, L Juha, et al. Clocking femtosecond collisional dynamics via resonant x-ray spectroscopy. Physical review letters, 120(5):055002, 2018.
[16] G. Gregori, S. H. Glenzer, W. Rozmus, R. W. Lee, and O. L. Landen. Theoretical model of x-ray scattering as a dense matter probe. Phys. Rev. E, 67:026412, Feb 2003.