Path Integral Optimiser: Global Optimisation via Neural Schr\"odinger-F\"ollmer Diffusion
Abstract
We present an early investigation into the use of neural diffusion processes for global optimisation, focusing on Zhang et al.'s Path Integral Sampler. One can use the Boltzmann distribution to formulate optimization as solving a Schr\"odinger bridge sampling problem, then apply Girsanov's theorem with a simple (single-point) prior to frame it in stochastic control terms, and compute the solution's integral terms via a neural approximation (a Fourier MLP). We provide theoretical bounds for this optimiser, results on toy optimisation tasks, and a summary of the stochastic theory motivating the model. Ultimately, we found the optimiser to display promising per-step performance at optimisation tasks between 2 and 1,247 dimensions, but struggle to explore higher-dimensional spaces when faced with a 15.9k parameter model, indicating a need for work on adaptation in such environments.
Cite
@article{arxiv.2506.06815,
title = {Path Integral Optimiser: Global Optimisation via Neural Schr\"odinger-F\"ollmer Diffusion},
author = {Max McGuinness and Eirik Fladmark and Francisco Vargas},
journal= {arXiv preprint arXiv:2506.06815},
year = {2025}
}
Comments
6 pages. Presented at the OPT Workshop, NeurIPS 2024, Vancouver, CA