A framework for mixed-type multioutcome prediction with applications in healthcare

Budhaditya Saha, Sunil Gupta, DInh Phung, Svetha Venkatesh

Research output: Contribution to journalArticleResearchpeer-review

5 Citations (Scopus)

Abstract

Health analysis often involves prediction of multiple outcomes of mixed type. The existing work is restrictive to either a limited number or specific outcome types. We propose a framework for mixed-type multioutcome prediction. Our proposed framework proposes a cumulative loss function composed of a specific loss function for each outcome type-as an example, least square (continuous outcome), hinge (binary outcome), Poisson (count outcome), and exponential (nonnegative outcome). To model these outcomes jointly, we impose a commonality across the prediction parameters through a common matrix normal prior. The framework is formulated as iterative optimization problems and solved using an efficient block-coordinate descent method. We empirically demonstrate both scalability and convergence. We apply the proposed model to a synthetic dataset and then on two real-world cohorts: A cancer cohort and an acute myocardial infarction cohort collected over a two-year period. We predict multiple emergency-related outcomes-as example, future emergency presentations (binary), emergency admissions (count), emergency length of stay days (nonnegative), and emergency time to next admission day (nonnegative). We show that the predictive performance of the proposed model is better than several state-of-the-art baselines.

Original languageEnglish
Article number7879827
Pages (from-to)1182-1191
Number of pages10
JournalIEEE Journal of Biomedical and Health Informatics
Volume21
Issue number4
DOIs
Publication statusPublished - Jul 2017
Externally publishedYes

Keywords

  • Health information management
  • Multi-task learning
  • Optimization

Cite this