cs.LGSep 17, 2026

MOSAIC-SR: Transformer-Guided Symbolic Regression for Scientific Equation Recovery

Authors: Peiyi Zheng, Yanming Kang, Hans De Sterck, Giang Tran

Organizations: University of Waterloo

Abstract

Symbolic regression aims to recover closed-form equations from observations, providing interpretable models for scientific discovery. Existing approaches struggle to combine flexible structural search with efficient inference. Search-based methods can refine expression structure but often rely on costly combinatorial optimization with random initialization. Pretrained neural models generate formulas almost instantly, but their predictions often contain symbolic errors. We introduce MOSAIC-SR, which uses a pretrained Transformer to propose multiple initial sketches. These sketches initialize searches in several promising regions, avoiding random starts in the vast expression space. Each search jointly recovers structure and constants through scale-aware constant optimization and local symbolic repair. We evaluate MOSAIC-SR on the SRSD-Feynman dataset with and without dummy variables and on six additional benchmarks. MOSAIC-SR obtains the highest symbolic solution rate on every dataset while ranking among the top two methods in predictive accuracy. This advantage persists in the presence of irrelevant dummy inputs. The results show that learned priors can focus search on promising equation structures, and that numerical optimization and symbolic repair are important for recovery.

Figures & tables

Appendix figures & tables39 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. FePySR: A Neural Feature Extraction Framework for Efficient and Scalable Symbolic Regression

    May 12, 2026Zhiming Yu, Wangtao Lu, Xin LaiNonlinear System IdentificationAutomated Feature Engineering

  2. Language models guide symbolic equation discovery by controlling search

    Jul 5, 2026Zikai Xie, Wenmei Li, Man Luo +2Large Language Model-Guided Scientific DiscoverySymbolic Regression

  3. AutoSR: Automatic Symbolic Regression by Searching Research States

    Aug 17, 2026Kejia Zhang, Youran Sun, Xinyu Ren +2Symbolic RegressionAutonomous Scientific Discovery