We present PEdestal Neural Network (PENN) as a machine learning model for tokamak pedestal predictions. Here, the model is trained using the EUROfusion JET pedestal database to predict the electron pedestal temperature and density from a set of global engineering and plasma parameters. Results show that PENN makes accurate predictions on the test set of the database, with R2 = 0.93 for the temperature, and R2 = 0.91 for the density. To demonstrate the applicability of the model, PENN is employed in the European transport simulator (ETS) to provide boundary conditions for the core of the plasma. In a case example in the ETS with varied neutral beam injection (NBI) power, results show that the model is consistent with previous studies regarding NBI power dependency on the pedestal. Additionally, we show how an uncertainty estimation method can be used to interpret the reliability of the predictions. Future work includes further analysis of how pedestal models, such as PENN, or other advanced deep learning models, can be more efficiently implemented in integrating modeling frameworks, and also how similar models may be generalized with respect to other tokamaks and future device scenarios.
Investigating pedestal dependencies at JET using an interpretable neural network architecture