Flow Matching for Posterior Inference with Simulator Feedback

• URL: http://arxiv.org/abs/2410.22573v1
• Date: Tue, 29 Oct 2024 22:26:39 GMT
• Title: Flow Matching for Posterior Inference with Simulator Feedback
• Authors: Benjamin Holzschuh, Nils Thuerey,
• Abstract summary: Flow-based generative modeling is a powerful tool for solving inverse problems in physical sciences. We propose to refine flows with additional control signals based on a simulator. We demonstrate that including feedback from the simulator improves the accuracy by $53%$, making it competitive with traditional techniques.
• Score: 20.315933488318986
• License:
• Abstract: Flow-based generative modeling is a powerful tool for solving inverse problems in physical sciences that can be used for sampling and likelihood evaluation with much lower inference times than traditional methods. We propose to refine flows with additional control signals based on a simulator. Control signals can include gradients and a problem-specific cost function if the simulator is differentiable, or they can be fully learned from the simulator output. In our proposed method, we pretrain the flow network and include feedback from the simulator exclusively for finetuning, therefore requiring only a small amount of additional parameters and compute. We motivate our design choices on several benchmark problems for simulation-based inference and evaluate flow matching with simulator feedback against classical MCMC methods for modeling strong gravitational lens systems, a challenging inverse problem in astronomy. We demonstrate that including feedback from the simulator improves the accuracy by $53\%$, making it competitive with traditional techniques while being up to $67$x faster for inference.

Related papers

• QuasiSim: Parameterized Quasi-Physical Simulators for Dexterous Manipulations Transfer [21.20658409070528]
  We explore the dexterous manipulation transfer problem by designing simulators. The task wishes to transfer human manipulations to dexterous robot hand simulations. We successfully enable a dexterous hand to track complex and diverse manipulations in high-fidelity simulated environments.
  arXiv  Detail & Related papers  (2024-04-11T17:59:40Z)
• Model-based deep reinforcement learning for accelerated learning from flow simulations [0.0]
  We demonstrate the benefits of model-based reinforcement learning for flow control applications. Specifically, we optimize the policy by alternating between trajectories sampled from flow simulations and trajectories sampled from an ensemble of environment models. The model-based learning reduces the overall training time by up to $85%$ for the fluidic pinball test case.
  arXiv  Detail & Related papers  (2024-02-26T13:01:45Z)
• Simulation-free Schr\"odinger bridges via score and flow matching [89.4231207928885]
  We present simulation-free score and flow matching ([SF]$2$M) Our method generalizes both the score-matching loss used in the training of diffusion models and the recently proposed flow matching loss used in the training of continuous flows. Notably, [SF]$2$M is the first method to accurately model cell dynamics in high dimensions and can recover known gene regulatory networks simulated data.
  arXiv  Detail & Related papers  (2023-07-07T15:42:35Z)
• Continual learning autoencoder training for a particle-in-cell simulation via streaming [52.77024349608834]
  upcoming exascale era will provide a new generation of physics simulations with high resolution. These simulations will have a high resolution, which will impact the training of machine learning models since storing a high amount of simulation data on disk is nearly impossible. This work presents an approach that trains a neural network concurrently to a running simulation without data on a disk.
  arXiv  Detail & Related papers  (2022-11-09T09:55:14Z)
• Neural Posterior Estimation with Differentiable Simulators [58.720142291102135]
  We present a new method to perform Neural Posterior Estimation (NPE) with a differentiable simulator. We demonstrate how gradient information helps constrain the shape of the posterior and improves sample-efficiency.
  arXiv  Detail & Related papers  (2022-07-12T16:08:04Z)
• DiSECt: A Differentiable Simulator for Parameter Inference and Control in Robotic Cutting [71.50844437057555]
  We present DiSECt: the first differentiable simulator for cutting soft materials. The simulator augments the finite element method with a continuous contact model based on signed distance fields. We show that the simulator can be calibrated to match resultant forces and fields from a state-of-the-art commercial solver.
  arXiv  Detail & Related papers  (2022-03-19T07:27:19Z)
• Constraint-based graph network simulator [9.462808515258464]
  We present a framework for constraint-based learned simulation. We implement our method using a graph neural network as the constraint function and gradient descent as the constraint solver. Our model achieves better or comparable performance to top learned simulators.
  arXiv  Detail & Related papers  (2021-12-16T19:15:11Z)
• Likelihood-Free Inference in State-Space Models with Unknown Dynamics [71.94716503075645]
  We introduce a method for inferring and predicting latent states in state-space models where observations can only be simulated, and transition dynamics are unknown. We propose a way of doing likelihood-free inference (LFI) of states and state prediction with a limited number of simulations.
  arXiv  Detail & Related papers  (2021-11-02T12:33:42Z)
• DriveGAN: Towards a Controllable High-Quality Neural Simulation [147.6822288981004]
  We introduce a novel high-quality neural simulator referred to as DriveGAN. DriveGAN achieves controllability by disentangling different components without supervision. We train DriveGAN on multiple datasets, including 160 hours of real-world driving data.
  arXiv  Detail & Related papers  (2021-04-30T15:30:05Z)
• Simulation-efficient marginal posterior estimation with swyft: stop wasting your precious time [5.533353383316288]
  We present algorithms for nested neural likelihood-to-evidence ratio estimation and simulation reuse. Together, these algorithms enable automatic and extremely simulator efficient estimation of marginal and joint posteriors.
  arXiv  Detail & Related papers  (2020-11-27T19:00:07Z)
• DISCO: Double Likelihood-free Inference Stochastic Control [29.84276469617019]
  We propose to leverage the power of modern simulators and recent techniques in Bayesian statistics for likelihood-free inference. The posterior distribution over simulation parameters is propagated through a potentially non-analytical model of the system. Experiments show that the controller proposed attained superior performance and robustness on classical control and robotics tasks.
  arXiv  Detail & Related papers  (2020-02-18T05:29:40Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.