Cargando…

Predictive and interpretable models via the stacked elastic net

MOTIVATION: Machine learning in the biomedical sciences should ideally provide predictive and interpretable models. When predicting outcomes from clinical or molecular features, applied researchers often want to know which features have effects, whether these effects are positive or negative and how...

Descripción completa

Detalles Bibliográficos
Autores principales: Rauschenberger, Armin, Glaab, Enrico, van de Wiel, Mark A
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Oxford University Press 2020
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8336997/
https://www.ncbi.nlm.nih.gov/pubmed/32437519
http://dx.doi.org/10.1093/bioinformatics/btaa535
_version_ 1783733419366154240
author Rauschenberger, Armin
Glaab, Enrico
van de Wiel, Mark A
author_facet Rauschenberger, Armin
Glaab, Enrico
van de Wiel, Mark A
author_sort Rauschenberger, Armin
collection PubMed
description MOTIVATION: Machine learning in the biomedical sciences should ideally provide predictive and interpretable models. When predicting outcomes from clinical or molecular features, applied researchers often want to know which features have effects, whether these effects are positive or negative and how strong these effects are. Regression analysis includes this information in the coefficients but typically renders less predictive models than more advanced machine learning techniques. RESULTS: Here, we propose an interpretable meta-learning approach for high-dimensional regression. The elastic net provides a compromise between estimating weak effects for many features and strong effects for some features. It has a mixing parameter to weight between ridge and lasso regularization. Instead of selecting one weighting by tuning, we combine multiple weightings by stacking. We do this in a way that increases predictivity without sacrificing interpretability. AVAILABILITY AND IMPLEMENTATION: The R package starnet is available on GitHub (https://github.com/rauschenberger/starnet) and CRAN (https://CRAN.R-project.org/package=starnet).
format Online
Article
Text
id pubmed-8336997
institution National Center for Biotechnology Information
language English
publishDate 2020
publisher Oxford University Press
record_format MEDLINE/PubMed
spelling pubmed-83369972021-08-09 Predictive and interpretable models via the stacked elastic net Rauschenberger, Armin Glaab, Enrico van de Wiel, Mark A Bioinformatics Original Papers MOTIVATION: Machine learning in the biomedical sciences should ideally provide predictive and interpretable models. When predicting outcomes from clinical or molecular features, applied researchers often want to know which features have effects, whether these effects are positive or negative and how strong these effects are. Regression analysis includes this information in the coefficients but typically renders less predictive models than more advanced machine learning techniques. RESULTS: Here, we propose an interpretable meta-learning approach for high-dimensional regression. The elastic net provides a compromise between estimating weak effects for many features and strong effects for some features. It has a mixing parameter to weight between ridge and lasso regularization. Instead of selecting one weighting by tuning, we combine multiple weightings by stacking. We do this in a way that increases predictivity without sacrificing interpretability. AVAILABILITY AND IMPLEMENTATION: The R package starnet is available on GitHub (https://github.com/rauschenberger/starnet) and CRAN (https://CRAN.R-project.org/package=starnet). Oxford University Press 2020-05-21 /pmc/articles/PMC8336997/ /pubmed/32437519 http://dx.doi.org/10.1093/bioinformatics/btaa535 Text en © The Author(s) 2020. Published by Oxford University Press. https://creativecommons.org/licenses/by/4.0/This is an Open Access article distributed under the terms of the Creative Commons Attribution License (http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) ), which permits unrestricted reuse, distribution, and reproduction in any medium, provided the original work is properly cited.
spellingShingle Original Papers
Rauschenberger, Armin
Glaab, Enrico
van de Wiel, Mark A
Predictive and interpretable models via the stacked elastic net
title Predictive and interpretable models via the stacked elastic net
title_full Predictive and interpretable models via the stacked elastic net
title_fullStr Predictive and interpretable models via the stacked elastic net
title_full_unstemmed Predictive and interpretable models via the stacked elastic net
title_short Predictive and interpretable models via the stacked elastic net
title_sort predictive and interpretable models via the stacked elastic net
topic Original Papers
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8336997/
https://www.ncbi.nlm.nih.gov/pubmed/32437519
http://dx.doi.org/10.1093/bioinformatics/btaa535
work_keys_str_mv AT rauschenbergerarmin predictiveandinterpretablemodelsviathestackedelasticnet
AT glaabenrico predictiveandinterpretablemodelsviathestackedelasticnet
AT vandewielmarka predictiveandinterpretablemodelsviathestackedelasticnet