- Open Access
Machine learning based closed orbit correction for steady state microbunching storage ring
Phys. Rev. Accel. Beams 29, 034602 – Published 17 March, 2026
DOI: https://doi.org/10.1103/h677-zjdg
Abstract
Closed orbit correction is the initial step in error correction, and its results significantly influence subsequent correction of higher-order errors. This is particularly important for an steady state microbunching (SSMB) storage ring, where both the transverse dimensions approach the diffraction limit and the longitudinal length is on the order of hundreds of nanometers. The precision and speed of closed orbit correction are essential for achieving high average power and high brightness radiation in the SSMB storage ring. Here, we introduce a new model to address this issue. Our model integrates the attention mechanism with convolutional neural network (CNN). This combination enables the model to capture both global dependencies and local patterns. Furthermore, we considered the effects of various error distributions on the lattice optics function. By incorporating diverse error scenarios into the training dataset, we enhance the model’s robustness. Through simulations of the SSMB storage ring, we compared the correction results of our model with the singular value decomposition method. It achieves the desired residual closed orbit after only two iterations and shows encouraging correction outcomes across various error distributions. We tested the stability of our model by adjusting the strengths of specific correctors and incorporating beam position monitor (BPM) noise. Additionally, to address the issue of BPM failures, we integrated a Gaussian process model to ensure that our correction process remains effective even when BPMs malfunction.
Physics Subject Headings (PhySH)
Article Text
References (35)
- D. F. Ratner and A. W. Chao, Steady-state microbunching in a storage ring for generating coherent radiation, Phys. Rev. Lett. 105, 154801 (2010).
- Y. Jiao, D. F. Ratner, and A. W. Chao, Terahertz coherent radiation from steady-state microbunching in storage rings with -band radio-frequency system, Phys. Rev. ST Accel. Beams 14, 110702 (2011).
- C. Tang, A. W. Chao, X. Deng, J. Feikes, C. Feng, E. Granados, A. Hoehl, W. Huang, B. Jiang, J. Li, D. Ratner, M. Ries, T. Rui, and X. Wang, An overview of the progress on SSMB, in Proceeding of 60th ICFA Advanced Beam Dynamics Workshop (FLS’18), Shanghai, China (JACoW, Geneva, Switzerland, 2018), pp. 166–170, 10.18429/JACoW-FLS2018-THP2WB02.
- X. Deng, A. W. Chao, J. Feikes, A. Hoehl, W. Huang, R. M. Klein, A. Kruschinski, J. Li, A. Matveenko, Y. Petenev, M. Ries, C. Tang, and L. Yan, Experimental demonstration of the mechanism of steady-state microbunching, Nature (London) 590, 576 (2021).
- R. Nagaoka, Machine stability and orbit correction in ELETTRA, in Proceedings of the 5th European Particle Accelerator Conference, Barcelona, Spain (IOP, Bristol, 1996).
- R. J. Nawrocky, J. W. Bittner, L. Ma, H. M. Rarback, D. P. Siddons, and L. H. Yu, Automatic beam steering in the NSLS storage rings using closed orbit feedback, Nucl. Instrum. Methods Phys. Res., Sect. A 266, 164 (1988).
- Y. Chung, G. Decker, and K. Evans, Closed orbit correction using singular value decomposition of the response matrix, in Proceedings of the 15th Particle Accelerator Conference, PAC-1993, Washington, DC, 1993 (IEEE, New York, 1993), Vol. 3, pp. 2263–2265.
- Y. Chung, G. Decker, K. Evans, J. Safranek, I. So, Y. Tang, W. Corbett, and R. Hettel, Global dc closed orbit correction experiments on the NSLS x-ray ring and spear, in Proceedings of the 15th Particle Accelerator Conference, PAC-1993, Washington, DC, 1993 (IEEE, New York, 1993), Vol. 3, pp. 2275–2277.
- M. Borland, elegant: A flexible SDDS-compliant code for accelerator simulation, 10.2172/761286 (2000).
- A. W. Chao, Lectures on Accelerator Physics (World Scientific, Singapore, 2020).
- E. S. Bozoki and A. Friedman, Neural networks and orbit control in accelerators, in Proceedings of the 4th European Particle Accelerator Conference, EPAC 94, London, England (World Scientific, River Edge, NJ, 1994).
- E. Meier, G. LeBlanc, and Y. E. Tan, Orbit correction studies using neural networks, in Proceedings of 3rd International Particle Accelerator Conference (IPAC’12) (JACoW, Geneva, Switzerland, 2012), pp. 2837–2839, https://proceedings.jacow.org/IPAC2012/papers/weppp057.pdf.
- Y. Bai, Y. Wei, W. Liu, G. Xu, and J. Wang, Research on the slow orbit feedback of BEPCII using machine learning, Radiat. Detect. Technol. Methods 6, 179 (2022).
- R. Li, Q. Zhang, B. Jiang, Z. Zhao, C. Li, K. Wang, and D. Huang, A machine-learning based closed orbit feedback for the SSRF storage ring, arXiv:2212.01010.
- D. Schirmer, Orbit correction with machine learning techniques at the synchrotron light source delta, in Proceedings of the 17th International Conference on Accelerator and Large Experimental Physics Control, ICALEPCS2019 (JACoW, Geneva, Switzerland, 2020).
- S. C. Leemann, P. Amstutz, W. Byrne, M. Ehrlichman, T. Hellert, A. Hexemer, S. Liu, M. A. Marcus, C. N. Melton, H. Nishimura, G. C. Penn, F. Sannibale, D. A. Shapiro, C. Sun, D. M. Ushizima, and M. Venturini, First attempts at applying machine learning to ALS storage ring stabilization, in Proceedings of the 10th International Particle Accelerator Conference, IPAC2019, Melbourne, Australia (JACoW, Geneva, Switzerland, 2019).
- A. Vaswani, N. M. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. N. Gomez, L. Kaiser, and I. Polosukhin, Attention is all you need, in Proceedings of the 31st Conference on Neural Information Processing Systems, NIPS 2017, Long Beach, CA, USA (Curran Associates, Inc., New York, 2017), https://papers.neurips.cc/paper/7181-attention-is-all-you-need.pdf.
- Y. LeCun, Y. Bengio, and G. Hinton, Deep learning, Nature (London) 521, 436 (2015).
- J. L. Elman, Finding structure in time, Cognit. Sci. 14, 179 (1990).
- J. Chung, K. Cho, and Y. Bengio, Empirical evaluation of gated recurrent neural networks on sequence modeling, arXiv:1412.3555.
- S. Hochreiter and J. Schmidhuber, Long short-term memory, Neural Comput. 9, 1735 (1997).
- F. Informatik, Y. Bengio, P. Frasconi, and J. Schmidhuber, Gradient flow in recurrent nets the difficulty of learning long-term dependencies, in A Field Guide to Dynamical Recurrent Neural Networks (Wiley-IEEE Press, Hoboken, NJ, 2003).
- D. Bahdanau, K. Cho, and Y. Bengio, Neural machine translation by jointly learning to align and translate, arXiv:1409.0473.
- L. Kaiser and S. Bengio, Can active memory replace attention?, in Neural Information Processing Systems (Curran Associates, Inc., New York, 2016).
- K. Cho, B. van Merriënboer, C. Gulcehre, D. Bahdanau, F. Bougares, H. Schwenk, and Y. Bengio, Learning phrase representations using RNN encoder–decoder for statistical machine translation, in Proceedings of the 2014 Conference on Empirical Methods in Natural Language Processing (EMNLP), edited by A. Moschitti, B. Pang, and W. Daelemans (Association for Computational Linguistics, Doha, Qatar, 2014), pp. 1724–1734.
- Z. Peng, Z. Guo, W. Huang, Y. Wang, L. Xie, J. Jiao, Q. Tian, and Q. Ye, Conformer: Local features coupling global representations for recognition and detection, IEEE Trans. Pattern Anal. Mach. Intell. 45, 9454 (2023).
- L. Sasal, T. Chakraborty, and A. Hadid, W-transformers: A wavelet-based transformer framework for univariate time series forecasting, in Proceedings of the 2022 21st IEEE International Conference on Machine Learning and Applications (ICMLA) (IEEE Computer Society, Los Alamitos, CA, 2022), pp. 671–676, 10.1109/ICMLA55696.2022.00111.
- H. Wang, J. Xu, R. Yan, C. Sun, and X. Chen, Intelligent bearing fault diagnosis using multi-head attention-based CNN, Procedia Manuf. 49, 112 (2020).
- V. Mnih, N. M. O. Heess, A. Graves, and K. Kavukcuoglu, Recurrent models of visual attention, in Neural Information Processing Systems (MIT Press, Cambridge, MA, 2014).
- K. He, X. Zhang, S. Ren, and J. Sun, Deep residual learning for image recognition, in Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) (IEEE Computer Society, Los Alamitos, CA, 2016), pp. 770–778, 10.1109/CVPR.2016.90.
- J. L. Ba, J. R. Kiros, and G. E. Hinton, Layer normalization, arXiv:1607.06450.
- X. Huang, M. Song, and Z. Zhang, Multi-objective multi-generation Gaussian process optimizer for design optimization, arXiv:1907.00250.
- A. Hanuka, X. Huang, J. Shtalenkova, D. Kennedy, A. Edelen, Z. Zhang, V. R. Lalchand, D. Ratner, and J. Duris, Physics model-informed Gaussian process for online optimization of particle accelerators, Phys. Rev. Accel. Beams 24, 072802 (2021).
- R. Roussel et al., Bayesian optimization algorithms for accelerator physics, Phys. Rev. Accel. Beams 27, 084801 (2024).
- L. Chen, the raw data for training and testing, Tsinghua Cloud (2025), https://cloud.tsinghua.edu.cn/d/82df2cce6c7b4c8eac6e.