Time-inversion of spatiotemporal beam dynamics using uncertainty-aware latent evolution reversal

التفاصيل البيبلوغرافية
العنوان: Time-inversion of spatiotemporal beam dynamics using uncertainty-aware latent evolution reversal
المؤلفون: Rautela, Mahindra, Williams, Alan, Scheinker, Alexander
سنة النشر: 2024
المجموعة: Computer Science
Physics (Other)
مصطلحات موضوعية: Physics - Accelerator Physics, Computer Science - Machine Learning
الوصف: Charged particle dynamics under the influence of electromagnetic fields is a challenging spatiotemporal problem. Many high performance physics-based simulators for predicting behavior in a charged particle beam are computationally expensive, limiting their utility for solving inverse problems online. The problem of estimating upstream six-dimensional phase space given downstream measurements of charged particles in an accelerator is an inverse problem of growing importance. This paper introduces a reverse Latent Evolution Model (rLEM) designed for temporal inversion of forward beam dynamics. In this two-step self-supervised deep learning framework, we utilize a Conditional Variational Autoencoder (CVAE) to project 6D phase space projections of a charged particle beam into a lower-dimensional latent distribution. Subsequently, we autoregressively learn the inverse temporal dynamics in the latent space using a Long Short-Term Memory (LSTM) network. The coupled CVAE-LSTM framework can predict 6D phase space projections across all upstream accelerating sections based on single or multiple downstream phase space measurements as inputs. The proposed model also captures the aleatoric uncertainty of the high-dimensional input data within the latent space. This uncertainty, which reflects potential uncertain measurements at a given module, is propagated through the LSTM to estimate uncertainty bounds for all upstream predictions, demonstrating the robustness of the LSTM against in-distribution variations in the input data.
Comment: arXiv admin note: text overlap with arXiv:2403.13858
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2408.07847
رقم الأكسشن: edsarx.2408.07847
قاعدة البيانات: arXiv