Scenario trajectory planning for NSTX-U using machine-learning-accelerated models and genetic optimization*

Between-shots and real-time actuator trajectory planning will be critical to achieving reliable high performance scenarios in present-day tokamaks, ITER, and beyond. A requirement for this effort is the availability of models that are accurate enough for useful decision making and fast enough for optimization algorithms to meet between-shots and real-time deadlines. While integrated modeling codes are progressing toward the accuracy and completeness needed for these applications, they are too computationally intensive for this purpose. To address this, a novel accelerated simulation capability has been developed for NSTX-U by applying machine learning techniques to both empirical data and TRANSP simulations, enabling profile and equilibrium predictions at real-time relevant time scales. The approach includes machine learning surrogates for high-fidelity TRANSP modules that accelerate calculations by orders of magnitude while maintaining fidelity. For quantities not accurately modeled by TRANSP modules, machine learning is applied to an experimental database to create empirical models. Initial results trajectory optimization using the learning-based model are presented, including a strategy for mitigating the effect of leaving the validity range of the model.