vix.ing · top · new · best · stats · spec

Gradient estimators for normalising flows

2022/02/02 by P. Białas, Bialas, Piotr, Piotr Korcyl +3
Computer Science · #Adversarial Robustness in Machine Learning #Explainable Artificial Intelligence (XAI) #FOS: Computer and information sciences #FOS: Physical sciences #Generative Adversarial Networks and Image Synthesis #High Energy Physics - Lattice (hep-lat) #Machine Learning (cs.LG) #Machine Learning (stat.ML) #Statistical Mechanics (cond-mat.stat-mech)

paper · pdf · doi:10.48550/arxiv.2202.01314

openalex publication_date 2022/02/02 · openalex created_date 2025/10/10 · openalex updated_date 2026/08/05

Abstract

Recently a machine learning approach to Monte-Carlo simulations called Neural Markov Chain Monte-Carlo (NMCMC) is gaining traction. In its most popular form it uses neural networks to construct normalizing flows which are then trained to approximate the desired target distribution. In this contribution we present new gradient estimator for Stochastic Gradient Descent algorithm (and the corresponding PyTorch implementation) and show that it leads to better training results for ϕ4 model. For this model our estimator achieves the same precision in approximately half of the time needed in standard approach and ultimately provides better estimates of the free energy. We attribute this effect to the lower variance of the new estimator. In contrary to the standard learning algorithm our approach does not require estimation of the action gradient with respect to the fields, thus has potential of further speeding up the training for models with more complicated actions.

Related