Current browse context:
stat
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Deep symbolic regression: Recovering mathematical expressions from data via risk-seeking policy gradients
(Submitted on 10 Dec 2019 (v1), last revised 5 Apr 2021 (this version, v4))
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.
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.