Grammar-Constrained LLM Generation for Reliable and Efficient Probabilistic Program Synthesis
Probabilistic programming languages (PPLs) provide a principled framework for expressing Bayesian models and automating inference, yet writing correct programs remains challenging. Large language models (LLMs) can accelerate this process by drafting candidate programs, but their outputs frequently fail due to semantic and syntactic errors. We introduce RefineStat, a framework for reliable probabilistic program synthesis in PyMC using compact, open-weight models. By enforcing statistical semantics during generation and applying targeted refinements guided by Bayesian diagnostics, RefineStat produces programs that are both syntactically sound and statistically reliable. Across different datasets and multiple small open source models, RefineStat consistently improves run rate, convergence behavior, and predictive stability compared to unconstrained or syntax-only baselines, while remaining cost-efficient.
Sun 11 JanDisplayed time zone: Brussels, Copenhagen, Madrid, Paris change
16:00 - 18:00 | |||
16:00 10mTalk | Multi-Agent Systems for Traceable Bayesian Workflow LAFI Xianda Sun University of Cambridge, Andrew D. Gordon Cogna and University of Edinburgh, Hong Ge University of Cambridge File Attached | ||
16:12 10mTalk | Grammar-Constrained LLM Generation for Reliable and Efficient Probabilistic Program Synthesis LAFI Madhav Kanda University of Illinois Urbana-Champaign, Shubham Ugare Meta, Sasa Misailovic University of Illinois at Urbana-Champaign | ||
16:24 10mTalk | Language-Model Probabilistic Programming for Improving Autoformalization via Cycle Consistency and Incremental Type-Checking LAFI Mauricio Barba da Costa MIT, Fabian Zaiser MIT, Katherine Collins MIT, Romir Patel MIT, Timothy O'Donnell , Alexander K. Lew Yale University, Joshua B. Tenenbaum Massachusetts Institute of Technology, Vikash K. Mansinghka Massachusetts Institute of Technology, Cameron Freer Massachusetts Institute of Technology | ||
16:35 80mPoster | Poster Session LAFI | ||