Stabilizing high-dimensional prediction models using feature graphs

© 2014 IEEE. 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...

Full description

Bibliographic Details
Main Authors: Gopakumar, S., Tran, The Truyen, Nguyen, T., Phung, D., Venkatesh, S.
Format: Journal Article
Published: Institute of Electrical and Electronics Engineers Inc. 2015
Online Access:http://hdl.handle.net/20.500.11937/27001
Description
Summary:© 2014 IEEE. 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.