Stabilizing high-dimensional prediction models using feature graphs

Shivapratap Gopakumar, Truyen Tran, Tu Dinh Nguyen, Dinh Phung, Svetha Venkatesh

Research output: Contribution to journalArticleResearchpeer-review

6 Citations (Scopus)

Abstract

We investigate feature stability in the context of clinical prognosis derived from high-dimensional electronic medical records. To reduce variance in the selected features that are predictive, we introduce Laplacian-based regularization into a regression model. The Laplacian is derived on a feature graph that captures both the temporal and hierarchic relations between hospital events, diseases, and interventions. Using a cohort of patients with heart failure, we demonstrate better feature stability and goodness-of-fit through feature graph stabilization.

Original languageEnglish
Article number6887285
Pages (from-to)1044-1052
Number of pages9
JournalIEEE Journal of Biomedical and Health Informatics
Volume19
Issue number3
DOIs
Publication statusPublished - May 2015
Externally publishedYes

Cite this