Saved in:
Bibliographic Details
Main Authors: Seber, Pedro, Braatz, Richard D.
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2402.17120
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911292859613184
author Seber, Pedro
Braatz, Richard D.
author_facet Seber, Pedro
Braatz, Richard D.
contents Interpretable models can have advantages over black-box models, and interpretability is essential for the application of machine learning in critical settings, such as aviation or medicine. This article introduces the LASSO-Clip-EN (LCEN) algorithm for nonlinear, interpretable feature selection and machine learning modeling. In a wide variety of artificial and empirical datasets, LCEN constructed sparse and frequently more accurate models than other methods, including sparse, nonlinear methods, on tested datasets. LCEN was empirically observed to be robust against many issues typically present in datasets and modeling, including noise, multicollinearity, and data scarcity. As a feature selection algorithm, LCEN matched or surpassed the thresholded elastic net but was, on average, 10.3-fold faster based on our experiments. LCEN for feature selection can also rediscover multiple physical laws from empirical data. As a machine learning algorithm, when tested on processes with no known physical laws, LCEN achieved better results than many other dense and sparse methods -- including being comparable to or better than ANNs on multiple datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2402_17120
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LCEN: A Nonlinear, Interpretable Feature Selection and Machine Learning Algorithm
Seber, Pedro
Braatz, Richard D.
Machine Learning
Interpretable models can have advantages over black-box models, and interpretability is essential for the application of machine learning in critical settings, such as aviation or medicine. This article introduces the LASSO-Clip-EN (LCEN) algorithm for nonlinear, interpretable feature selection and machine learning modeling. In a wide variety of artificial and empirical datasets, LCEN constructed sparse and frequently more accurate models than other methods, including sparse, nonlinear methods, on tested datasets. LCEN was empirically observed to be robust against many issues typically present in datasets and modeling, including noise, multicollinearity, and data scarcity. As a feature selection algorithm, LCEN matched or surpassed the thresholded elastic net but was, on average, 10.3-fold faster based on our experiments. LCEN for feature selection can also rediscover multiple physical laws from empirical data. As a machine learning algorithm, when tested on processes with no known physical laws, LCEN achieved better results than many other dense and sparse methods -- including being comparable to or better than ANNs on multiple datasets.
title LCEN: A Nonlinear, Interpretable Feature Selection and Machine Learning Algorithm
topic Machine Learning
url https://arxiv.org/abs/2402.17120