stat.MEFeb 27, 2026

VaSST: Variational Inference for Symbolic Regression using Soft Symbolic Trees

Authors: Somjit RoyPritam DeyBani K. Mallick

Organizations: Department of Statistics, Texas A&M University, College Station, Texas, USA

Abstract

Symbolic regression (SR) has gained recent traction in AI-driven scientific discovery for learning closed-form physical laws. Yet existing methods are dominated by heuristic search or data-intensive approaches that often assume low-noise regimes and lack principled uncertainty quantification, while fully probabilistic SR formulations remain scarce. We introduce a scalable probabilistic framework for SR, VaSST, based on variational inference. VaSST uses soft symbolic trees, a continuous relaxation of symbolic expression trees in which discrete operator and feature assignments are replaced by probability distributions over allowable components. This transforms combinatorial symbolic search through an astronomically large expression space into efficient gradient-based optimization while preserving a coherent probabilistic interpretation. The learned soft representations induce posterior distributions over symbolic structures, enabling uncertainty quantification across plausible symbolic forms through posterior-aware symbolic model selection. On simulated experiments and the Feynman Symbolic Regression Database, VaSST achieves strong structural recovery and predictive accuracy compared to state-of-the-art competing SR methods.

Explore similar work

CardsList
  1. Deep Variational Inference Symbolic Regression

    May 1, 2026James Butterworth, Gevik Grigorian, Alejandro DiazDelaOSymbolic RegressionVariational Inference