Nonparametric IPSS: Fast, flexible feature selection with false discovery control

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Melikechi, Omar, Dunson, David B., Miller, Jeffrey W.
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918095352758272
author Melikechi, Omar
Dunson, David B.
Miller, Jeffrey W.
author_facet Melikechi, Omar
Dunson, David B.
Miller, Jeffrey W.
contents Feature selection is a critical task in machine learning and statistics. However, existing feature selection methods either (i) rely on parametric methods such as linear or generalized linear models, (ii) lack theoretical false discovery control, or (iii) identify few true positives. Here, we introduce a general feature selection method with finite-sample false discovery control based on applying integrated path stability selection (IPSS) to arbitrary feature importance scores. The method is nonparametric whenever the importance scores are nonparametric, and it estimates q-values, which are better suited to high-dimensional data than p-values. We focus on two special cases using importance scores from gradient boosting (IPSSGB) and random forests (IPSSRF). Extensive nonlinear simulations with RNA sequencing data show that both methods accurately control the false discovery rate and detect more true positives than existing methods. Both methods are also efficient, running in under 20 seconds when there are 500 samples and 5000 features. We apply IPSSGB and IPSSRF to detect microRNAs and genes related to cancer, finding that they yield better predictions with fewer features than existing approaches.
format Preprint
id arxiv_https___arxiv_org_abs_2410_02208
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Nonparametric IPSS: Fast, flexible feature selection with false discovery control
Melikechi, Omar
Dunson, David B.
Miller, Jeffrey W.
Machine Learning
Applications
Methodology
Feature selection is a critical task in machine learning and statistics. However, existing feature selection methods either (i) rely on parametric methods such as linear or generalized linear models, (ii) lack theoretical false discovery control, or (iii) identify few true positives. Here, we introduce a general feature selection method with finite-sample false discovery control based on applying integrated path stability selection (IPSS) to arbitrary feature importance scores. The method is nonparametric whenever the importance scores are nonparametric, and it estimates q-values, which are better suited to high-dimensional data than p-values. We focus on two special cases using importance scores from gradient boosting (IPSSGB) and random forests (IPSSRF). Extensive nonlinear simulations with RNA sequencing data show that both methods accurately control the false discovery rate and detect more true positives than existing methods. Both methods are also efficient, running in under 20 seconds when there are 500 samples and 5000 features. We apply IPSSGB and IPSSRF to detect microRNAs and genes related to cancer, finding that they yield better predictions with fewer features than existing approaches.
title Nonparametric IPSS: Fast, flexible feature selection with false discovery control
topic Machine Learning
Applications
Methodology
url https://arxiv.org/abs/2410.02208