skip to main content
OSTI.GOV title logo U.S. Department of Energy
Office of Scientific and Technical Information

Title: Fusion RF Modeling Machine Learning (FusionML_RF) v1.0

Software ·
DOI:https://doi.org/10.11578/dc.20230713.12· OSTI ID:1989436 · Code ID:110025
 [1];  [1];  [1];  [1];  [2];  [2];  [3];  [3]
  1. Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States)
  2. Massachusetts Inst. of Technology (MIT), Cambridge, MA (United States)
  3. Princeton Univ., NJ (United States)

FusionML_RF consists of multiple codes and trained machine learning (ML) models that perform low-cost output modeling from the Genray-CQL3D. Three machine learning techniques (multilayer perceptron, random forest, and Gaussian process) provide fast surrogate models for lower hybrid current drive (LHCD) simulations. For example, completing a single GENRAY/CQL3D simulation without radial diffusion of fast electrons requires several minutes of wall-clock time. On the other hand, these ML models achieve ~ms of inference time with high accuracy across the input parameter space. This software collection consists of multiple components. (1) codes that use ML methods and precomputed Genray-CQL3D simulation output to build regression models that enable approximate computations of Genray-CLQ3D outputs from arbitrary but physically meaningful input parameters (surrogate modeling); (2) three trained models created by the team, using a database of 16,000+ GENRAY/CQL3D simulations, to study the performance of ML models for surrogate modeling; (3) codes that load the trained models and simulation data, and then compute mean squared error between the models' predictions and the ground truth of simulation output data. This collection is being made available in conjunction with a scientific publication about the work to promote reusability and provide an artifact of the scientific work.

Short Name / Acronym:
FusionML_RF v1.0
Project Type:
Open Source, Publicly Available Repository
Site Accession Number:
2022-082
Software Type:
Scientific
License(s):
BSD 3-clause "New" or "Revised" License
Research Organization:
Princeton Univ., NJ (United States); Massachusetts Inst. of Technology (MIT), Cambridge, MA (United States); Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States)
Sponsoring Organization:
USDOE

Primary Award/Contract Number:
AC02-05CH11231
DOE Contract Number:
AC02-05CH11231
Code ID:
110025
OSTI ID:
1989436
Country of Origin:
United States