Multi-objective Bayesian active learning for MeV-ultrafast electron diffraction.
Journal
Nature communications
ISSN: 2041-1723
Titre abrégé: Nat Commun
Pays: England
ID NLM: 101528555
Informations de publication
Date de publication:
03 Jun 2024
03 Jun 2024
Historique:
received:
06
09
2023
accepted:
16
05
2024
medline:
4
6
2024
pubmed:
4
6
2024
entrez:
3
6
2024
Statut:
epublish
Résumé
Ultrafast electron diffraction using MeV energy beams(MeV-UED) has enabled unprecedented scientific opportunities in the study of ultrafast structural dynamics in a variety of gas, liquid and solid state systems. Broad scientific applications usually pose different requirements for electron probe properties. Due to the complex, nonlinear and correlated nature of accelerator systems, electron beam property optimization is a time-taking process and often relies on extensive hand-tuning by experienced human operators. Algorithm based efficient online tuning strategies are highly desired. Here, we demonstrate multi-objective Bayesian active learning for speeding up online beam tuning at the SLAC MeV-UED facility. The multi-objective Bayesian optimization algorithm was used for efficiently searching the parameter space and mapping out the Pareto Fronts which give the trade-offs between key beam properties. Such scheme enables an unprecedented overview of the global behavior of the experimental system and takes a significantly smaller number of measurements compared with traditional methods such as a grid scan. This methodology can be applied in other experimental scenarios that require simultaneously optimizing multiple objectives by explorations in high dimensional, nonlinear and correlated systems.
Identifiants
pubmed: 38830874
doi: 10.1038/s41467-024-48923-9
pii: 10.1038/s41467-024-48923-9
doi:
Types de publication
Journal Article
Langues
eng
Sous-ensembles de citation
IM
Pagination
4726Informations de copyright
© 2024. This is a U.S. Government work and not under copyright protection in the US; foreign copyright protection may apply.
Références
Filippetto, D. et al. Ultrafast electron diffraction: visualizing dynamic states of matter. Rev. Mod. Phy. 94, 045004 (2022).
doi: 10.1103/RevModPhys.94.045004
SLAC. Mega-electronvolt Ultrafast Electron Diffraction Instrument. https://lcls.slac.stanford.edu/instruments/mev-ued (2024).
Shen, X. et al. Femtosecond gas-phase mega-electron-volt ultrafast electron diffraction. Struct. Dyn. 6, 054305 (2019).
doi: 10.1063/1.5120864
pubmed: 31649964
pmcid: 6796191
Kim, H. W. et al. Towards jitter-free ultrafast electron diffraction technology. Nat. Photonics 14, 245–249 (2020).
doi: 10.1038/s41566-019-0566-4
Qi, F. et al. Breaking 50 femtosecond resolution barrier in MeV ultrafast electron diffraction with a double bend achromat compressor. Phys. Rev. Lett. 124, 134803 (2020).
doi: 10.1103/PhysRevLett.124.134803
pubmed: 32302182
Ji, F. et al. Ultrafast relativistic electron nanoprobes. Comm. Phys. 2, 54 (2019).
doi: 10.1038/s42005-019-0154-4
Nunes, J. P. F. et al. Liquid-phase mega-electron-volt ultrafast electron diffraction. Struct. Dyn. 7, 024301 (2020).
doi: 10.1063/1.5144518
pubmed: 32161776
pmcid: 7062553
Wolf, T. J. A. et al. The photochemical ring-opening of 1,3-cyclohexadiene imaged by ultrafast electron diffraction. Nat. Chem. 11, 504 (2019).
doi: 10.1038/s41557-019-0252-7
pubmed: 30988415
Yang, J. et al. Simultaneous observation of nuclear and electronic dynamics by ultrafast electron diffraction. Science 368, 885–889 (2020).
doi: 10.1126/science.abb2235
pubmed: 32439793
Yang, J. et al. Direct observation of ultrafast hydrogen bond strengthening in liquid water. Nature 596, 531–535 (2021).
doi: 10.1038/s41586-021-03793-9
pubmed: 34433948
Sie, E. J. et al. An ultrafast symmetry switch in a Weyl semimetal. Nature 565, 61–66 (2019).
doi: 10.1038/s41586-018-0809-4
pubmed: 30602749
Aditya Sood. Universal phase dynamics in VO2 switches revealed by ultrafast operando diffraction. Science 373, 352–355 (2021).
doi: 10.1126/science.abc0652
pubmed: 34437156
Vaswani, A. et al. Attention is all you need. In Proceedings of the 2017 International Conference on Neural Information Processing Systems (NeurIPS 2017). 6000–6010, https://doi.org/10.5555/3295222.3295349 (2017).
Jumper, J. Highly accurate protein structure prediction with AlphaFold. Nature 596, 583–589 (2021).
doi: 10.1038/s41586-021-03819-2
pubmed: 34265844
pmcid: 8371605
Degrave, J. Magnetic control of tokamak plasmas through deep reinforcement learning. Nature 602, 414–419 (2022).
doi: 10.1038/s41586-021-04301-9
pubmed: 35173339
pmcid: 8850200
Edelen, A. et al. Machine learning for orders of magnitude speedup in multiobjective optimization of particle accelerator systems. Phys. Rev. Accel. Beams 23, 044601 (2020).
doi: 10.1103/PhysRevAccelBeams.23.044601
Sheinker, A. Demonstration of model-independent control of the longitudinal phase space of electron beams in the linac-coherent light source with femtosecond resolution. Phys. Rev. Lett 121, 044801 (2018).
doi: 10.1103/PhysRevLett.121.044801
Leemann, S. C. Demonstration of machine learning-based model-independent stabilization of source properties in synchrotron light sources. Phys. Rev. Lett. 123, 194801 (2019).
doi: 10.1103/PhysRevLett.123.194801
pubmed: 31765214
Duris, J. Bayesian optimization of a free-electron laser. Phys. Rev. Lett 124, 124801 (2020).
doi: 10.1103/PhysRevLett.124.124801
pubmed: 32281869
Roussel, R. Multiobjective Bayesian optimization for online accelerator tuning. Phys. Rev. Accel. Beams 24, 062801 (2021).
doi: 10.1103/PhysRevAccelBeams.24.062801
Hanuka, A. Physics model-informed Gaussian process for online optimization of particle accelerators. Phys. Rev. Accel. Beams 24, 072802 (2021).
doi: 10.1103/PhysRevAccelBeams.24.072802
Roussel, R. et al. Turn-key constrained parameter space exploration for particle accelerators using Bayesian active learning. Nat. Commun. 12, 5621 (2021).
doi: 10.1038/s41467-021-25757-3
Roussel, R. et al. Bayesian optimization algorithms for accelerator physics. arXiv:2312.05667 https://doi.org/10.48550/arXiv.2312.05667 (2023).
Rasmussen, C. E. & Williams, K. I. Gaussian Processes for Machine Learning, Adaptive Computation and Machine Learning. Ch. 2. 13, (MIT Press, Cambridge, MA, 2006).
Emmerich, M., Yang, K., Deutz, A., Wang, H. & Fonseca, C. M. A multi-criteria generalization of Bayesian global optimization. In Advances in Stochastic and Deterministic Global Optimization, Springer Optimization and Its Applications Vol. 106 (eds Pardalos, P., Zhigljavsky, A. & Žilinskas, J.). 229–242 (Springer, Cham, 2024).
Kusne, A. G. et al. On-the-fly closed-loop materials discovery via Bayesian active learning. Nat. Commun. 11, 5966 (2020).
doi: 10.1038/s41467-020-19597-w
pubmed: 33235197
pmcid: 7686338
Abolhasani, M. & Kumacheva, E. The rise of self-driving labs in chemical and materials sciences. Nat. Synth 2, 483–492 (2023).
doi: 10.1038/s44160-022-00231-0
Maier, H. R. et al. Introductory overview: optimization using evolutionary algorithms and other metaheuristics. Environ. Model. Softw. 114, 195 (2019).
doi: 10.1016/j.envsoft.2018.11.018
Jalas, S. et al. Tuning curves for a laser-plasma accelerator. Phys. Rev. Accel. Beams 26, 071302 (2023).
doi: 10.1103/PhysRevAccelBeams.26.071302
Yang, K., Emmerich, M., Deutz, A. & Bäck, T. Multi-objective Bayesian global optimization using expected hypervolume improvement gradient. Swarm Evol. Comput. 44, 945 (2019).
doi: 10.1016/j.swevo.2018.10.007
Kogar, A. et al. Light-induced charge density wave in LaTe3. Nat. Phys. 16, 159–163 (2020).
doi: 10.1038/s41567-019-0705-3
Sood, A. et al. Bidirectional phonon emission in two-dimensional heterostructures triggered by ultrafast charge transfer. Nat. Nanotechnol. 18, 29–35 (2023).
doi: 10.1038/s41565-022-01253-7
pubmed: 36543882
Cao, Y. et al. Unconventional superconductivity in magic-angle graphene superlattices. Nature 556, 43–50 (2018).
doi: 10.1038/nature26160
pubmed: 29512651
Daulton, S., Eriksson, D., Balandat, M. & Bakshy, E. Multi-Objective Bayesian Optimization over High-Dimensional Search Spaces,in Proceedings of the Thirty-Eighth Conference on Uncertainty in Artificial Intelligence, PMLR 180, 507–517, https://proceedings.mlr.press/v180/daulton22a/daulton22a.pdf (2022).
Xu, C., Roussel, R. & Edelen, A. Neural network prior mean for particle accelerator injector tuning. in Machine Learning And The Physical Sciences Workshop, NeurIPS 2022. (2022).
Yang, K. et al. Efficient computation of expected hypervolume improvement using box decomposition algorithms. J. Glob. Optim. 75, 3–34 (2019).
doi: 10.1007/s10898-019-00798-7
Daulton, S., Balandat, M. & Bakshy, E. Differentiable expected hypervolume improvement for parallel multi-objective Bayesian optimization. In Proceedings of the 2020 International Conference on Neural Information Processing Systems (NeurIPS 2020). 9851–9864, arXiv https://doi.org/10.48550/arXiv.2006.05078 (2020).
Snively, E. C. et al. Femtosecond compression dynamics and timing jitter suppression in a THz-driven electron bunch compressor. Phys. Rev. Lett 124, 054801 (2020).
doi: 10.1103/PhysRevLett.124.054801
pubmed: 32083908
Li, R. et al. Terahertz-based subfemtosecond metrology of relativistic electron beams. Phys. Rev. Accel. Beams 22, 012803 (2019).
doi: 10.1103/PhysRevAccelBeams.22.012803
Othman, M. A. K. et al. Measurement of femtosecond dynamics of ultrafast electron beams through terahertz compression and time-stamping. Appl. Phys. Lett 122, 141107 (2023).
doi: 10.1063/5.0134733
Williams, J. et al. Active control of bright electron beams with RF optics for femtosecond microscopy. Struct. Dyn. 4, 044035 (2017).
doi: 10.1063/1.4999456
pubmed: 28868325
pmcid: 5565489
Ji, F. et al. Raw dataset for 'multi-objective Bayesian active learning for MeV-ultrafast electron diffraction’. Zenodo https://doi.org/10.5281/zenodo.11095450 (2024).
Balandat, M. et al. BoTorch: a framework for efficient Monte-Carlo Bayesian optimization. In Proceedings of the 2020 International Conference on Neural Information Processing Systems (NeurIPS 2020). 21524–21538 https://doi.org/10.48550/arXiv.1910.06403 (2020).
Roussel, R. et al. Xopt: A simplified framework for optimization of accelerator problems using advanced algorithms. In Proceedings of 14th International Particle Accelerator Conference (IPAC 2023). 4796–4799 (JACoW Publishing, 2023).