Predicting Resistive Wall Mode Stability in NSTX through Balanced Random Forests and Counterfactual Explanations

Piccione, Andrea ; Sabbagh, Steven; Andreopoulos, Yiannis
Issue date: 2021
Rights:
Creative Commons Attribution 4.0 International (CC BY)
Cite as:
Piccione, Andrea, Sabbagh, Steven, & Andreopoulos, Yiannis. (2021). Predicting Resistive Wall Mode Stability in NSTX through Balanced Random Forests and Counterfactual Explanations [Data set]. Princeton Plasma Physics Laboratory, Princeton University. https://doi.org/10.11578/1888266
@electronic{piccione_andrea_2021,
  author      = {Piccione, Andrea and
                Sabbagh, Steven and
                Andreopoulos, Yiannis},
  title       = {{Predicting Resistive Wall Mode Stability
                 in NSTX through Balanced Random Forests
                 and Counterfactual Explanations}},
  publisher   = {{Princeton Plasma Physics Laboratory, Pri
                nceton University}},
  year        = 2021,
  url         = {https://doi.org/10.11578/1888266}
}
Description:

Recent progress in the disruption event characterization and forecasting framework has shown that machine learning guided by physics theory can be easily implemented as a supporting tool for fast computations of ideal stability properties of spherical tokamak plasmas. In order to extend that idea, a customized random forest (RF) classifier that takes into account imbalances in the training data is hereby employed to predict resistive wall mode (RWM) stability for a set of high beta discharges from the NSTX spherical tokamak. More specifically, with this approach each tree in the forest is trained on samples that are balanced via a user-defined over/under-sampler. The proposed approach outperforms classical cost-sensitive methods for the problem at hand, in particular when used in conjunction with a random under-sampler, while also resulting in a threefold reduction in the training time. In order to further understand the model's decisions, a diverse set of counterfactual explanations based on determinantal point processes (DPP) is generated and evaluated. Via the use of DPP, the underlying RF model infers that the presence of hypothetical magnetohydrodynamic activity would have prevented the RWM from concurrently going unstable, which is a counterfactual that is indeed expected by prior physics knowledge. Given that this result emerges from the data-driven RF classifier and the use of counterfactuals without hand-crafted embedding of prior physics intuition, it motivates the usage of counterfactuals to simulate real-time control by generating the βN levels that would have kept the RWM stable for a set of unstable discharges.

Show More
# Filename Filesize
1 readme.txt 3.17 KB
2 Piccione_NF105028.zip 37.1 MB