Discovering hidden physics using ML-based multimodal super-resolution measurement and its application to fusion plasmas

Abstract

A non-linear complex system governed by multi-spatial and multi-temporal physics scales cannot be fully understood with a single diagnostic, as each provides only a partial view and much information is lost during data extraction. Combining multiple diagnostics also results in imperfect projections of the system’s physics. By identifying hidden inter-correlations between diagnostics, we can leverage mutual support to fill in these gaps, but uncovering these inter-correlations analytically is too complex. We introduce a groundbreaking machine learning methodology to address this issue. Our multimodal approach generates super-resolution data encompassing multiple physics phenomena, capturing detailed structural evolution and responses to perturbations previously unobservable. This methodology addresses a critical problem in fusion plasmas: the Edge Localized Mode (ELM), a plasma instability that can severely damage reactor walls. One method to stabilize ELM is using resonant magnetic perturbation to trigger magnetic islands. However, low spatial and temporal resolution of measurements limits the analysis of these magnetic islands due to their small size, rapid dynamics, and complex interactions within the plasma. With super-resolution diagnostics, we can experimentally verify theoretical models of magnetic islands for the first time, providing unprecedented insights into their role in ELM stabilization. This advancement aids in develo** effective ELM suppression strategies for future fusion reactors like ITER and has broader applications, potentially revolutionizing diagnostics in fields such as astronomy, astrophysics, and medical imaging.

1 Introduction

In complex physical systems, diagnostic measurements are often intricately interconnected through fundamental physical principles. These underlying connections stem from the laws of nature that govern the behavior of matter and energy. For instance, electromagnetic events couple the measured signals, and equations of state relate variables such as pressure, volume, and temperature, providing a framework to infer one quantity from others. Similarly, coupled differential equations in fluid dynamics or plasma physics describe how multiple system parameters evolve interdependently over time. Such relationships are particularly evident in nuclear fusion, the focus field of this work, which is characterized by its intricate interplay of various physical phenomena.

The nuclear fusion technology aimed at producing eco-friendly energy is rapidly advancing through the synergy of academia and industry [1, 2, 3]. The success of nuclear fusion is fundamentally based on maintaining high-temperature, high-pressure hydrogen plasma without becoming unstable. It was recently shown that Artificial Intelligence (AI) can be a helpful tool to achieve that goal [4, 5, 6, 3]. Experimental fusion reactors like DIII-D [7] utilize various diagnostics for effective plasma monitoring necessary for this AI application [8]. For example, the Electron Cyclotron Emission (ECE) diagnostic system measures electron temperature [9], CO2 interferometer (Interferometer) measures electron density and its fluctuations [10], Motional Stark Effect (MSE) measures the magnetic field [11], and Thomson Scattering (TS) measures the electron temperature and density [12]. The different measurements each capture different physical properties, and form a complementary set for extracting as much information from the plasma as possible. Although it is likely that there exists some kind of correlation or coupling between the measurements of different diagnostics (see Section 6.1), our current scientific understanding is still not capable of specifying some of these relationships analytically. Machine Learning (ML) is a powerful tool for identifying hidden relationships in data [13]. Learning the hidden relationships among different diagnostics would be a great asset to enhance their measurements, and it also helps to find a minimal set of diagnostics for a future reactor in which the availability of diagnostics is limited due to the cost and hardware constraints.

One of the most critical issues for fusion reactors is the Edge Localized Mode (ELM), an instability that occurs at the plasma edge under high-confinement conditions. This edge instability delivers transient and intense heat flux outward, which can cause irreversible damage to the wall in a reactor-scale device. Therefore, understanding and controlling this phenomenon is a major challenge that must be resolved [5, 14]. However, the detailed physical mechanism of ELM s and structural response to the external field occurring within milliseconds are still subjects of ongoing debate. High-frequency diagnostics like ECE and Interferometer possess sufficient time resolution to track these fast dynamics, but their limited spatial resolution and measurement conditions pose challenges in clearly observing the structural characteristics of ELM s. On the other hand, TS offers high spatial resolution capable of observing detailed structures, but its temporal resolution is too low to elucidate the exact mechanism of ELM s.

The current remedy to this issue is a specific operational method for TS, known as “bunch mode”, to increase the sampling rate of up to $10\text{\,}\mathrm{kHz}$ [15, 16]. Despite its high pulse repetition, firing TS in “bunch mode” is limited by the heat capacity of the laser medium and limited measurement repetition. Therefore such an approach is typically reserved for very short periods of time or specific experiments where high-resolution temporal data is crucial [16].

Instead, we hypothesize that a data-driven ML model, so-called Diag2Diag, with multimodal inputs comprising the high-frequency diagnostics can effectively make use of internal correlations in order to estimate TS. This can enhance the temporal resolution of the existing TS diagnostics without upgrading hardwares, so-called Super-Resolution TS (SRTS) diagnostics, which enables deeper physical analysis of plasma behavior.

Various fields have developed ML-based spatial or temporal resolution enhancement techniques, but these mostly involve uni-modal resolution enhancement by learning linear or nonlinear interpolation of the given information [17, 18, 19, 20, 21]. These are applicable only to regularly sampled data and are challenging to generate finer-scale phenomena undetectable at the time resolution of the target sensor (more detail in Section 6.2). Our work goes beyond plausible interpolation; it is a physics-preserving super-resolution to reconstruct events missed by target diagnostics, by learning the correlation between different diagnostic measurements, which is, to our best knowledge, the first attempt of its kind.

Figure 1: Main methodology. DiagA is essential to capture fast transient events near the edge of plasma. But due to its low temporal resolution and accuracy it fails to track the evolution of such events. Diag2Diag solves this problem by generating synthetic super resolution of DiagA by learning the correlation between DiagA data and other diagnostic measurements with higher resolutions and better accuracy.

Figure 1 summarizes the main methodology for this work. DIII-D utilizes hundreds of diagnostics for monitoring the plasma. These diagnostics measure various characteristics of plasma at different temporal resolutions. A potential ML model can learn the intrinsic correlations among diagnostics data and thus generate one from others. This works for both, time-series and spectrograms, although different variants of Artificial Neural Network (ANN) are used. The design choices and the optimization and training strategies are described in the following sections.

2 ML-based map** between different diagnostics

For develo** an ML-based SRTS diagnostic from other diagnostics, it is essential to verify the existence, strength, and robustness of correlation among them. We therefore approach this in several steps as described subsequently.

The aim of the first step is to show that we can reconstruct the spectrograms of one diagnostic based on another. As was discussed in Section 1, it is very likely that different diagnostics data have intrinsic correlations. Certain plasma instabilities and modes, such as ELM and Alfvén Eigenmode (AE), affect both electron temperature, which is measured by ECE, and density fluctuations measured by Interferometer. We now show that an ANN is able to learn this relationship during AE modes by map** from ECE spectrograms to Interferometer spectrograms as illustrated in Figure 2.

Figures 2(b) and (d) show example spectrograms obtained from the raw signals of ECE and Interferometer, respectively, whose measurement positions and paths can be seen in Figure 2(a). We designed and trained a Convolutional Neural Network (CNN) that takes 40 ECE spectrograms as input and reconstructs 4 target Interferometer spectrograms, as shown in Figure 2(c). The reconstructed synthetic Interferometer spectrograms visually confirm the plausible reconstruction of features such as frequency chir** and harmonics as seen in Figure 2(d).

Figure 2: Reconstructing Interferometer spectrograms from ECE spectrograms of DIII-D shot 170669 using convolutional neural networks. (a) The configuration of 4 ECE and 40 ECE probes at DIII-D . (b) A tensor of $(40\times time\times frequency)$ is supplied to CNN. (c) The configuration of CNN. (d) Visual comparison of measured and reconstructed spectrograms (e) Comparison of the Alfven Eigenmode detector output [22] supplied with the measured and reconstructed spectrograms.

Besides the visual comparison, we are also interested in how much the underlying physical information is preserved using this method. Therefore, we evaluated the preservation of physical information by performing a downstream task, AE instability detection [22], based on the measured and the reconstructed Interferometer spectrograms. In Figure 2(e), we can see that the AE scores obtained from the reconstructed spectrograms (blue) closely match those from the measured spectrograms (red). This demonstrates that the results generated by ML contain sufficient hidden physical information, and thus it is supported that ML can extract the intrinsic correlation among diagnostic data. We can now take a step further and tackle the generation of TS signals from other diagnostics based on raw time-series, not a spectrogram.

3 Multimodal super-resolution diagnostic

In this section, we switch from spectrograms to time-series signals and show that the amplitude of a diagnostics can be reconstructed from other diagnostics, while preserving intrinsic physics. More importantly, we will show that if the input diagnostics are of much higher temporal resolution compared to the target one, such a model can be used to increase the time resolution of the target signals in a much more intelligent way compared to the conventional uni-modal interpolations. As a use case, we target TS, one of the most important diagnostics that measure the density and pressure profile of plasma. However as mentioned earlier, its low temporal resolution is a bottleneck in studying the plasma evolution in the rapidly changing events such as ELM.

We consider a suite of input diagnostics available at DIII-D including Interferometer, ECE, Magnetic probes (Magnetics), Charge Exchange Recombination (CER), and MSE with typical sampling rates of $1.66\text{\,}\mathrm{MHz}$, $500\text{\,}\mathrm{kHz}$, $2\text{\,}\mathrm{MHz}$, $200\text{\,}\mathrm{Hz}$, and $4\text{\,}\mathrm{kHz}$, respectively. To obtain a dataset suitable for this task, all the included diagnostics are aligned with the TS sampling time steps by matching their most recent measured sample. In this way, we create a dataset with which we train the Diag2Diag ANN for this task. Since the sampling steps of TS are not always uniform in time (See Figure 6), we opted for a memory-less neural network instead of the recurrent neural network commonly used in time-series analysis. However, we included the first and second derivatives of the high-resolution input diagnostics, ECE and Interferometer, to include the temporal evolution information.

Figure 3: (a) Comparison of the electron density by the measured TS and the synthetic SRTS, for the DIII-D shot 153761 [15] near the edge ($Z=0.71m$). $D_{\alpha}$ with arbitrary units is plotted as an indicator of ELM s. An example of ELM event captured by both diagnostics, and another example only captured by SRTS are highlighted in green and purple, respectively. (b-c) Aggregating the measured TS density and temperature in three locations of plasma near the edge for several ELM cycles of the DIII-D shot 174832. The circle highlights the measures TS for one selected ELM cycle and the solid lines present the SRTS which agreeably match the measures TS. $t=0$ represents the time when ELM is identified by $D_{\alpha}$. (d-e) The evolution of SRTS between two consecutive measured TS near one ELM cycle across the plasma location.

Figure 3(a) shows, in blue, synthetic TS signals (or SRTS) reconstructed through the trained Diag2Diag from other high-frequency diagnostics, where the original TS signals are also shown with black dots. We can observe that the synthetic signals closely follow the original signals. Furthermore, while the original signals sometimes fail to capture ELM events (identifiable by spectral emission ($D_{\alpha}$)), the synthetic signals accurately capture the events missed between the original signals.

Validation: Investigating ELM cycles in DIII-D

When ELM instability occurs, a large amount of plasma quickly escapes from the boundary within milliseconds, and then the plasma gradually recovers. TS diagnostics can observe the density and temperature structure at this edge region, but are limited in capturing dynamics occurring over milliseconds. Recent research [23] overcame these resolution limits by aggregating the measurements from multiple repeated cycles of the fast activity under almost identical conditions to observe a complete evolution. In this subsection, over 20 highly reproduced cycles of ELM crash and recovery were aggregated from DIII-D shot 174823 to assume a ground truth of a complete evolution of an ELM cycle.

The aggregated density and temperature evolution measured by TS in three locations of plasma near the edge are shown in Figures 3(b-c) with transparent crosses, while measurements from a single cycle are shown as solid dots, with different colors for different measurement locations.

We used the Diag2Diag model to generate synthetic SRTS, shown with solid lines in Figure 3(b-c). The SRTS signal from a single cycle around time $3795\text{\,}\mathrm{m}\mathrm{s}$ not only follows the trend of the aggregated multiple TS measurements but also well overlies the TS measurements within that cycle. Figures 3(d-e) show the detailed evolution of plasma density and temperature across the plasma plasma location captured by SRTS in the same ELM cycle at $3795\text{\,}\mathrm{m}\mathrm{s}$ which is missed by TS between its two consecutive measurements at $3791\text{\,}\mathrm{m}\mathrm{s}$ and $3800\text{\,}\mathrm{m}\mathrm{s}$.

In a more typical tokamak discharge, the plasma state continually changes, and ELM s occur more irregularly as shown in Figure 3(a). In such cases, it is not possible to reconstruct a single ELM cycle by aggregating multiple cycles, and our SRTS method will be highly beneficial.

4 Science discovery: Unveiling diagnostic evidence of RMP mechanism on the plasma boundary

In what follows, we investigate whether the synthetic super-resolution diagnostics can help to verify the hypotheses on the mechanism of plasma response to external field perturbations in fusion plasma physics that have been proposed theoretically or by simulations but have never been visualized with the experimental data due to the lack of diagnostic resolution.

One promising strategy to control ELM s is employing Resonant Magnetic Perturbation (RMP) s [24, 25, 26, 27, 28] generated by external 3D field coils depicted in Figure 4(a). These fields effectively reduce the temperature and density at the confinement pedestal, stabilizing the energy bursts in the edge region. Consequently, ITER will rely on RMP s to maintain a burst-free burning plasma in a tokamak, making it essential for the fusion community to understand and predict the its physics mechanism[29]. However, this issue has been remained a challenge for decades.

The leading theory [30, 31, 32, 33] for explaining the reduced pedestal by RMP s is the formation of magnetic islands by an external 3D field. The magnetic island is a ubiquitous feature in an electromagnetic system with plasmas [34] formed by field reconnection [35, 36]. This structure allows rapid heat (or temperature) and particle (or density) transport between adjacent magnetic field lines, strongly reducing the gradient of local heat and particle distribution or, in other words, profile flattening[37]. The existing theories explain that RMP forms static magnetic islands at the pedestal top and foot region, therefore reducing the pedestal by local profile flattening. As illustrated in Figure 4(a), the theory predicts that RMP s can create magnetic islands near the plasma boundary where pedestal sit in. This model has been successful in quantitatively explaining and predicting the RMP-induced pedestal degradation in real experiments [38, 39], reinforcing magnetic islands as a promising mechanism for RMP-induced pedestal degradation. Nevertheless, measuring the direct evidence of island or local profile flattening still remains a challenge. Extensive experimental efforts have been conducted for this reason and were able to capture the local flattening electron temperature profile [40] near the pedestal top, strongly supporting this theory. However, simultaneously measuring electron temperature and density both at the pedestal top and foot was not possible. In a previous study, rough evidence was observed in TS [33], but it was insufficient to derive a concrete conclusion, mainly due to a large uncertainty of measurement originating from narrow structure (expected from theory, see Fig.4(a)) and oscillatory nature of the plasma boundary. To address the diagnostic uncertainties caused by such system oscillation, one method is to increase the time sampling rate and use time averaging. However, in conventional TS, increasing the time resolution results in a trade-off with measured accuracy, eventually leading to observational limitations.

Figure 4: Structure of 3D coils and islands by perturbed field (a), and the evidence in the simulation (b-d) and the SRTS diagnostic (e-g) for RMP-induced island mechanism on the plasma boundary in DIII-D shot 157545. Here, the Chebyshev filter is used to derive a statistically reliable time trace of the profile, leveraging the enhanced temporal resolution.

Interestingly, the SRTS has once again illuminated the profile evolution by RMP application, providing the first evidence of ”simultaneous” flattening of temperature and density profile at both top and the foot of the pedestal, strongly supporting the theoretical prediction of magnetic islands effect. This is possible by capturing the statically reliable time trace of profile with Chebyshev time filter, leveraging the enhanced temporal resolution by SRTS.

Figures 4(b-g) illustrate the recovery of temperature and density pedestals within $10\text{\,}\mathrm{ms}$ after deactivating RMP, as captured through numerical modeling (Figure 4(b-d)) and SRTS (Figure 4(e-g)). The simulations reveal that the recovery of temperature and density pedestals begins at the top and foot, coinciding with the disappearance of islands. As depicted in Figures 4(d) and (g), the profile gradient recovers at these island locations, enhancing the overall profile. For instance, the measured temperature pedestal shows recovery at both the top and foot through an increasing gradient, displaying qualitative alignment with the simulation results. However, some discrepancies are noted, particularly in the density evolution at the pedestal foot in the SRTS, even though its gradient remains consistent with the modeling. These quantitative differences may stem from the TS’s limited spatial resolution at the boundary and the modeling assumptions such as fixed boundary conditions [41]. Nevertheless, the gradient evolution directly indicates a change in transport due to the RMP-induced islands during this perturbative profile evolution, highlighting that the SRTS successfully reveals the experimental island effect. This provides the ‘first‘ diagnostic evidence of profile flattening at magnetic islands, a key mechanism of RMP-induced pedestal degradation. This successful application of SRTS underscores its potential to reveal new physics beyond the limitations of conventional diagnostic techniques.

5 Conclusion

This study introduces a transformative approach in the field of signal processing and diagnostics through the development of a multimodal neural network, Diag2Diag, which significantly enhances temporal resolution. By leveraging the intrinsic correlations among various diagnostic measurements, we have demonstrated the potential to increase the temporal resolution of the Thomson Scattering diagnostics in fusion plasma from a standard $0.2\text{\,}\mathrm{kHz}$ to an unprecedented $1\text{\,}\mathrm{MHz}$. This improvement has unlocked new potentials in analyzing fast transient phenomena in plasma, such as the ELM s and the effects of RMP s on pedestal degradation, which were previously blurred or missed in lower resolution data. The ability to inspect these dynamics in greater detail provides new insights into plasma behavior, particularly in conditions where key physics is hidden in the milliseconds. This enhancement is not merely a technical improvement but a crucial enabler for deeper insights into plasma behaviors that are pivotal for advancing fusion reactors. Furthermore, the model’s ability to reconstruct and predict diagnostics from multimodal inputs opens new avenues for cost-effective and less hardware-dependent diagnostic systems. This is particularly beneficial for experimental setups where space and resources are limited, such as in smaller fusion test facilities or in environments where installing multiple high-resolution diagnostics is impractical.

The implications of this work extend well beyond the immediate application to magnetic fusion. The super-resolution capabilities developed here can significantly impact areas such as laser fusion data analysis, accelerator data analysis, and molecular dynamics research. In these fields, similar challenges exist where the time resolution of diagnostics is inadequate to capture fast phenomena effectively. By applying our method, researchers can potentially uncover new physical phenomena or confirm theoretical predictions that were previously unverifiable through experiments due to resolution constraints.

In conclusion, the Diag2Diag model not only addresses a critical need within the fusion community but also sets a precedent for the broader application of AI and machine learning in physical sciences. By pushing the boundaries of what can be observed and measured, this work contributes to the foundational technologies necessary for the realization of fusion energy and advances our understanding of complex physical systems across various scientific domains.

6 Methods

6.1 Underlying Physics in Coupling of Diagnostic Measurements in Plasma System

Diagnostics of electromagnetic systems involve measuring photons or waves to determine the physical quantities of these systems through post-processing. Due to the nature of the systems, these diagnostics are connected. Firstly, the measured signals are interconnected through electromagnetic interactions during system events. Additionally, the physical quantities obtained from signal processing are closely linked through momentum balances. Electromagnetic plasma quantities are governed by a series of momentum equations that encompass variables such as density, flow, temperature, and higher-order terms. Figure 5 illustrates the momentum equations for plasma density ($n$) and temperature ($T$), where $D$ represents particle diffusivity, $v$ is plasma flow, $S_{n}$ is the particle source, $q$ denotes heat flux, $B$ stands for magnetic field, $j$ is plasma current, $S_{T}$ is the heat source, and ($\alpha$, $\beta$) are constant coefficients determined by plasma properties [42]. These equations demonstrate how the measured plasma quantities are interrelated both spatially and temporally. For instance, the line-averaged density obtained from Interferometer diagnostics is geometrically linked to the local density measured by TS by its definition. Simultaneously, temperatures measured by TS and ECE diagnostics, which are positioned differently, are spatially coupled through the gradient term in the momentum equations. Although the TS density and temperature do not directly interact in the equations, they are tightly linked via diffusive fluxes influenced by turbulence, flow, and sources in a self-consistent manner. This intricate physical coupling of various diagnostic measurements allows ML to identify and predict their interconnections effectively.

Figure 5: Schematic of couplings between diagnostics of the plasma system. They are connected through electromagnetic interactions between signals. Simultaneously, derived quantities from these signals are coupled via geometric definitions, momentum balances, and high-order physics, including turbulence, flow, and source in the system.

6.2 Related works

In recent years different kinds of ANN have been used for upsampling visual data [43, 44, 45, 46, 47] and for radar data [48, 49, 50, 51]. These approaches are typically some kind of non-linear interpolation to add frames between existing video frames. More examples for ML-based upsampling were proposed for medical data [52] and for audio data [53, 54, 55, 56]. Similar to the video upsampling approaches, these approaches can be considered a subcategory of non-linear interpolation as well. In [57], an alternative to interpolation is suggested to estimate missing data in temporal data streams. It is to some extent a multimodal approach, because it fuses different kinds of information. However, the algorithm is limited towards estimating missing data or dealing with irregularly sampled data. Approaches like these work well for enhancing existing sequences, which are quasi-stationary in a way such that consecutive frames or samples do not change very fast.

However, in nuclear fusion, many spurious events like ELM can happen between two TS samples. By interpolating between consecutive TS samples, regardless of linearly or non-linearly, it is likely that we would miss such spurious events. In our work, we thus develop a novel method to generate additional TS samples based on other diagnostics. This is roughly inspired by other multimodal ML approaches, such as [58], where it was proposed to fuse Radar and camera data for an enhanced distance estimation. This is a multimodal approach and thus related to our approach, or [59], where machine learning was used to reveal the control mechanics of an insect wing hinge. This was also a multimodal approach in a way that the ML algorithm received different features recorded from flying insects. However, similar to the other approaches, no attempts to upsampling or estimating missing/in-between data are made. Also [60] presents an artificial neural network method to enhance historical electron temperature data from the decommissioned C-2U fusion device. The model significantly increases the effective sampling rate of TS temperature measurements, utilizing data from multiple diagnostics including the measured TS. The method’s effectiveness is demonstrated through comparisons with ensemble-averaged data for micro-burst instability study. The model’s main drawbacks include limited generalization to only temperature profile study for one specific plasma regime. Notably, the work does not explore the model’s potential for discovering new physics in fusion plasmas.

6.3 Diagnostic details for ELM

In order to let fusion energy be a viable energy source, it must achieve significant fusion gain through continuous fusion reactions. A prominent method to reach this objective is operating a tokamak in high-confinement mode (H-mode), which has a narrow edge transport barrier, also known as the pedestal. This feature significantly boosts plasma confinement within the reactor, enhancing fusion power and efficiency. However, operating in H-mode introduces a steep pressure gradient at the pedestal, leading to substantial operational risks. This gradient drives hazardous edge energy bursts due to a plasma instability known as ELM s. These bursts lead to sudden drops in the energy at the pedestal, causing severe, transient heat fluxes on the reactor walls. This results in damaging material, potential surface erosion and melting, with heat energy reaching approximately $20\text{\,}\mathrm{MJ}\text{\,}{\mathrm{m}}^{-2}$, which is an unacceptable level for fusion reactors. From ITER, future machines will not allow even the first ELM. Therefore, to advance tokamak designs toward practical application in fusion energy, it is crucial to develop dependable methods to consistently suppress these edge burst events.

A limitation of some diagnostics, such as TS is the low temporal resolution of only $200\text{\,}\mathrm{Hz}$, which does not allow for detecting and tracking fast events like ELM ($\leq$1\text{\,}\mathrm{ms}$$). Figure 6 shows an example of missing ELM in a discharge, due to the low temporal resolution of TS. Nevertheless, it is still important to detect such events reliably, as they can have a strong impact on plasma behavior.

Figure 6: (a) Configuration of some diagnostics at DIII-D . (b) Example of TS signal for DIII-D discharge 190736, in red, along with the $D_{\alpha}$ measurements, as an indicator of ELM, and a collection of other diagnostics that is used to increase the resolution of TS. (c) The same data as (b) but zoomed in to show the sampling points of the diagnostics around two examples of ELM event. Due to the low sampling rate, this ELM is not observed by TS. However, thanks to their high temporal resolution, diagnostics including ECE, Interferometer, and Magnetics capture that.

On the other hand, diagnostics like Interferometer, ECE, have much higher temporal resolution with sampling frequencies around $\mathrm{MHz}$, which allows for a much more detailed analysis of the plasma. However, these diagnostics have different characteristics compared to TS. While TS offers detailed insights into both electron density and temperature with high accuracy, it requires complex setups and is usually more resource-intensive. A Interferometer provides a more straightforward approach to measuring electron density, excelling in situations that require rapid response and continuous monitoring. Furthermore, ECE and TS are both pivotal diagnostic tools used in tokamaks for measuring electron temperature, yet they operate on distinctly different principles and offer unique advantages. ECE utilizes the natural microwave emissions from electrons gyrating around magnetic field lines to provide excellent temporal resolution, allowing for the monitoring of rapid plasma changes and instabilities, though its effectiveness can be limited by variations in magnetic field strength. On the other hand, TS involves firing a laser into the plasma and analyzing the scattered light, which provides robust, absolute measurements of both electron temperature and density with less susceptibility to magnetic influences. While ECE excels in continuous data collection and fine temporal analysis, TS offers superior spatial resolution and is less dependent on external conditions, making it invaluable for comprehensive, though typically less frequent, plasma evaluations. If it would be possible to find a correlation between the measurements of those high-resolution diagnostics and TS, this would be useful for develo** new physical analyses.

6.4 Data acquisition

For this experiment, we used discharges from the DIII-D tokamak that include all data from the key diagnostics of interest (CER, Interferometer, ECE, MSE, and TS). We randomly selected $4000$ discharges recorded between the years $2017$ and $2022$ to ensure a diverse and representative dataset. The diagnostic data was collected using the DIII-D MDSplus [61] and PTDATA [62] systems. These diagnostics are generally provided as time-series data streams with varying sampling frequencies, ranging from $200\text{\,}\mathrm{Hz}$ for TS up to $1.66\text{\,}\mathrm{MHz}$ for Interferometer. The specific pre-processing steps applied to the data for the different experiments conducted in this study are detailed in the following sections.

6.5 Feature extraction

For the spectrogram experiments, we consider the Interferometer and ECE diagnostics. We compute logarithmic magnitude spectrogram from time-series of the raw diagnostics. For each channel ($40$ ECE channels and $4$ Interferometer channels), we therefore used hamming windows of $1\text{\,}\mathrm{ms}$ with $0.5\text{\,}\mathrm{ms}$ overlap. In this way, it was ensured that the different magnitude spectrogrames are aligned in time. The spectrograms were afterwards converted to a logarithmic scale, clipped and rescaled to the range of [0, 1]. Given the noisy nature of the ECE signals and after rescaling the spectrograms to the range of [0,1], the spectrograms are enhanced using a pipeline of image processing filters that includes

• Quantile Filtering with a threshold of 0.9,

• Gaussian Blur Filtering on patches of size 31x3,

• Subtracting average per frequency bin

We used the ECE spectrograms as inputs to our model. Since we treated every ECE channel independently during feature extraction, we obtained one spectrogram per channel, resulting in $40$ input spectrograms (one per ECE channel). Since our model is designed to estimate the Interferometer spectrograms, it predicts four output spectrogram channels corresponding to the four Interferometer interferometer channels.

For the time-series models, the different diagnostic measurements have varying sampling rates, and some are even non-uniformly sampled in time. Since the aim of time-series data analysis was to increase the resolution of TS, we used its timestamps as a reference and aligned all diagnostic modalities to TS by matching their most recent measured samples in time. This resulted in an amount of $135\,233$ training, $22\,084$ validation, and $18\,721$ test samples.

For Interferometer and ECE, we also included the first and second temporal derivatives. Therefore, we smoothed the signals with a moving average window of $1\text{\,}\mathrm{ms}$ ($1660$ Interferometer samples and $500$ ECE samples), and then computed the first and second temporal derivatives of the smoothed signal also with a window of $1\text{\,}\mathrm{ms}$. In this way, we can consider a temporal context of $4\text{\,}\mathrm{ms}$.

The diagnostics CER and MSE have a low temporal resolution, i.e., sampling frequencies of $200\text{\,}\mathrm{Hz}$ and $4\text{\,}\mathrm{kHz}$, respectively. In this paper, we assume that they evolve only slowly in time. For the upsampling experiments, we thus pad these diagnostics after a measured sample with constant values until the next measured sample arrived.

The diagnostics (CER, Interferometer, ECE, and MSE) together with the derivatives of Interferometer ($4$ channels $\rightarrow$ $12$ dimensions including derivatives) and ECE ($42$ channels $\rightarrow$ $126$ dimensions) lead to an input size of $192$. From there, we map to TS with $288$ dimensions for plasma density and temperature.

6.6 Spectrogram model development

The resulting multi-channel ECE spectrograms were used as the input to a CNN, and the multi-channel Interferometer spectrograms were used as the target outputs. We optimized all important hyper-parameters based on the $\mathcal{L}1$ loss to minimize the difference between the ground truth and the estimated outputs on the validation set.

The optimization process of the model involved several key steps:

• The model underwent training for up to $500$ epochs.

• We implemented early stop** with a patience threshold of $20$ epochs, during which we monitored the validation loss for any improvements.

• The AdamW optimizer [63], known for decoupling weight decay from the learning rate, was utilized to minimize the $\mathcal{L}1$ loss function.

• We conducted a comprehensive hyper-parameter optimization through a randomized search across $1000$ iterations for all hyper-parameters listed in Table 1.

The exact search space of the hyper-parameters and their optimized values obtained from the randomized search are summarized in Table 1.

Table 1: Optimized hyperparameters for the spectrogram prediction CNN model.

Hyper-parameter	Search space	Optimized value
Batch size	$18$, random integers	$2$
Kernel size	$315$ odd integers	$7$
Learning rate	$1\text{\times}{10}^{-5}1$ log uniform	$0.482\text{\times}{10}^{-3}$
Final $\mathcal{L}1$ loss	–	$1.2\text{\times}{10}^{-3}$

To reduce the amount of training time, we randomly selected $518$ discharges from the entire dataset to conduct the hyperparameter optimization. The model with the best performing hyperparameter setting (achieving an $\mathcal{L}1$ loss of $1.2\text{\times}{10}^{-3}$ on the validation set) was then re-trained on all available discharges.

The best-performing model is a CNN that transforms the ECE spectrograms with $40$ channels subsequently to $32168$ feature maps and finally to the Interferometer spectrograms with $4$ channels. For each feature map, 2D filter kernels with a size of $7\times 7$ are used. Batch normalization was used separately for each channel, and parametric $\mathrm{ReLU}$ activation functions were used after each batch normalization layer. The model had in total $95\,823$ trainable parameters (i.e., filter kernels for each feature map, batch normalization parameters, and negative slope of the parametric $\mathrm{ReLU}$ activation function).

6.7 Time-series model development

For the time-series prediction task, we employed a Multilayer Perceptron (MLP) model. The input data to the MLP comprised the CER, Interferometer, ECE, MSE, and magnetic diagnostics, along with the first and second temporal derivatives of the Interferometer and ECE signals, resulting in a total input size of $236$ dimensions. The target output was the TS diagnostic data, which had $80$ dimensions representing electron temperature and density across various spatial locations. The target data were augmented by factor $2$ by using the upper and lower intervals of each sample as additional targets.

The MLP model was trained for a maximum of $500$ epochs, with an early stop** mechanism implemented to halt the training process if the validation loss did not improve for $20$ consecutive epochs. The AdamW optimizer [63] was employed to minimize the $\mathcal{L}1$ loss function during training.

As for the spectrogram model, a comprehensive hyperparameter optimization was undertaken using a randomized search approach spanning $2000$ iterations. The hyper-parameters jointly optimized included the batch size, hidden layer size, dropout rate, and learning rate.

Table 2 summarizes the optimized hyperparameter values obtained from the randomized search process.

Table 2: Optimized hyperparameters for the time-series MLP model.

Hyper-parameter	Search space	Optimized value
Batch size	$12048$, powers of 2	$1024$
Hidden layer size	$1922048$ integers	$952$
Dropout	$01$ uniform	$0.076$
Learning rate	$1\text{\times}{10}^{-5}1$ log uniform	$1.998\text{\times}{10}^{-3}$
Final $R^{2}$ score	–	$0.92$

6.8 Uncertainty quantification

To estimate the uncertainty in super-resolution TS, we employed a Bayesian Neural Network (BNN) with the architecture described in Section 6.7. For each TS channel, we calculated the standard deviation of the BNN outputs, particularly focusing on the pedestal area.

Figure 7 illustrates the average standard deviations of the neural network outputs per channel over the validation set of discharges, depicted as red error bars. The TS channels are represented by their relative distance from the core of the plasma ($\psi_{n}=0$) to the edge ($\psi_{n}=1$). For comparison, we also show the empirically measured diagnostic errors, which include contributions from background light, dark noise, and pulse error.

Although the sources of diagnostic uncertainty differ from those of model uncertainty, our results indicate that the model uncertainty falls in the range of the diagnostic errors that are generally accepted by physicists. This suggests that proposed model provides a reliable estimate of uncertainty that enhances the confidence in the super-resolution TS measurements.

Figure 7: Comparison of the SRTS and the measured TS diagnostic uncertainty for both electron density $n_{e}$ and temperature $T_{e}$. The dashed line shows the average measured TS.

6.9 Thomson Scattering in “bunch mode”

Thomson Scattering, as a popular and reliable diagnostic technique, has successfully measured electron temperatures and electron number densities of plasmas for many years. However, conventional TS techniques operate only at tens of hertz. To accurately resolve the fast transient dynamics, the Thomson scattering lasers can be fired in a bunch mode, which enabled temporal resolution of up to $10\text{\,}\mathrm{\SIUnitSymbolMicro s}$. This increase in temporal resolution is achieved by using multiple lasers in the same path with pulses interleaved closely in time. Normally, the lasers are phased to produce pulses at fairly regular intervals (exact regularity is not possible with the specific combination of $20\text{\,}\mathrm{Hz}$ and $50\text{\,}\mathrm{Hz}$ lasers being used at DIII-D ). In bunch mode, the phase shifts are adjusted so that all lasers fire in rapid succession, followed by a cool down. This bunch mode encompasses between 3 and 7 laser pulses depending on the time in the discharges.

Figure 8(a) presents the comparison of SRTS with the measured TS fired at bunch mode for the DIII-D shot 153761. Figure 8(b-e) are zoomed in of different window times. Figure 8 (f-g) are further zoomed in frames up to one TS fire of the TS lasers. The matches between the measured TS and SRTS confirms the reliability of SRTS. Also we observe the SRTS superiority in capturing ELM s in comparison to the measured TS even in “bunch mode”.

Figure 8: Comparison of the SRTS and the TS fired in “bunch mode” for measuring the electron density in DIII-D shot 153761 at the pedestal (Z=0.71m).

6.10 Research method

To avoid any bias during model development and evaluation, each of the following steps in this research was conducted independently by separate researchers in a feed-forward manner as presented in Figure 9:

Figure 9: Research steps

1. The data scientists developed diagnostic dataset for training the neural network aiming for generating synthetic super resolution Thomson Scattering (SRTS). In this phase, the evaluation metric was simply the similarity between the model’s output and the measured TS, whenever the measurement was available.

2. For physics validation we generated the super resolution diagnostic for a known ELMy discharge and asked an ELM-expert physicist to validate the behavior of the super resolution diagnostic.

3. The data scientist then delivered the generated super-resolution diagnostic for the target plasma discharge to an experimental physicist to extract the plasma profile from that.

4. We then asked another physicist with expertise in simulation to obtain the simulation results for the target plasma discharge.

5. In the final phase, we compared the plasma profiles extracted from our generated diagnostics and the simulation results. They matched nicely!

This indicates that our results are not biased based on prior physics knowledge, and we also did not rework our ML model to match our results with the simulation.

Acknowledgements

This material is based upon work supported by the U.S. Department of Energy, Office of Science, Office of Fusion Energy Sciences, using the National Fusion Facility, a DOE Office of Science user facility, under Award DE-FC02-04ER54698. In addition this research was supported by the U.S. Department of Energy, under Awards DE-SC0024527, DE-SC0015480, DE-SC0022270 and DE-SC0022272, as well as the National Research Foundation of Korea (NRF) Award RS-2024-00346024 funded by the Korea government (MSIT).

Disclaimer: This report was prepared as an account of work sponsored by an agency of the United States Government. Neither the United States Government nor any agency thereof, nor any of their employees, makes any warranty, express or implied, or assumes any legal liability or responsibility for the accuracy, completeness, or usefulness of any information, apparatus, product, or process disclosed, or represents that its use would not infringe privately owned rights. Reference herein to any specific commercial product, process, or service by trade name, trademark, manufacturer, or otherwise does not necessarily constitute or imply its endorsement, recommendation, or favoring by the United States Government or any agency thereof. The views and opinions of authors expressed herein do not necessarily state or reflect those of the United States Government or any agency thereof.

Author contributions

A.J. is the main author of the manuscript and contributed to develo** the multimodal model and data science analyses. S.K. contributed to the physics analysis of Diag2Diag for the RMP mechanism on the plasma boundary. J.S contributed to the general physics analysis and writing the manuscript. Q.H. contributed to the physics simulation of the RMP mechanism on the plasma boundary. M.C. and P.S. contributed to the DIII-D data collection and data preprocessing for the multimodal model development and writing the manuscript. A.O.N contributed the physics analysis of Diag2Diag for ELM cycles. Y.S.N and E.K. contributed to the conception of this work, analyses, and writing the manuscript.

Competing interests

The authors declare no competing interests.

Data availability

The data that support the findings of this study are available from the corresponding author upon reasonable request. The source code for the models developed in this work can be found in [64].