We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Deep symbolic regression: Recovering mathematical expressions from data via risk-seeking policy gradients

Abstract: Discovering the underlying mathematical expressions describing a dataset is a core challenge for artificial intelligence. This is the problem of $\textit{symbolic regression}$. Despite recent advances in training neural networks to solve complex tasks, deep learning approaches to symbolic regression are underexplored. We propose a framework that leverages deep learning for symbolic regression via a simple idea: use a large model to search the space of small models. Specifically, we use a recurrent neural network to emit a distribution over tractable mathematical expressions and employ a novel risk-seeking policy gradient to train the network to generate better-fitting expressions. Our algorithm outperforms several baseline methods (including Eureqa, the gold standard for symbolic regression) in its ability to exactly recover symbolic expressions on a series of benchmark problems, both with and without added noise. More broadly, our contributions include a framework that can be applied to optimize hierarchical, variable-length objects under a black-box performance metric, with the ability to incorporate constraints in situ, and a risk-seeking policy gradient formulation that optimizes for best-case performance instead of expected performance.
Comments: Published at International Conference on Learning Representations, 2021
Subjects: Machine Learning (cs.LG); Machine Learning (stat.ML)
Journal reference: International Conference on Learning Representations, 2021
Report number: LLNL-CONF-790457
Cite as: arXiv:1912.04871 [cs.LG]
  (or arXiv:1912.04871v4 [cs.LG] for this version)

Submission history

From: Brenden Petersen [view email]
[v1] Tue, 10 Dec 2019 18:25:48 GMT (510kb,D)
[v2] Fri, 28 Feb 2020 17:16:24 GMT (3486kb,D)
[v3] Wed, 31 Mar 2021 04:00:32 GMT (4087kb,D)
[v4] Mon, 5 Apr 2021 22:29:16 GMT (4087kb,D)

Link back to: arXiv, form interface, contact.