-
Practical applications of machine-learned flows on gauge fields
Authors:
Ryan Abbott,
Michael S. Albergo,
Denis Boyda,
Daniel C. Hackett,
Gurtej Kanwar,
Fernando Romero-López,
Phiala E. Shanahan,
Julian M. Urban
Abstract:
Normalizing flows are machine-learned maps between different lattice theories which can be used as components in exact sampling and inference schemes. Ongoing work yields increasingly expressive flows on gauge fields, but it remains an open question how flows can improve lattice QCD at state-of-the-art scales. We discuss and demonstrate two applications of flows in replica exchange (parallel tempe…
▽ More
Normalizing flows are machine-learned maps between different lattice theories which can be used as components in exact sampling and inference schemes. Ongoing work yields increasingly expressive flows on gauge fields, but it remains an open question how flows can improve lattice QCD at state-of-the-art scales. We discuss and demonstrate two applications of flows in replica exchange (parallel tempering) sampling, aimed at improving topological mixing, which are viable with iterative improvements upon presently available flows.
△ Less
Submitted 17 April, 2024;
originally announced April 2024.
-
Applications of flow models to the generation of correlated lattice QCD ensembles
Authors:
Ryan Abbott,
Aleksandar Botev,
Denis Boyda,
Daniel C. Hackett,
Gurtej Kanwar,
Sébastien Racanière,
Danilo J. Rezende,
Fernando Romero-López,
Phiala E. Shanahan,
Julian M. Urban
Abstract:
Machine-learned normalizing flows can be used in the context of lattice quantum field theory to generate statistically correlated ensembles of lattice gauge fields at different action parameters. This work demonstrates how these correlations can be exploited for variance reduction in the computation of observables. Three different proof-of-concept applications are demonstrated using a novel residu…
▽ More
Machine-learned normalizing flows can be used in the context of lattice quantum field theory to generate statistically correlated ensembles of lattice gauge fields at different action parameters. This work demonstrates how these correlations can be exploited for variance reduction in the computation of observables. Three different proof-of-concept applications are demonstrated using a novel residual flow architecture: continuum limits of gauge theories, the mass dependence of QCD observables, and hadronic matrix elements based on the Feynman-Hellmann approach. In all three cases, it is shown that statistical uncertainties are significantly reduced when machine-learned flows are incorporated as compared with the same calculations performed with uncorrelated ensembles or direct reweighting.
△ Less
Submitted 28 May, 2024; v1 submitted 19 January, 2024;
originally announced January 2024.
-
Normalizing flows for lattice gauge theory in arbitrary space-time dimension
Authors:
Ryan Abbott,
Michael S. Albergo,
Aleksandar Botev,
Denis Boyda,
Kyle Cranmer,
Daniel C. Hackett,
Gurtej Kanwar,
Alexander G. D. G. Matthews,
Sébastien Racanière,
Ali Razavi,
Danilo J. Rezende,
Fernando Romero-López,
Phiala E. Shanahan,
Julian M. Urban
Abstract:
Applications of normalizing flows to the sampling of field configurations in lattice gauge theory have so far been explored almost exclusively in two space-time dimensions. We report new algorithmic developments of gauge-equivariant flow architectures facilitating the generalization to higher-dimensional lattice geometries. Specifically, we discuss masked autoregressive transformations with tracta…
▽ More
Applications of normalizing flows to the sampling of field configurations in lattice gauge theory have so far been explored almost exclusively in two space-time dimensions. We report new algorithmic developments of gauge-equivariant flow architectures facilitating the generalization to higher-dimensional lattice geometries. Specifically, we discuss masked autoregressive transformations with tractable and unbiased Jacobian determinants, a key ingredient for scalable and asymptotically exact flow-based sampling algorithms. For concreteness, results from a proof-of-principle application to SU(3) lattice gauge theory in four space-time dimensions are reported.
△ Less
Submitted 3 May, 2023;
originally announced May 2023.
-
Aspects of scaling and scalability for flow-based sampling of lattice QCD
Authors:
Ryan Abbott,
Michael S. Albergo,
Aleksandar Botev,
Denis Boyda,
Kyle Cranmer,
Daniel C. Hackett,
Alexander G. D. G. Matthews,
Sébastien Racanière,
Ali Razavi,
Danilo J. Rezende,
Fernando Romero-López,
Phiala E. Shanahan,
Julian M. Urban
Abstract:
Recent applications of machine-learned normalizing flows to sampling in lattice field theory suggest that such methods may be able to mitigate critical slowing down and topological freezing. However, these demonstrations have been at the scale of toy models, and it remains to be determined whether they can be applied to state-of-the-art lattice quantum chromodynamics calculations. Assessing the vi…
▽ More
Recent applications of machine-learned normalizing flows to sampling in lattice field theory suggest that such methods may be able to mitigate critical slowing down and topological freezing. However, these demonstrations have been at the scale of toy models, and it remains to be determined whether they can be applied to state-of-the-art lattice quantum chromodynamics calculations. Assessing the viability of sampling algorithms for lattice field theory at scale has traditionally been accomplished using simple cost scaling laws, but as we discuss in this work, their utility is limited for flow-based approaches. We conclude that flow-based approaches to sampling are better thought of as a broad family of algorithms with different scaling properties, and that scalability must be assessed experimentally.
△ Less
Submitted 14 November, 2022;
originally announced November 2022.
-
Gauge-equivariant flow models for sampling in lattice field theories with pseudofermions
Authors:
Ryan Abbott,
Michael S. Albergo,
Denis Boyda,
Kyle Cranmer,
Daniel C. Hackett,
Gurtej Kanwar,
Sébastien Racanière,
Danilo J. Rezende,
Fernando Romero-López,
Phiala E. Shanahan,
Betsy Tian,
Julian M. Urban
Abstract:
This work presents gauge-equivariant architectures for flow-based sampling in fermionic lattice field theories using pseudofermions as stochastic estimators for the fermionic determinant. This is the default approach in state-of-the-art lattice field theory calculations, making this development critical to the practical application of flow models to theories such as QCD. Methods by which flow-base…
▽ More
This work presents gauge-equivariant architectures for flow-based sampling in fermionic lattice field theories using pseudofermions as stochastic estimators for the fermionic determinant. This is the default approach in state-of-the-art lattice field theory calculations, making this development critical to the practical application of flow models to theories such as QCD. Methods by which flow-based sampling approaches can be improved via standard techniques such as even/odd preconditioning and the Hasenbusch factorization are also outlined. Numerical demonstrations in two-dimensional U(1) and SU(3) gauge theories with $N_f=2$ flavors of fermions are provided.
△ Less
Submitted 16 October, 2022; v1 submitted 18 July, 2022;
originally announced July 2022.
-
Flow-based sampling in the lattice Schwinger model at criticality
Authors:
Michael S. Albergo,
Denis Boyda,
Kyle Cranmer,
Daniel C. Hackett,
Gurtej Kanwar,
Sébastien Racanière,
Danilo J. Rezende,
Fernando Romero-López,
Phiala E. Shanahan,
Julian M. Urban
Abstract:
Recent results suggest that flow-based algorithms may provide efficient sampling of field distributions for lattice field theory applications, such as studies of quantum chromodynamics and the Schwinger model. In this work, we provide a numerical demonstration of robust flow-based sampling in the Schwinger model at the critical value of the fermion mass. In contrast, at the same parameters, conven…
▽ More
Recent results suggest that flow-based algorithms may provide efficient sampling of field distributions for lattice field theory applications, such as studies of quantum chromodynamics and the Schwinger model. In this work, we provide a numerical demonstration of robust flow-based sampling in the Schwinger model at the critical value of the fermion mass. In contrast, at the same parameters, conventional methods fail to sample all parts of configuration space, leading to severely underestimated uncertainties.
△ Less
Submitted 23 February, 2022;
originally announced February 2022.
