A flexible model-free prediction-based framework for feature ranking.

Jingyi Jessica Li, Yiling Elaine Chen, Xin Tong
Author Information
  1. Jingyi Jessica Li: Department of Statistics, University of California, Los Angeles.
  2. Yiling Elaine Chen: Department of Statistics, University of California, Los Angeles.
  3. Xin Tong: Department of Data Sciences and Operations, Marshall Business School, University of Southern California.

Abstract

Despite the availability of numerous statistical and machine learning tools for joint feature modeling, many scientists investigate features marginally, i.e., one feature at a time. This is partly due to training and convention but also roots in scientists' strong interests in simple visualization and interpretability. As such, marginal feature ranking for some predictive tasks, e.g., prediction of cancer driver genes, is widely practiced in the process of scientific discoveries. In this work, we focus on marginal ranking for binary classification, one of the most common predictive tasks. We argue that the most widely used marginal ranking criteria, including the Pearson correlation, the two-sample test, and two-sample Wilcoxon rank-sum test, do not fully take feature distributions and prediction objectives into account. To address this gap in practice, we propose two ranking criteria corresponding to two prediction objectives: the classical criterion (CC) and the Neyman-Pearson criterion (NPC), both of which use model-free nonparametric implementations to accommodate diverse feature distributions. Theoretically, we show that under regularity conditions, both criteria achieve sample-level ranking that is consistent with their population-level counterpart with high probability. Moreover, NPC is robust to sampling bias when the two class proportions in a sample deviate from those in the population. This property endows NPC good potential in biomedical research where sampling biases are ubiquitous. We demonstrate the use and relative advantages of CC and NPC in simulation and real data studies. Our model-free objective-based ranking idea is extendable to ranking feature subsets and generalizable to other prediction tasks and learning objectives.

Keywords

References

  1. Int J Oncol. 2016 Sep;49(3):1046-56 [PMID: 27573352]
  2. Genome Biol. 2014;15(8):435 [PMID: 25146004]
  3. Nucleic Acids Res. 2002 Jan 1;30(1):207-10 [PMID: 11752295]
  4. Sci Adv. 2018 Feb 02;4(2):eaao1659 [PMID: 29423442]
  5. Epigenetics. 2014 Aug;9(8):1149-62 [PMID: 25089541]
  6. Cell. 2013 Nov 7;155(4):948-62 [PMID: 24183448]
  7. J Natl Cancer Inst. 2020 Mar 1;112(3):247-255 [PMID: 31161221]
  8. Ann Appl Stat. 2009 Jan 1;3(4):1266-1269 [PMID: 20574547]
  9. Int J Med Sci. 2018 Jan 1;15(1):46-58 [PMID: 29333087]
  10. Nucleic Acids Res. 2019 Jan 8;47(D1):D1005-D1012 [PMID: 30445434]
  11. Sci Adv. 2020 Nov 11;6(46): [PMID: 33177077]

Grants

  1. R01 GM120507/NIGMS NIH HHS

Word Cloud

Created with Highcharts 10.0.0rankingfeaturemarginalpredictionNPCmodel-freetaskscriteriatwosamplinglearningeonepredictivewidelybinaryclassificationtwo-sampletestdistributionsobjectivesclassicalcriterionCCNeyman-PearsonusebiasDespiteavailabilitynumerousstatisticalmachinetoolsjointmodelingmanyscientistsinvestigatefeaturesmarginallyitimepartlyduetrainingconventionalsorootsscientists'stronginterestssimplevisualizationinterpretabilitygcancerdrivergenespracticedprocessscientificdiscoveriesworkfocuscommonargueusedincludingPearsoncorrelationWilcoxonrank-sumfullytakeaccountaddressgappracticeproposecorrespondingobjectives:nonparametricimplementationsaccommodatediverseTheoreticallyshowregularityconditionsachievesample-levelconsistentpopulation-levelcounterparthighprobabilityMoreoverrobustclassproportionssampledeviatepopulationpropertyendowsgoodpotentialbiomedicalresearchbiasesubiquitousdemonstraterelativeadvantagessimulationrealdatastudiesobjective-basedideaextendablesubsetsgeneralizableflexibleprediction-basedframeworkparadigms

Similar Articles

Cited By