- Physical Review Journals
All JournalsPhysics Magazine

Physical Review Accelerators and Beams

Reuse & Permissions

It is not necessary to obtain permission to reuse this article or its components as it is available under the terms of the Creative Commons Attribution 4.0 International license. This license permits unrestricted use, distribution, and reproduction in any medium, provided attribution to the author(s) and the published article's title, journal citation, and DOI are maintained. Please note that some figures may have been included with permission from other third parties. It is your responsibility to obtain the proper permission from the rights holder directly for these figures.

Export citation

Export citation

Choose format for download:

Download Citation
  • Letter
  • Open Access

Efficient dynamic and momentum aperture optimization for lattice design using multipoint Bayesian algorithm execution

Z. Zhang1, I. Agapov2, S. Gasiorowski1, T. Hellert3, W. Neiswanger4, X. Huang1, and D. Ratner5,6,*

  • *Contact author: ratner@jlab.org
PDF

Phys. Rev. Accel. Beams 29, L102001 – Published 1 October, 2026

DOI: https://doi.org/10.1103/4t6l-7jmg

Abstract

We demonstrate that multipoint Bayesian algorithm execution (multipointBAX) can overcome fundamental computational challenges in storage ring design optimization. Dynamic aperture and momentum aperture optimization is a multipoint, multiobjective design task for storage rings, ultimately informing the flux of x-ray sources and luminosity of colliders. Current production black-box optimization methods require extensive particle-tracking simulations for each trial configuration; the high computational cost restricts the extent of the search to ∼103 configurations and therefore limits the quality of the final design. We remove this bottleneck using multipointBAX, which selects, simulates, and models each trial configuration at the single-particle level. We demonstrate our approach on a design for a fourth-generation light source, with neural-network-powered multipointBAX achieving equivalent Pareto front results using more than 2 orders of magnitude fewer tracking computations compared to genetic algorithms. The significant reduction in cost positions multipointBAX as a promising alternative to black-box optimization, and we anticipate multipointBAX will be instrumental in the design of future light sources, colliders, and large-scale scientific facilities.

View figure in article

Physics Subject Headings (PhySH)

Article Text

Supplemental Material

References (51)

  1. M. Borland, V. Sajaev, L. Emery, and A. Xiao, Direct methods of optimization of storage ring dynamic and momentum aperture, in Proceedings of PAC09, Vancouver, BC, Canada (JACoW Publishing, Geneva, Switzerland, 2009), pp. 3850–3852, paper TH6PFP062.
  2. L. Yang, D. Robin, F. Sannibale, C. Steier, and W. Wan, Global optimization of an accelerator lattice using multiobjective genetic algorithms, Nucl. Instrum. Methods Phys. Res., Sect. A 609, 50 (2009).
  3. X. Huang and J. Safranek, Nonlinear dynamics optimization with particle swarm and genetic algorithms for SPEAR3 emittance upgrade, Nucl. Instrum. Methods Phys. Res., Sect. A 757, 48 (2014).
  4. S. Nagaitsev, Z. Huang, J. Power, J.-L. Vay, P. Piot, L. Spentzouris, and J. Rosenzweig, Accelerator and beam physics research goals and opportunities, Fermi National Accelerator Laboratory, Tech. Rep. FERMILAB-FN-1123, 2021.
  5. S. Nagaitsev, V. Shiltsev, A. Valishev, T. Zolkin, J.-L. Vay, M. Bai, Y. Cai, M. J. Hogan, Z. Huang, J. Seeman et al., Accelerator and beam physics: Grand challenges and research opportunities, arXiv:2203.06816.
  6. S. A. Miskovich, W. Neiswanger, W. Colocho, C. Emma, J. Garrahan, T. Maxwell, C. Mayes, S. Ermon, A. Edelen, and D. Ratner, Multipoint-BAX: A new approach for efficiently tuning particle accelerator emittance via virtual objectives, Mach. Learn.: Sci. Technol. 5, 015004 (2024).
  7. C. Steier, D. Robin, L. Nadolski, W. Decking, Y. Wu, and J. Laskar, Measuring and optimizing the momentum aperture in a particle accelerator, Phys. Rev. E 65, 056506 (2002).
  8. J. Kim, X. Huang, K. Tian, M. Song, J. Safranek, and P. Raimondi, A hybrid multi-bend achromat lattice design for SSRL-X, in Proceedings of IPAC2022 (JACoW Publishing, Bangkok, Thailand, 2022), pp. 207–209, paper MOPOST054.
  9. T. Murata and H. Ishibuchi, MOGA: Multi-objective genetic algorithms, in Proceedings of 1995 IEEE International Conference on Evolutionary Computation (IEEE, Perth, WA, Australia, 1995), Vol. 1, pp. 289–294.
  10. C. A. Coello Coello and M. S. Lechuga, MOPSO: A proposal for multiple objective particle swarm optimization, in Proceedings of the 2002 Congress on Evolutionary Computation (CEC’02) (IEEE, Honolulu, HI, USA, 2002), Vol. 2, pp. 1051–1056.
  11. R. Stevens, V. Taylor, J. Nichols, A. B. Maccabe, K. Yelick, and D. Brown, AI for science: Report on the department of energy (DOE) town halls on artificial intelligence (AI) for science, Argonne National Laboratory, Argonne, IL, Tech. Rep. ANL-20/17, 2020.
  12. R. Roussel, A. L. Edelen, T. Boltz, D. Kennedy, Z. Zhang, F. Ji, X. Huang, D. Ratner, A. Santamaria Garcia, C. Xu et al., Bayesian optimization algorithms for accelerator physics, Phys. Rev. Accel. Beams 27, 084801 (2024).
  13. X. Huang, Z. Zhang, and M. Song, Multi-objective multi-generation Gaussian process optimizer, in Proceedings of IPAC2021 (JACoW Publishing, Campinas, Brazil, 2021), pp. 3383–3386, paper WEPAB304.
  14. Y. Li, W. Cheng, L. H. Yu, and R. Rainer, Genetic algorithm enhanced by machine learning in dynamic aperture optimization, Phys. Rev. Accel. Beams 21, 054601 (2018).
  15. J. Laskar, Frequency analysis for multi-dimensional systems. Global dynamics and diffusion, Physica D 67, 257 (1993).
  16. Y. Papaphilippou, Detecting chaos in particle accelerators through the frequency map analysis method, Chaos 24, 024412 (2014).
  17. Y. Li, Y. Hao, K. Hwang, R. Rainer, A. He, and A. Liu, Fast dynamic aperture optimization with forward-reversal integration, Nucl. Instrum. Methods Phys. Res., Sect. A 988, 164936 (2021).
  18. J. Qiang, J. Wan, A. Qiang, and Y. Hao, Fast chaos indicator from auto-differentiation for dynamic aperture optimization, Nucl. Instrum. Methods Phys. Res., Sect. A 1087, 171427 (2026).
  19. K. Hwang, C. Mitchell, and R. Ryne, Rapidly converging chaos indicator for studying dynamic aperture in a storage ring with space charge, Phys. Rev. Accel. Beams 23, 084601 (2020).
  20. C. E. Montanari, R. B. Appleby, A. Bazzani, A. Fornara, M. Giovannozzi, S. Redaelli, G. Sterbini, and G. Turchetti, Chaos indicators for nonlinear dynamics in circular particle accelerators, Eur. Phys. J. Plus 140, 603 (2025).
  21. E. Andrés Sánchez, A. Flores, J. Hernández-Cobos, and A. Antillón, Multiple quasi-invariant surfaces for dynamic aperture optimization in low-emittance synchrotrons, Commun. Nonlinear Sci. Numer. Simul. 154, 109533 (2026).
  22. M. Kranjčević, B. Riemann, A. Adelmann, and A. Streun, Multiobjective optimization of the dynamic aperture using surrogate models based on artificial neural networks, Phys. Rev. Accel. Beams 24, 014601 (2021).
  23. J. Wan, P. Chu, and Y. Jiao, Neural network-based multiobjective optimization algorithm for nonlinear beam dynamics, Phys. Rev. Accel. Beams 23, 081601 (2020).
  24. Y. Lu, S. C. Leemann, C. Sun, M. P. Ehrlichman, T. Hellert, H. Nishimura, and M. Venturini, Enhancing the MOGA optimization process at ALS-U with machine learning, in Proceedings of IPAC2021 (JACoW Publishing, Campinas, Brazil, 2021), pp. 387–390.
  25. Y. Lu, S. C. Leemann, C. Sun, M. P. Ehrlichman, H. Nishimura, M. Venturini, and T. Hellert, Demonstration of machine learning-enhanced multi-objective optimization of ultrahigh-brightness lattices for 4th-generation synchrotron light sources, Nucl. Instrum. Methods Phys. Res., Sect. A 1050, 168192 (2023).
  26. M. Schenk, L. Coyle, M. Giovannozzi, A. Mereghetti, T. Pieloni, E. Krymova, and G. Obozinski, Modeling particle stability plots for accelerator optimization using adaptive sampling, in Proceedings of IPAC2021 (JACoW Publishing, Campinas, Brazil, 2021), pp. 1923–1926.
  27. F. F. Van der Veken, M. Giovannozzi, E. H. Maclean, C. E. Montanari, and G. Valentino, Using machine learning to improve dynamic aperture estimates, in Proceedings of IPAC2021 (JACoW Publishing, Campinas, Brazil, 2021), pp. 134–137.
  28. J. Wan and Y. Jiao, Machine learning enabled fast evaluation of dynamic aperture for storage ring accelerators, New J. Phys. 24, 063030 (2022).
  29. D. D. Croce, M. Giovannozzi, T. Pieloni, M. Seidel, and F. F. Van der Veken, Accelerating dynamic aperture evaluation using deep neural networks, J. Phys. Conf. Ser. 2687, 062032 (2024).
  30. M. Casanova, B. Dalena, L. Bonaventura, and M. Giovannozzi, Ensemble reservoir computing for dynamical systems: Prediction of phase-space stable region for hadron storage rings, Eur. Phys. J. Plus 138, 559 (2023).
  31. D. Di Croce, M. Giovannozzi, E. Krymova, T. Pieloni, S. Redaelli, M. Seidel, R. Tomás, and F. F. Van der Veken, Optimizing dynamic aperture studies with active learning, J. Instrum. 19, P04004 (2024).
  32. M. Nemec, D. W. Zingg, and T. H. Pulliam, Multipoint and multi-objective aerodynamic shape optimization, AIAA J. 42, 1057 (2004).
  33. G. K. W. Kenway and J. R. R. A. Martins, Multipoint high-fidelity aerostructural optimization of a transport aircraft configuration, J. Aircr. 51, 144 (2014).
  34. R. P. Liem, G. K. W. Kenway, and J. R. R. A. Martins, Multimission aircraft fuel-burn minimization via multipoint aerostructural optimization, AIAA J. 53, 104 (2015).
  35. X. Chai, X. Yu, and Y. Wang, Multipoint optimization on fuel efficiency in conceptual design of wide-body aircraft, Chin. J. Aeronaut. 31, 99 (2018).
  36. T. Yamashita, N. Sato, H. Kino, T. Miyake, K. Tsuda, and T. Oguchi, Crystal structure prediction accelerated by Bayesian optimization, Phys. Rev. Mater. 2, 013803 (2018).
  37. K. Terayama, H. Iwata, M. Araki, Y. Okuno, K. Tsuda, and B. Rost, Machine learning accelerates MD-based binding pose prediction between ligands and proteins, Bioinformatics 34, 770 (2018).
  38. S. A. Miskovich, F. Montes, G. P. A. Berg, J. Blackmon, K. A. Chipps, M. Couder, C. M. Deibel, K. Hermansen, A. A. Hood, R. Jain, T. Ruland, H. Schatz, M. S. Smith, P. Tsintari, and L. Wagner, Online Bayesian optimization for a recoil mass separator, Phys. Rev. Accel. Beams 25, 044601 (2022).
  39. R. Roussel, D. Kennedy, A. Edelen, A. Ody, and E. Wisniewski, Autonomous beam alignment through quadrupole triplets using Bayesian algorithm execution, in Proceedings of the 32nd Linear Accelerator Conference (LINAC2024), Chicago, IL, USA (JACoW Publishing, Geneva, Switzerland, 2024), pp. 765–767, paper THPB066.
  40. E. Matzoukas, C. Xu, E. Blomley, E. Bründermann, J. Gethmann, G. De Carne, and A.-S. Müller, Efficient accelerator operation with artificial intelligence based optimization methods, in Proceedings of the 16th International Particle Accelerator Conference (IPAC’25), Taipei, Taiwan (JACoW Publishing, Geneva, Switzerland, 2025), pp. 2722–2725, paper THPM018.
  41. W. Neiswanger, K. A. Wang, and S. Ermon, Bayesian algorithm execution: Estimating computable properties of black-box functions using mutual information, in Proceedings of the 38th International Conference on Machine Learning, Proceedings of Machine Learning Research, edited by M. Meila and T. Zhang (PMLR, 2021). Vol. 139, pp. 8005–8015.
  42. S. R. Chitturi, A. Ramdas, Y. Wu, B. Rohr, S. Ermon, J. Dionne, F. H. da Jornada, M. Dunne, C. Tassone, W. Neiswanger, and D. Ratner, Targeted materials discovery using Bayesian algorithm execution, npj Comput. Mater. 10, 156 (2024).
  43. See Supplemental Material at http://link.aps.org/supplemental/10.1103/4t6l-7jmg for the full multipointBAX algorithm and hyperparameters, the acquisition function, the neural-network architecture, a comparison on an alternative single-configuration metric, the early-stopping indicator, and the average time cost per particle-tracking simulation for BAX and NSGA-II, which includes Ref. [44].
  44. D. Ginsbourger, R. Le Riche, and L. Carraro, Kriging is well-suited to parallelize optimization, Computational Intelligence in Expensive Optimization Problems, Adaptation, Learning, and Optimization, edited by Y. Tenne and C.-K. Goh (Springer, Berlin, 2010), Vol. 2, pp. 131–162.
  45. Y. Xie, T. Takikawa, S. Saito, O. Litany, S. Yan, N. Khan, F. Tombari, J. Tompkin, V. Sitzmann, and S. Sridhar, Neural fields in visual computing and beyond, Comput. Graphics Forum 41, 641 (2022).
  46. W. Rogers, N. Carmignani, L. Farvacque, and B. Nash, pyAT: A python build of accelerator toolbox, in Proceedings of IPAC2017 (JACoW Publishing, Copenhagen, Denmark, 2017), pp. 3855–3857, paper THPAB060.
  47. P. Raimondi, X. Huang, J. Kim, J. Safranek, and T. Rabedeau, Advanced storage ring lattice options based on hybrid six-bend achromat for stanford synchrotron radiation lightsource upgrade, Nucl. Instrum. Methods Phys. Res., Sect. A 1061, 169137 (2024).
  48. P. Raimondi and S. M. Liuzzo, Toward a diffraction limited light source, Phys. Rev. Accel. Beams 26, 021601 (2023).
  49. X. Huang and J. Safranek, Online optimization of storage ring nonlinear beam dynamics, Phys. Rev. ST Accel. Beams 18, 084001 (2015).
  50. L. McInnes, J. Healy, and J. Melville, UMAP: Uniform manifold approximation and projection for dimension reduction, arXiv:1802.03426.
  51. M. Benedikt, A. Blondel, P. Janot, M. Mangano, and F. Zimmermann, Future circular colliders succeeding the LHC, Nat. Phys. 16, 402 (2020).

Outline

Information

Phys. Rev. Accel. Beams 29, L102001– Published 1 October, 2026
  • Received 12 December 2025
  • Accepted 1 September 2026
Crossmark - Check for updates button
Creative Commons Logo

Published by the American Physical Society under the terms of the Creative Commons Attribution 4.0 International license. Further distribution of this work must maintain attribution to the author(s) and the published article’s title, journal citation, and DOI.

Published by the American Physical Society

Sign In to Your Journals Account

Filter

Filter

Article Lookup

Enter a citation