Symbolic regression as a feature engineering method for machine and deep learning regression tasks

Assaf Shmuel, Oren Glickman, Teddy Lazebnik

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

In the realm of machine and deep learning (DL) regression tasks, the role of effective feature engineering (FE) is pivotal in enhancing model performance. Traditional approaches of FE often rely on domain expertise to manually design features for machine learning (ML) models. In the context of DL models, the FE is embedded in the neural network’s architecture, making it hard for interpretation. In this study, we propose to integrate symbolic regression (SR) as an FE process before a ML model to improve its performance. We show, through extensive experimentation on synthetic and 21 real-world datasets, that the incorporation of SR-derived features significantly enhances the predictive capabilities of both machine and DL regression models with 34%-86% root mean square error (RMSE) improvement in synthetic datasets and 4%-11.5% improvement in real-world datasets. In an additional realistic use case, we show the proposed method improves the ML performance in predicting superconducting critical temperatures based on Eliashberg theory by more than 20% in terms of RMSE. These results outline the potential of SR as an FE component in data-driven models, improving them in terms of performance and interpretability.

Original languageEnglish
Article number025065
JournalMachine Learning: Science and Technology
Volume5
Issue number2
DOIs
StatePublished - 1 Jun 2024

Keywords

  • data science
  • data-driven physics
  • feature engineering
  • neural network
  • symbolic regression

Fingerprint

Dive into the research topics of 'Symbolic regression as a feature engineering method for machine and deep learning regression tasks'. Together they form a unique fingerprint.

Cite this