Probabilistic Symbolic Regression for Equation Discovery via Operator-induced and Regularized Symbolic Forests
stat.ME, cs.LG, cs.SC, math.ST, stat.ML, stat.TH
Submitted: 2025-09-24
Updated: 2026-08-28
Comments: 152 pages, 37 figures, 24 tables, Submitted
Code: https://github.com/Roy-SR-007/BayeSymX
Project page: https://autoresearch.github.io/autora/user-guide/theorists/bms
License: http://creativecommons.org/licenses/by/4.0/
The gist: Symbolic regression has emerged as a powerful tool for artificial intelligence-driven scientific discovery by learning interpretable analytical expressions that reveal governing relationships
Terminology
Abstract
Symbolic regression has emerged as a powerful tool for artificial intelligence-driven scientific discovery by learning interpretable analytical expressions that reveal governing relationships directly from data. Existing methods, however, often rely on heuristic search, struggle to balance predictive accuracy with expression complexity in noisy settings, and offer limited characterization of symbolic uncertainty. Probabilistic approaches that address these challenges in a unified manner remain underexplored. We introduce a probabilistic symbolic regression framework that represents mathematical expressions as ensembles of symbolic trees. A regularizing prior over tree topology controls expression complexity, while an Occam's window-based posterior summary captures uncertainty across multiple plausible symbolic models. Given the limited existing theoretical treatment of symbolic regression, we develop posterior concentration guarantees when symbolic expressions approximate the underlying relationship arbitrarily well, with a near-parametric rate when an exact finite formula exists. Additionally, we establish a sharp oracle concentration result under symbolic misspecification. Comparisons of our proposed framework with state-of-the-art competitors demonstrate superior predictive accuracy, optimal symbolic complexity, and stable structural recovery when learning benchmark scientific equations, together with the identification of scientifically interpretable descriptor formulas in a challenging materials discovery application.
Sources
- An Approach to Symbolic Regression Using Feyn
- Interpretable Machine Learning for Science with PySR and SymbolicRegression.jl
- Bayesian Symbolic Regression
- VaSST: Variational Inference for Symbolic Regression using Soft Symbolic Trees
Related papers
- Doubly robust inference via calibration
- Bayesian Empirical Bayes: Simultaneous Inference from Probabilistic Symmetries
- Flexible Nonparametric Inference for Causal Effects under the Front-Door Model
- Deployment of AI-Assisted Interventions: Capacity Constraints and Noisy Compliance
- A Survey on Archetypal Analysis
- Dynamic Spatial Bayesian Machine Learning Model: Applications to Intergenerational Economic Mobility and Geographic Income Inequality in the United States