Microsecond-Latency Feedback at a Particle Accelerator by Online Reinforcement Learning on Hardware

TL;DR

This paper demonstrates a microsecond-latency reinforcement learning system deployed on hardware for real-time control in a particle accelerator, enabling ultra-fast feedback and autonomous operation.

Contribution

It introduces a hardware-accelerated RL architecture for real-time control in particle accelerators, achieving microsecond latency and demonstrating viability in a real experimental setting.

Findings

Performance comparable to existing feedback systems

Real-time control of betatron oscillations achieved

Potential for application in various large-scale facilities

Abstract

The commissioning and operation of future large-scale scientific experiments will challenge current tuning and control methods. Reinforcement learning (RL) algorithms are a promising solution thanks to their capability of autonomously tackling a control problem based on a task parameterized by a reward function. The conventionally utilized machine learning (ML) libraries are not intended for microsecond latency applications, as they mostly optimize for throughput performance. On the other hand, most of the programmable logic implementations are meant for computation acceleration, not being intended to work in a real-time environment. To overcome these limitations of current implementations, RL needs to be deployed on-the-edge, i.e. on to the device gathering the training data. In this paper we present the design and deployment of an experience accumulator system in a particle accelerator. In this system deep-RL algorithms run using hardware acceleration and act within a few microseconds, enabling the use of RL for control of ultra-fast phenomena. The training is performed offline to reduce the number of operations carried out on the acceleration hardware. The proposed architecture was tested in real experimental conditions at the Karlsruhe research accelerator (KARA), serving also as a synchrotron light source, where the system was used to control induced horizontal betatron oscillations in real-time. The results showed a performance comparable to the commercial feedback system available at the accelerator, proving the viability and potential of this approach. Due to the self-learning and reconfiguration capability of this implementation, its seamless application to other control problems is possible. Applications range from particle accelerators to large-scale research and industrial facilities.