Symbolic Regression using Mixed-Integer Nonlinear Optimization

Author/Creator ORCID

Date

2020-06-11

Department

Program

Citation of Original Publication

Austel, Vernon; Cornelio, Cristina; Dash, Sanjeeb; Goncalves, Joao; Horesh, Lior; Josephson, Tyler R.; Megiddo, Nimrod; Symbolic Regression using Mixed-Integer Nonlinear Optimization; Machine Learning (2020); https://arxiv.org/abs/2006.06813

Rights

This item is likely protected under Title 17 of the U.S. Copyright Law. Unless on a Creative Commons license, for uses protected by Copyright Law, contact the copyright holder or the author.

Subjects

Abstract

The Symbolic Regression (SR) problem, where the goal is to find a regression function that does not have a pre-specified form but is any function that can be composed of a list of operators, is a hard problem in machine learning, both theoretically and computationally. Genetic programming based methods, that heuristically search over a very large space of functions, are the most commonly used methods to tackle SR problems. An alternative mathematical programming approach, proposed in the last decade, is to express the optimal symbolic expression as the solution of a system of nonlinear equations over continuous and discrete variables that minimizes a certain objective, and to solve this system via a global solver for mixed-integer nonlinear programming problems. Algorithms based on the latter approach are often very slow. We propose a hybrid algorithm that combines mixed-integer nonlinear optimization with explicit enumeration and incorporates constraints from dimensional analysis. We show that our algorithm is competitive, for some synthetic data sets, with a state-of-the-art SR software and a recent physics-inspired method called AI Feynman.