Organizations: C2DL, Institute of Automation, Chinese Academy of Sciences · School of Artificial Intelligence, University of Chinese Academy of Sciences · State Key Laboratory of Aerodynamics · University of Chinese Academy of Sciences, Nanjing · AiRiA
Symbolic regression (SR) aims to discover compact and meaningful mathematical equations from data, but searching the vast combinatorial space of symbolic structures remains challenging. Existing methods typically guide this process using expression-level objectives, such as fitting error, which assess a candidate equation as a whole but provide little information about whether an individual term contributes genuinely new structure or is largely redundant with the rest of the expression. We introduce Sobolev Novelty, a term-level measure of structural independence for symbolic equations. For each term, we construct an empirical Sobolev signature from its function values and exact derivatives over the observed inputs, and quantify how much of this behavior cannot be reconstructed by the remaining terms. We further derive a theory-calibrated threshold, yielding a principled and tuning-free criterion for identifying structurally novel terms. Using this threshold, 92.6% of terms in benchmark ground-truth equations exhibit sufficient structural novelty, compared with only 38.3% on average for expressions produced by 15 SR methods, revealing a substantial gap between scientific equations and current SR solutions. As a lightweight plug-in, Sobolev Novelty can be incorporated into diverse SR paradigms to support term pruning, search guidance, LLM feedback, and data selection, yielding consistent performance gains and demonstrating broad applicability.
Figures & tables
Figure 1: Mean complexity and term-level Sobolev Novelty across ground truth and SR methods.
Figure 2: Accuracy–complexity rank comparisons on the SRBench white-box and black-box benchmarks. Lower is better on both axes. Dashed lines indicate Pareto tiers, and arrows connect each backbone with its Sobolev-guided variant. Accuracy ranks are computed from mean R2 , and complexity ranks from mean expression complexity.
Figure 3: Generalization performance of E2ESR under different pretraining formula distributions. Endpoint R2 values are annotated on the right.
Figure 4
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
Method
Multi-term Expressions
Total Terms
Qualified Terms
Qualification Rate (%)
Ground Truth
41
94
87
92.6
AFP
451
1,496
1,039
69.5
AFP-FE
563
1,814
1,295
71.4
AIFeynman2
388
6,130
448
7.3
BSR
524
4,335
1,657
38.2
DSR
339
990
677
68.4
Appendix
Table 2: Per-method term qualification statistics at the theory-calibrated threshold τid=1/10 . A term is qualified when νi>τid .
Type
Algorithm
Mean R2↑
Complexity ↓
SN ↑
Regression
FEAT
0.9192
195.300
0.1686
Generative
E2ESR
0.8194
89.630
0.3675
NeurSR
0.4090
31.420
0.4348
Search
AFP
0.9553
37.010
0.8185
AFP-FE
0.9767
40.630
0.8197
AIFeynman2
0.9000
113.200
0.7781
Appendix
Table 3: White-box results under noise-free conditions.
Type
Algorithm
Mean R2↑
Complexity ↓
SN ↑
Regression
FEAT
0.7621
82.489
0.6555
Generative
E2ESR
0.3612
61.094
0.6331
NeurSR
0.1228
13.333
0.6590
Search
AFP
0.6333
34.894
0.7094
AFP-FE
0.6400
36.039
0.7078
AIFeynman2
0.2110
2239.752
0.4286
Appendix
Table 4: Black-box results.
Figure 5: Mean test NMSE (left group, left axis) and mean SN calls per run (right group, right axis) under the same time budget.
Symbolic Regression (SR) plays a central role in scientific knowledge discovery by distilling mathematical equations from observational data. Most existing SR methods function within a bi-level optimization framework: an outer loop that searches for the discrete equation structure, and an inner loop that optimizes the continuous parameters of that structure. Crucially, parameter-fitting quality directly determines a structure's score and thus the outer-loop search. However, nonlinear operators make the inner loop highly non-convex, and budget-driven reliance on fast local solvers (e.g., BFGS) often yields poor local minima and underestimated scores for correct structures. This ``Good Structure, Bad Score'' phenomenon becomes a key bottleneck, degrading efficiency and misguiding the search away from the true equation. To resolve this, we propose SAGE-Fit (Structure-Aware and Semantics-Guided Evaluator for Symbolic Regression), an SR-native fitting framework that exploits the dual native priors of symbolic expressions. By capitalizing on the structural and semantic priors unique to SR, we design tailored modules for each property, thereby effectively mitigating this optimization bottleneck. Extensive experiments demonstrate that our approach, as a plug-and-play module, significantly enhances evaluation fidelity and universally improves the performance of various SR systems.
Boxiao Wang, Kai Li, Zhiwei Chen +5
Institute of Automation, Chinese Academy of Sciences
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.
A fundamental challenge in symbolic regression (SR) is efficiently recovering complex mathematical expressions from observational data. Although this problem is NP-hard, many expressions of practical interest decompose naturally into combinations of nonlinear feature modules, concentrating structural complexity into a small number of reusable components. Here, we introduce FePySR, a two-stage framework that reduces the SR search space by extracting valid features prior to equation search. FePySR first employs a heterogeneous neural network to constrain observational data to a set of candidate expressions, then performs structural optimization within this refined expression space using PySR. Across five standard benchmarks, FePySR outperforms state-of-the-art methods by achieving higher equation recovery rates. On a set of 75 highly complex synthesized equations, FePySR recovers 36 equations, while producing substantially smaller mean squared errors on the remaining unrecovered cases, with reduced computation time compared to PySR. FePySR's first stage also maintains consistent performance under varying numbers of selected top features and increasing levels of noise in the observational data. Applied to ordinary differential equations governing biological systems, FePySR successfully identifies governing equations in 24 out of 100 tests where PySR recovers none. Taken together, FePySR is a generalizable framework that can enhance the SR solvers, enabling the efficient and reliable recovery of symbolic expressions across scientific domains.
Zhiming Yu, Wangtao Lu, Xin Lai
School of Mathematical Sciences, Zhejiang University, Hangzhou, Zhejiang, China · Faculty of Medicine and Health Technology, Tampere University, Tampere, Finland