LGMLMar 29

AutoStan: Autonomous Bayesian Model Improvement via Predictive Feedback

arXiv:2603.2776620.11 citationsh-index: 4
AI Analysis

For Bayesian practitioners, AutoStan automates model improvement, reducing manual effort in iterative modeling.

AutoStan is a framework where a CLI coding agent autonomously builds and iteratively improves Bayesian models in Stan using NLPD and sampler diagnostics as feedback. It matches or outperforms TabPFN on a synthetic regression dataset and discovers complex models across five datasets without search algorithms or domain-specific instructions.

We present AutoStan, a framework in which a command-line interface (CLI) coding agent autonomously builds and iteratively improves Bayesian models written in Stan. The agent operates in a loop, writing a Stan model file, executing MCMC sampling, then deciding whether to keep or revert each change based on two complementary feedback signals: the negative log predictive density (NLPD) on held-out data and the sampler's own diagnostics (divergences, R-hat, effective sample size). We evaluate AutoStan on five datasets with diverse modeling structures. On a synthetic regression dataset with outliers, the agent progresses from naive linear regression to a model with Student-t robustness, nonlinear heteroscedastic structure, and an explicit contamination mixture, matching or outperforming TabPFN, a state-of-the-art black-box method, while remaining fully interpretable. Across four additional experiments, the same mechanism discovers hierarchical partial pooling, varying-slope models with correlated random effects, and a Poisson attack/defense model for soccer. No search algorithm, critic module, or domain-specific instructions are needed. This is, to our knowledge, the first demonstration that a CLI coding agent can autonomously write and iteratively improve Stan code for diverse Bayesian modeling problems.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes