Bayesian Optimization and Convolutional Neural Networks for Zernike-Based Wavefront Correction in High Harmonic Generation
Guilherme Grancho D. Fernandes, Duarte Alexandrino, Eduardo Silva, João Matias, Joaquim Pereira
TL;DR
This work tackles aberration-induced beam quality degradation in high-harmonic generation by applying machine-learning-based aberration correction via an SLM. It compares Bayesian optimization, which provides a structured, interpretable baseline via sequential Zernike-mode optimization, with a CNN that learns a direct PSF-to-Zernike mapping to predict correction coefficients from focal-spot images. The CNN achieved 80.39% test accuracy on a small dataset (506 images), illustrating the potential for real-time aberration correction, while Bayesian optimization demonstrates feasibility but is limited by single-objective optimization and mode interactions. The study identifies the dataset size as a primary constraint and outlines future work including additional Zernike modes, closed-loop adaptive optics, transfer learning, multi-objective optimization, and hybrid approaches to combine interpretability with rapid predictions.
Abstract
High harmonic generation (HHG) is a nonlinear process that enables table-top generation of tunable, high-energy, coherent, ultrashort radiation pulses in the extreme ultraviolet (EUV) to soft X-ray range. These pulses find applications in photoemission spectroscopy in condensed matter physics, pump-probe spectroscopy for high-energy-density plasmas, and attosecond science. However, optical aberrations in the high-power laser systems required for HHG degrade beam quality and reduce efficiency. We present a machine learning approach to optimize aberration correction using a spatial light modulator. We implemented and compared Bayesian optimization and convolutional neural network (CNN) methods to predict optimal Zernike polynomial coefficients for wavefront correction. Our CNN achieved promising results with 80.39% accuracy on test data, demonstrating the potential for automated aberration correction in HHG systems.
