hep-latSep 24, 2026
SaveAccurate Sampling from Diffusion Models
Organizations: Institute of Physics, NAWI Graz, University of Graz, Universitätsplatz 5, 8010 Graz, Austria
Abstract
A new proposal called DM-SMC (Diffusion Model - Sequential Monte Carlo) is investigated, which samples ensembles defined in terms of an action, using diffusion models trained on samples from the ensemble. The SMC setup allows for accurate sampling in spite of an approximate diffusion model and the finite stepsize used in the numerical solution of the stochastic process. Improved update strategies are also investigated. Results are presented for a symmetric scalar field theory in 2 dimensions near its 2nd order phase transition.
Figures & tables
Figure 1: The effective sample size over the number of total samples on a 4x4 lattice for and and various Langevin stepsizes, as indicated by the total number of steps. The first order Euler-Maruyama update is used. On the left, the learning of the model was stopped 200 learning steps. On the right, the learning was allowed to converge using learning steps, and the parameters of the model were averaged for a further 1200 steps. No resampling was used.
Figure 2: Histogram of the field average for a 4x4 lattice for 2d scalar field theory using and . Configurations are sampled using denoising (without SMC setup) with the two models in Fig. 1 and steps for the “bad” model and steps with improved update steps for the “good” model. Results from an HMC simulation are shown as well.
Figure 3: Histogram of the field average (left panel) and histogram for the action density (right panel) for a 4x4 lattice for 2d scalar field theory using and . Measurements using the DM-SMC setup use the model corresponding to the left panel of Fig. 1 , , and one resampling at the end of the denoising process. Results from an HMC simulation are shown as well.
Figure 4: The ESS is shown for an improved denoising update for various Langevin stepsizes, as indicated. In the left panel, the determinant of the Jacobian is calculated exactky. In the right panel, the denterminant of the Jacobian is approximated using 4 noise vectors with the formulas in eqs. ( 25 ) and ( 26 ).
Figure 5: The average of is shown for various stepsizes, as calculated from DM-SMC using the 2nd order update and the approximation of the Jacobian with 4 noise vectors. Extrapolation to zero stepsize, using stepsizes , and an HMC benchmark value is also shown.
Figure 6: Histogram of the field average (left panel) and histogram for the action density (right panel) for a lattice for 2d scalar field theory using and . The diffusion process uses . Denoising is done using , and systematic resampling when the ESS/ decreases below 0.5. Results from an HMC simulation are shown as well.