Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper addresses the source distribution estimation (SDE) problem in simulation-based science by employing an expectation maximization approach that iteratively refines the source distribution using fresh simulations. Unlike traditional methods that rely on a fixed likelihood surrogate, the proposed method trains an amortized posterior during the E-step and updates the source distribution in the M-step, effectively adapting to inaccuracies in parameter space. The evaluation on benchmark tasks, particularly the Lotka-Volterra model, demonstrates significant improvements over existing methods, achieving data-space C2ST scores as low as 0.64-0.68 compared to 0.96 for baseline approaches.
Expectation maximization for source distribution estimation outperforms traditional fixed surrogate methods, achieving unprecedented accuracy in challenging simulation tasks.
Simulation-based science often requires a distribution over simulator parameters whose push-forward reproduces a set of real observations: this is the source distribution estimation (SDE) problem. Existing methods fit the source against a likelihood surrogate trained once from a fixed proposal prior. Their objective is therefore stated only in terms of the surrogate instead of the true simulator, which may fail for inaccurate areas in parameter space where the surrogate was never trained. We instead solve SDE by expectation maximization: an E-step trains an amortized posterior on fresh simulations from the current source estimate, and an M-step refits the source to the average of that posterior over the observed data. We give two parameterizations, (1) separate source and posterior flows and (2) a single shared conditional flow. We evaluate our method on three benchmark tasks under both broad and misspecified initial priors. Both improve on existing fixed surrogate approaches and on iterated variants of each, most clearly on Lotka--Volterra, where no baseline falls below 0.96 data-space C2ST while our methods reach 0.64-0.68 in three of four initial-prior settings.