Supervised mixture of experts models for population health.

Abstract

We propose a machine learning driven approach to derive insights from observational healthcare data to improve public health outcomes. Our goal is to simultaneously identify patient subpopulations with differing health risks and to find those risk factors within each subpopulation. We develop two supervised mixture of experts models: a Supervised Gaussian Mixture model (SGMM) for general features and a Supervised Bernoulli Mixture model (SBMM) tailored to binary features. We demonstrate the two approaches on an analysis of high cost drivers of Medicaid expenditures for inpatient stays. We focus on the three diagnostic categories that accounted for the highest percentage of inpatient expenditures in New York State (NYS) in 2016. When compared with state-of-the-art learning methods (random forests, boosting, neural networks), our approaches provide comparable prediction performance while also extracting insightful subpopulation structure and risk factors. For problems with binary features the proposed SBMM provides as good or better performance than alternative methods while offering insightful explanations. Our results indicate the promise of such approaches for extracting population health insights from electronic health care records.