A Probabilistic Framework for LLM-Based Model Discovery
Stefan Wahl, Raphaela Schenk, Ali Farnoud, Jakob Macke, Daniel Gedon
Abstract
Automated methods for discovering mechanistic simulator models from observational data offer a promising path toward accelerating scientific progress. Such methods often take the form of agentic-style iterative workflows that repeatedly propose and revise candidate models by imitating human discovery processes. However, existing LLM-based approaches typically implement such workflows via hand-crafted heuristic procedures, without an explicit probabilistic formulation. We recast model discovery as probabilistic inference, i.e., as sampling from an unknown distribution over mechanistic models capable of explaining the data. This perspective provides a unified way to reason about model proposal, refinement, and selection within a single inference framework. As a concrete instantiation of this view, we introduce ModelSMC, an algorithm based on Sequential Monte Carlo sampling. ModelSMC represents candidate models as particles which are iteratively proposed and refined by an LLM, and weighted using likelihood-based criteria. Experiments on real-world scientific systems illustrate that this formulation discovers models with interpretable mechanisms and improves posterior predictive checks. More broadly, this perspective provides a probabilistic lens for understanding and developing LLM-based approaches to model discovery.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 479a31cc-7b91-4e95-bbfb-374ef2225b38Builds on10
- End-to-end Symbolic Regression with TransformersPierre-Alexandre Kamienny, Stéphane d'Ascoli, Guillaume Lample, François ChartonNeurIPS 2022 · 320 citations
- AutoDiscovery: Open-ended Scientific Discovery via Bayesian SurpriseDhruv Agarwal, Bodhisattwa Prasad Majumder, Reece Adamson, Megha Chakravorty et al.NeurIPS 2025 · 35 citations
- Generalized Bayesian Inference for Scientific Simulators via Amortized Cost EstimationRichard Gao, Michael Deistler, Jakob H. MackeNeurIPS 2023 · 19 citations
- Data-Driven Discovery of Dynamical Systems in Pharmacology using Large Language ModelsSamuel Holt, Zhaozhi Qian, Tennison Liu, James Weatherall et al.NeurIPS 2024 · 16 citations
- Effortless, Simulation-Efficient Bayesian Inference using Tabular Foundation ModelsJulius Vetter, Manuel Glöckler, Daniel Gedon, Jakob H. MackeNeurIPS 2025 · 14 citations
Related papers
- Discovering Ordinary Differential Equations with LLM-Based Qualitative and Quantitative EvaluationSum Kyun Song, Bong Gyun Shin, JaeYong LeeICML 2026
- Automated Statistical Model Discovery with Language ModelsMichael Y. Li, Emily B. Fox, Noah D. GoodmanICML 2024 · 36 citations
- Agentic Monte Carlo: Simulating Reinforcement Learning for Black-Box AgentsDaeyon Hwang, Raunaq Suri, Valentin Villecroze, Anthony Caterini et al.ICML 2026
- LLM and Simulation as Bilevel Optimizers: A New Paradigm to Advance Physical Scientific DiscoveryPingchuan Ma, Tsun-Hsuan Wang, Minghao Guo, Zhiqing Sun et al.ICML 2024 · 76 citations
- Speculative Sampling For Faster Molecular DynamicsArthur Kosmala, Stephan Günnemann, Meng Gao, Brandon WoodICML 2026
