Search papers, labs, and topics across Lattice.
This work introduces a novel approach to probabilistic modeling of physical fields by combining flow-matching models with a potential-induced velocity to derive explicit scalar energies at all transport times. The authors demonstrate that their method allows for effective data generation, out-of-distribution detection, and inverse problem solving without the need for variational forms or additional MCMC steps, thereby simplifying the training and sampling process. Key results show that their energy functions improve sampling accuracy and detection performance compared to traditional flow ODE baselines, highlighting the synergy between data-driven and physics-based approaches.
Flow-matching models can yield explicit scalar energies that enhance data generation and out-of-distribution detection while simplifying the sampling process in PDE fields.
Probabilistic modeling of physical fields benefits from both a data-driven prior and known physical structure such as the governing equations. Energy-based models (EBMs) are a natural fit since energies compose additively, which enables augmenting physics information during inference. However, EBMs have been difficult to train and sample from due to the intractable partition function. We show in this work that flow matching models with a potential-induced velocity yield an explicit scalar energy at all transport times, whose gradient is exactly the converted learned score and which recovers the marginal negative log-density at the population optimum. The time-dependent energy functions are obtained purely from the matching regression objective on an independent linear Gaussian interpolation, without a variational form or additional MCMC steps, and the sampling retains the flow ODE. Access to the energy function from a trained model serves three roles: energy-corrected data generation, energy as a scoring function for out-of-distribution (OOD) detection, and energy compositional posterior sampling for inverse problems. In particular, we show the explicit energy permits general MCMC samplers in the predictor-corrector sampling framework, reducing PDE residual and spectral distance compared to the flow ODE baseline. Furthermore, we demonstrate utilizing the data energy and physics-based energy (e.g., PDE residuals) as complementary mechanisms to improve detection accuracy for OOD tasks. In addition, we explore the connection to MCMC-based inference for inverse problems by composing the energy with a quadratic observational likelihood that yields a posterior energy, used as an explicitly chosen family of inference-time targets.
