Real-Time Plasma State Prediction via FPGA-Accelerated Quantized Recurrent Probabilistic Neural Networks
Not on the map. Why?
Abstract (arXiv)
Real time plasma state estimation for control of Tokamak devices are challenging due to the stringent latency requirements of the plasma control system (PCS). We present an end-to-end workflow for deploying a recurrent probabilistic neural network (RPNN) on FPGA hardware. We combine architecture size reduction with quantization-aware training via QKeras. The model is then synthesized using hls4ml, targeting a Xilinx Alveo U50 device. We report a design that fits comfortably within all four resource budgets (DSP, LUT, FF, BRAM) at deterministic sub-10~μs single-timestep latency, meeting the requirements for real-time inference inside a model-predictive-control-style plasma control loop.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 15 references that are in the map. First posted on arXiv 19 Sep 2026.
Builds on
11 of 15 references.
References · 11
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 774
- 2018Fast inference of deep neural networks in FPGAs for particle physicscited by 133
- 2020Automatic heterogeneous quantization of deep neural networks for low-latency inference on the edge for particle detectorscited by 80
- 2021Uncertainty Toolbox: an Open-Source Library for Assessing, Visualizing, and Improving Uncertainty Quantificationcited by 57
- 2021Applications and Techniques for Fast Machine Learning in Sciencecited by 32
- 2022Ultra-low latency recurrent neural network inference on FPGAs for physics applications with hls4mlcited by 14
- 2024Full Shot Predictions for the DIII-D Tokamak via Deep Recurrent Networkscited by 9
- 2026Offline Reinforcement Learning for Rotation Profile Control in Tokamakscited by 2
- 2026Real-Time Detection of Charge Jumps in Superconducting Qubits with a Convolutional Neural Networkcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Learning Plasma Dynamics and Robust Rampdown Trajectories with Predict-First Experiments at TCV
- 2026Offline Reinforcement Learning for Plasma Control in Nuclear Fusion: Codebase and Benchmark
- 2025Gym-TORAX: Open-source software for integrating reinforcement learning with plasma control simulators in tokamak research
- 2025Multi-Timescale Dynamics Model Bayesian Optimization for Plasma Stabilization in Tokamaks
- 2023End-to-end codesign of Hessian-aware quantized neural networks for FPGAs and ASICs
- 2026Dynamic Plasma Shape Control with Arbitrary Sensor Subsets ↗
- 2025Enabling Integrated AI Control on DIII-D: A Control System Design with State-of-the-art Experiments
- 2025Plasma Shape Control via Zero-shot Generative Reinforcement Learning