13.–16. Jan. 2025
Department of Physics, University of Oxford
Europe/London Zeitzone

This event is part of the Laser-Plasma Accelerator Seminars. Click here for more information, including data protection.

Enabling Large-Scale Data-Driven AI Workflows with PIConGPU & openPMD: Demonstrating Experimental Parameter Inference and In-Transit Learning

14.01.2025, 11:45
15m
Department of Physics, University of Oxford

Department of Physics, University of Oxford

Parks Rd, Oxford OX1 3PU, UK

Sprecher

Richard Pausch (HZDR)

Beschreibung

In this talk, we will explore the current development of PIConGPU in machine learning-based simulations for plasma acceleration and highlight three key applications. These projects mark significant advances in the integration of AI and advanced data workflows into plasma physics research with PIConGPU. They illustrate not only our current methods, but also our vision for future in-transit AI-assisted simulation analysis.

First, we present automated optimizations of large three-dimensional particle-in-cell (PIC) simulations to find parameters consistent with self-truncated ionization injection experiments during laser wakefield acceleration. By combining Bayesian optimization with the PIConGPU framework and automating the entire parameter tuning workflow with Snakemake, we efficiently achieve excellent convergence with experimental measurements.

Second, we present radINN, an invertible neural network designed to use emitted radiation spectra to quantify injected charge. Trained with synthetic data from PIConGPU simulations, radINN excels at identifying injection processes and provides valuable insights for improving future experimental diagnostic.

Finally, we present a new heterogeneous streaming workflow that streams plasma simulation data directly into a machine learning application, without writing any data to disk. This approach leverages openPMD and ADIOS2 to facilitate real-time model training during data transfer, overcoming the challenges of petabyte-scale data storage and I/O bottlenecks. We demonstrate this workflow by tackling the inverse problem of predicting particle dynamics from radiation in a PIConGPU simulation of the Kelvin-Helmholtz instability.

Hauptautor

Co-Autoren

Dr. Jeffrey Kelling (HZDR) Friedrich Bethke (HZDR) Vicente Bolea (Kitware Inc.) Dr. Michael Bussmann (HZDR, CASUS) Ankush Checkervarty (HZDR) Dr. Alexander Debus (HZDR) Jan Ebert (Forschungszentrum Jülich) Greg Eisenhauer (Georgia Institute of Technology) Vineeth Gutta (University of Delaware) Stefan Kesselheim (Forschungszentrum Jülich) Scott Klasky (Oak Ridge National Laboratory) Norbert Podhorszki (Oak Ridge National Laboratory) Franz Pöschel (HZDR, CASUS) David Rogers (Oak Ridge National Laboratory) Jeyhun Rustamov (HZDR) Steve Schmerler (HZDR) Prof. Ulrich Schramm (HZDR) Dr. Klaus Steiniger (HZDR, CASUS) Jessica Tiebel (HZDR, TU Dresden) René Widera (HZDR) Anna Willmann (HZDR) Prof. Sunita Chandrasekaran (University of Delaware)

Präsentationsmaterialien

Es gibt derzeit keine Materialien.