Patient risk prediction model via top-κ stability selection

Jiayu Zhou, Jimeng Sun, Yashu Liu, Jianying Hu, Jieping Ye

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

The patient risk prediction model aims at assessing the risk of a patient in developing a target disease based on his/her health profile. As electronic health records (EHRs) become more prevalent, a large number of features can be constructed in order to characterize patient profiles. This wealth of data provides unprecedented opportunities for data mining researchers to address important biomedical questions. Practical data mining challenges include: How to correctly select and rank those features based on their prediction power? What predictive model performs the best in predicting a target disease using those features? In this paper, we propose top-κ stability selection, which generalizes a powerful sparse learning method for feature selection by overcoming its limitation on parameter selection. In particular, our proposed top-κ stability selection includes the original stability selection method as a special case given κ = 1. Moreover, we show that the top-κ stability selection is more robust by utilizing more information from selection probabilities than the original stability selection, and provides stronger theoretical properties. In a large set of real clinical prediction datasets, the top-κ stability selection methods outperform many existing feature selection methods including the original stability selection. We also compare three competitive classification methods (SVM, logistic regression and random forest) to demonstrate the effectiveness of selected features by our proposed method in the context of clinical prediction applications. Finally, through several clinical applications on predicting heart failure related symptoms, we show that top-κ stability selection can successfully identify important features that are clinically meaningful.

Original languageEnglish (US)
Title of host publicationProceedings of the 2013 SIAM International Conference on Data Mining, SDM 2013
EditorsSrinivasan Parthasarathy, Joydeep Ghosh, Zhi-Hua Zhou, Jennifer Dy, Zoran Obradovic, Chandrika Kamath
PublisherSiam Society
Pages55-63
Number of pages9
ISBN (Electronic)9781611972627
DOIs
StatePublished - Jan 1 2013
Externally publishedYes
EventSIAM International Conference on Data Mining, SDM 2013 - Austin, United States
Duration: May 2 2013May 4 2013

Publication series

NameProceedings of the 2013 SIAM International Conference on Data Mining, SDM 2013

Other

OtherSIAM International Conference on Data Mining, SDM 2013
Country/TerritoryUnited States
CityAustin
Period5/2/135/4/13

ASJC Scopus subject areas

  • Computer Science Applications
  • Software
  • Theoretical Computer Science
  • Information Systems
  • Signal Processing

Fingerprint

Dive into the research topics of 'Patient risk prediction model via top-κ stability selection'. Together they form a unique fingerprint.

Cite this