QuaLiKiz Neural Network regression
A neural network regression allows to emulate the results of first-principle-based turbulent transport predictions. Model output databases are compiled with inputs spanning a portion of the experimentally relevant parameter space. The neural network then learns the mapping from code inputs to outputs. The speed of the quasilinear models allows for the production of sufficiently large training sets for this application.
QuaLiKiz was used to create a large database of 3.10^8 flux calculations using 1.3 MCPUh on HPC resources (Edison@NERSC). Embedding known physical constraints in the training of the networks is essential for the surrogate model to perform well in transport predictions. As such, we show the importance of choosing the right cost function and more fundamentally, choosing which target variables the networks have to be trained on. Custom figures of merit and visualisation tools were developed to aid with neural network accuracy verification, such as the data slicer which you can use to visualize QuaLiKiz results over the 9D database. Ongoing work is underway for development of higher input dimension (13D) QuaLiKiz-neural-networks, where due to the curse of dimensionality, the relevant input subspace is constrained by an under-construction multi-machine profile database.
There are multiple flavours of QLKNN to try:
- "baseline": general for large-aspect ratio tokamaks: QLKNN-hyper
- "WIP advanced physics": general for large-aspect ratio tokamaks: QLKNN-HornNet
- "WIP jet specific": better QuaLiKiz approximation for JET only: QLKNN-jetexp
- "Deprecated no physics": Trained without QLKNN methodology, for reference only: QLKNN-fullflux
Journal publications
- NEW Fast modeling of turbulent transport in fusion plasmas using neural networks: van de Plassche PoP 2020
- Real-time-capable prediction of temperature and density profiles in a tokamak using RAPTOR and a first-principle-based transport model: Felici NF 2018
- 5D proof of principle demonstration: Citrin NF letter 2015
Presentations
- NEW Fast surrogate modelling of turbulent transport in fusion plasmas with physics-informed neural networks AAPPS invited talk pdf
- Towards a high input dimension database for Neural Network regression based on JET measurements: Ho poster EPS2017
- 9D database and 10D Neural Network regression including ExB shear in post-processing: van de Plassche poster TTF 2019