Data-efficient learning of exchange-correlation functionals with differentiable DFT

Machine Learning: Science and Technology IOP Publishing 7:2 (2026) 025001-025001

Authors:

Antonius von Strachwitz, Karim K Alaa El-Din, Ana CC Dutra, Sam M Vinko

Abstract:

Abstract Machine learning (ML) density functional approximations (DFAs) have seen a lot of interest in recent years, often being touted as the replacement for well-established non-empirical DFAs, which still dominate the field. Although highly accurate, ML-DFAs typically rely on large amounts of data, are computationally expensive, and fail to generalize beyond their training domain. In this work we show that differentiable DFT with Kohn鈥揝ham regularization can be used to accurately capture the behavior of known local density approximations from small sets of synthetic data without using localized density information. At the same time our analysis shows a strong dependence of the learning on both the amount and type of data as well as on model initialization. By enabling accurate learning from sparse energy data, this approach paves the way towards the development of custom ML-DFAs trained directly on limited experimental or high-level quantum chemistry datasets.

Time-embedded convolutional neural networks for modeling plasma heat transport

Physical Review E American Physical Society (APS) 113:3 (2026) 035303

Authors:

Mufei Luo, Charles Heaton, Yizhen Wang, Daniel Plummer, Mila Fitzgerald, Francesco Miniati, Sam M Vinko, Gianluca Gregori

Abstract:

We introduce a time-embedded convolutional neural network (TCNN) for modeling spatiotemporal heat transport in plasmas, particularly under strongly nonlocal conditions. In our earlier work, the Luciani-Mora-Virmont (LMV) Informed Neural Network (LINN) (Luo , ) combined prior knowledge from the LMV model with kinetic Particle-in-Cell (PIC) data to improve kernel-based heat-flux predictions. While effective under moderately nonlocal conditions, LINN produced physically inconsistent kernels in strongly time-dependent regimes due to its reliance on the quasistationary LMV formulation. To overcome this limitation, TCNN is designed to capture the coupled evolution of both the normalized heat flux and the characteristic nonlocality parameter using a unified neural architecture informed by underlying physical principles. Trained on fully kinetic PIC simulations, TCNN accurately reproduces nonlocal dynamics across a broad range of collisionalities. Our results demonstrate that the combination of time modulation, coupled prediction, and convolutional depth significantly enhances predictive performance, offering a data-driven yet physically consistent framework for multiscale plasma transport problems.

Batch Bayesian optimization of attosecond betatron pulses from laser wakefield acceleration

Communications Physics Springer Nature 9:1 (2026) 92

Authors:

Dominika Maslarova, Albert Hansson, Mufei Luo, Vojt臎ch Horn媒, Julien Ferri, Istvan Pusztai, T眉nde F眉l枚p

Abstract:

Laser wakefield acceleration can generate a femtosecond-scale broadband X-ray betatron radiation pulse from electrons accelerated by an intense laser pulse in a plasma. The micrometer-scale of the source makes wakefield betatron radiation well-suited for advanced imaging techniques, including diffraction and phase-contrast imaging. Recent progress in laser technology can expand these capabilities into the attosecond regime, where the practical applications would significantly benefit from the increased energy contained within the pulse. Here we use numerical simulations combined with batch Bayesian optimization to enhance the radiation produced by an attosecond betatron source. The method enables an efficient exploration of a multi-parameter space and identifies a regime in which a plasma density spike triggers the generation of a high-charge electron beam. This results in an improvement of more than one order of magnitude in the on-axis time-averaged power within the central time containing half of the radiated energy, compared to the reference case without the density spike.

A statistical theory of electronic degrees of freedom in wave packet molecular dynamics

(2026)

Authors:

Daniel Plummer, Pontus Svensson, Wiktor Jasniak, Patrick Hollebon, Sam M Vinko, Gianluca Gregori

Data-efficient learning of exchange-correlation functionals with differentiable DFT

Machine Learning: Science and Technology IOP Publishing (2026)

Authors:

Antonius v. Strachwitz, Karim Kacper Kacper Alaa El-Din, Ana C C. Dutra, Sam M Vinko

Abstract:

<jats:title>Abstract</jats:title> <jats:p>Machine learning (ML) density functional approximations (DFAs) have seen a lot of interest in recent years, often being touted as the replacement for well-established non-empirical DFAs, which still dominate the field. Although highly accurate, ML-DFAs typically rely on large amounts of data, are computationally expensive, and fail to generalize beyond their training domain. In this work we show that differentiable DFT with Kohn-Sham (KS) regularization can be used to accurately capture the behaviour of known local density approximations (LDA) from small sets of synthetic data without using localized density information. At the same time our analysis shows a strong dependence of the learning on both the amount and type of data as well as on model initialization. By enabling accurate learning from sparse energy data, this approach paves the way towards the development of custom ML-DFAs trained directly on limited experimental or high-level quantum chemistry datasets.</jats:p>