Path Integral Optimiser: Global Optimisation via Neural Schrödinger-Föllmer Diffusion
By: Max McGuinness, Eirik Fladmark, Francisco Vargas
Potential Business Impact:
Finds best answers to hard problems faster.
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.
Similar Papers
Proximal Diffusion Neural Sampler
Machine Learning (CS)
Helps computers draw complex pictures by learning step-by-step.
Neural Network Approach to Stochastic Dynamics for Smooth Multimodal Density Estimation
Machine Learning (CS)
Helps computers learn faster from complex data.
Sampling conditioned diffusions via Pathspace Projected Monte Carlo
Machine Learning (Stat)
Guides random paths to follow specific rules.