Skip to main content
bioRxiv
  • Home
  • About
  • Submit
  • ALERTS / RSS
Advanced Search
New Results

binomialRF: Scalable Feature Selection and Screening for Random Forests to Identify Biomarkers and Their Interactions

Samir Rachid Zaim, Colleen Kenost, Yves A. Lussier, Helen H. Zhang
doi: https://doi.org/10.1101/681973
Samir Rachid Zaim
1The Center for Biomedical Informatics & Biostatistics of the University of Arizona Health Sciences, 1230 N. Cherry Ave, Tucson, AZ, 85721, USA
2The Department of Medicine, College of Medicine, Tucson, 1501 N. Campbell Ave, Tucson, AZ, 85721, USA
3The Graduate Interdisciplinary Program in Statistics, 617 N. Santa Rita Ave. The University of Arizona, Tucson, AZ 85721, USA
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
Colleen Kenost
1The Center for Biomedical Informatics & Biostatistics of the University of Arizona Health Sciences, 1230 N. Cherry Ave, Tucson, AZ, 85721, USA
2The Department of Medicine, College of Medicine, Tucson, 1501 N. Campbell Ave, Tucson, AZ, 85721, USA
3The Graduate Interdisciplinary Program in Statistics, 617 N. Santa Rita Ave. The University of Arizona, Tucson, AZ 85721, USA
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
Yves A. Lussier
1The Center for Biomedical Informatics & Biostatistics of the University of Arizona Health Sciences, 1230 N. Cherry Ave, Tucson, AZ, 85721, USA
2The Department of Medicine, College of Medicine, Tucson, 1501 N. Campbell Ave, Tucson, AZ, 85721, USA
3The Graduate Interdisciplinary Program in Statistics, 617 N. Santa Rita Ave. The University of Arizona, Tucson, AZ 85721, USA
4The Center for Applied Genetic and Genomic Medicine, 1295 N. Martin, Tucson, AZ 85721, USA
5The University of Arizona Cancer Center, 3838 N. Campbell Ave, Tucson, AZ 85721, USA
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
  • For correspondence: yves@email.arizona.edu hzhang@math.arizona.edu
Helen H. Zhang
6The Department of Mathematics, College of Sciences, 617 N. Santa Rita Ave. The University of Arizona, Tucson, AZ 85721, USA
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
  • For correspondence: yves@email.arizona.edu hzhang@math.arizona.edu
  • Abstract
  • Full Text
  • Info/History
  • Metrics
  • Preview PDF
Loading

Abstract

Background Feature selection is becoming increasingly important in machine learning as users want more interpretation and post-hoc analyses. However, existing feature selection techniques in random forests are computationally intensive and high RAM memory requirements requiring specialized (uncommon) high throughput infrastructures. In bioinformatics, random forest classifiers are widely used as it is a flexible, self-regulating and self-contained machine learning algorithm that is robust to the “predictors(features) P ≫ subjects N” problem with minimal tuning parameters. The current feature selection options are used extensively for biomarker detection and discovery; however, they are limited to variants of permutation tests or heuristic rankings with arbitrary cutoffs. In this work, we propose a novel paradigm using the binomial framework for feature selection in random forests, binomialRF, which is designed to produce significance measures devoid of expensive and uninterpretable permutation tests. Furthermore, it offers a highly flexible framework for efficiently identifying and ranking multi-way interactions via dynamic tree programming.

Methods We propose a novel and scalable feature selection technique that exploits the tree structure in the random forest, treats each tree as a binomial stochastic process, and determines feature significance by conducting a one-sided binomial exact test to determine if a feature was detected more often than expected by random chance. Since each tree is an independent and identically distributed random sample in a binomial process (from the perspective of choosing a splitting variable in the root node), the test statistic is constructed based on the frequency of a feature being selected (each tree is a Bernoulli trial for selecting a feature), and then the features are ranked based on the observed test statistics, its resulting nominal p-values, and its multiplicity-adjusted q-values. Furthermore, the binomialRF framework provides a general selection framework to identify 2-way, 3-way, and K-way interactions by generalizing the test statistic to count sub-trees in the random forest using dynamic tree programming.

Results In simulation studies, the binomialRF algorithm performs competitively with respect to the state of the art in terms of classification accuracy, true model coverage, and controlling for false selection in identifying main effects while attaining substantial computational performance gains (between 30 to 600 times faster in high dimensional settings than the state of the art). In addition, extending the binomialRF using model averaging identified the true model on average with greater accuracy (>20% improvement in reducing false positive feature selection at high dimensions while maximizing true model coverage) and attained greater classification accuracy (between 4-9% improvement across all techniques) without sacrificing computational speed (2nd fastest performance after binomialRF). In addition, the framework easily scales and extends to identifying 2-way and K-way interactions (i) without additional memory requirements (only requires storing original predictor matrix), and (ii) with minimal additional computational complexity cost due to efficient dynamic tree programming interaction searches. The algorithm was validated in a case study to predict bronchospasm-related hospitalization from blood transcriptomes where the binomialRF algorithm correctly identified the previously published relevant physiological pathways, presented comparable classification accuracy in a validation set, and extended previous work in this area by looking at pathway-pathway interaction.

Conclusion The proposed binomialRF proposes a novel and efficient feature selection method devoid of permutation tests – that scales linearly in the number of trees, with minimal computational complexity, thus outperforming alternate conventional methods from a computational perspective while attaining competitive model selection and classification accuracies and enabling computations on common of cost-effective high throughput infrastructures. Furthermore, the binomialRF model averaging framework greatly improves the accuracy of the feature predictions, controlling for false selection and substantially improving model and classification accuracy. Validated in numerical studies and retrospectively in a clinical trial (case study), the binomialRF paradigm offers a binomial framework to detect feature significance and easily extends to search for K-way interactions in a linear fashion, reducing a known non-polynomial time exploration to linear approximations.

The binomialRF R package is freely available on GitHub and has been submitted to BioConductor, with all associated documentations and help files.

Github: https://github.com/SamirRachidZaim/binomialRF

BioConductor: binomialRF

  • 10 List of acronyms

    RF
    random forest
    BH
    Benjamini Hochberg adjustment
    BY
    Benjamini Yekutieli adjustment
    BMA
    Bayesian Model Averaging
    SOIL
    Sparsity Oriented Importance Learning
    HRV
    Human Rhinovirus
    FSR
    False Selection Rate
    LASSO
    Least Absolute Shrinkage and Selection Operator
    RAMP
    regularization algorithm under marginality principle
    DTP
    Dynamic Tree Programming
    GO
    Gene Ontology
    GO-BP
    Gene Ontology Biological Processes
  • Copyright 
    The copyright holder for this preprint is the author/funder, who has granted bioRxiv a license to display the preprint in perpetuity. It is made available under a CC-BY-NC 4.0 International license.
    Back to top
    PreviousNext
    Posted June 26, 2019.
    Download PDF
    Email

    Thank you for your interest in spreading the word about bioRxiv.

    NOTE: Your email address is requested solely to identify you as the sender of this article.

    Enter multiple addresses on separate lines or separate them with commas.
    binomialRF: Scalable Feature Selection and Screening for Random Forests to Identify Biomarkers and Their Interactions
    (Your Name) has forwarded a page to you from bioRxiv
    (Your Name) thought you would like to see this page from the bioRxiv website.
    CAPTCHA
    This question is for testing whether or not you are a human visitor and to prevent automated spam submissions.
    Share
    binomialRF: Scalable Feature Selection and Screening for Random Forests to Identify Biomarkers and Their Interactions
    Samir Rachid Zaim, Colleen Kenost, Yves A. Lussier, Helen H. Zhang
    bioRxiv 681973; doi: https://doi.org/10.1101/681973
    Digg logo Reddit logo Twitter logo Facebook logo Google logo LinkedIn logo Mendeley logo
    Citation Tools
    binomialRF: Scalable Feature Selection and Screening for Random Forests to Identify Biomarkers and Their Interactions
    Samir Rachid Zaim, Colleen Kenost, Yves A. Lussier, Helen H. Zhang
    bioRxiv 681973; doi: https://doi.org/10.1101/681973

    Citation Manager Formats

    • BibTeX
    • Bookends
    • EasyBib
    • EndNote (tagged)
    • EndNote 8 (xml)
    • Medlars
    • Mendeley
    • Papers
    • RefWorks Tagged
    • Ref Manager
    • RIS
    • Zotero
    • Tweet Widget
    • Facebook Like
    • Google Plus One

    Subject Area

    • Bioinformatics
    Subject Areas
    All Articles
    • Animal Behavior and Cognition (3590)
    • Biochemistry (7562)
    • Bioengineering (5503)
    • Bioinformatics (20753)
    • Biophysics (10308)
    • Cancer Biology (7964)
    • Cell Biology (11625)
    • Clinical Trials (138)
    • Developmental Biology (6598)
    • Ecology (10177)
    • Epidemiology (2065)
    • Evolutionary Biology (13589)
    • Genetics (9530)
    • Genomics (12830)
    • Immunology (7917)
    • Microbiology (19525)
    • Molecular Biology (7651)
    • Neuroscience (42025)
    • Paleontology (307)
    • Pathology (1254)
    • Pharmacology and Toxicology (2195)
    • Physiology (3261)
    • Plant Biology (7028)
    • Scientific Communication and Education (1294)
    • Synthetic Biology (1949)
    • Systems Biology (5422)
    • Zoology (1113)