Patient Risk Prediction Model via Top-k Stability Selection

Previous chapter Next chapter Full AccessProceedings Proceedings of the 2013 SIAM International Conference on Data Mining (SDM)Patient Risk Prediction Model via Top-k Stability SelectionJiayu Zhou, Jimeng Sun, Yashu Liu, Jianying Hu, and Jieping YeJiayu Zhou, Jimeng Sun, Yashu Liu, Jianying Hu, and Jieping Yepp.55 - 63Chapter DOI:https://doi.org/10.1137/1.9781611972832.7PDFBibTexSections ToolsAdd to favoritesExport CitationTrack CitationsEmail SectionsAboutAbstract 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-k stability selection, which generalizes a powerful sparse learning method for feature selection by overcoming its limitation on parameter selection. In particular, our proposed top-k stability selection includes the original stability selection method as a special case given k = 1. Moreover, we show that the top-k 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-k 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-k stability selection can successfully identify important features that are clinically meaningful. Previous chapter Next chapter RelatedDetails Published:2013ISBN:978-1-61197-262-7eISBN:978-1-61197-283-2 https://doi.org/10.1137/1.9781611972832Book Series Name:ProceedingsBook Code:PRDT13Book Pages:1-804