1. Madlon-Kay, Seth, Lauren Brent, Michael Montague, Katherine Heller, and Michael Platt. Using Machine Learning to Discover Latent Social Phenotypes in Free-Ranging Macaques Brain sciences 7 (2017): 91.
  2. Kai Fan*, Qi Wei*, Lawrence Carin, Katherine Heller. Inner-loop free ADMM using Auxiliary Deep Neural Networks. NIPS 2017.
  3. Futoma, Joseph, Sanjay Hariharan, and Katherine Heller. Learning to Detect Sepsis with a Multitask Gaussian Process RNN Classifier. ICML 2017.
  4. Lorenzi, Elizabeth C., Stephanie L. Brown, Zhifei Sun, and Katherine Heller. Predictive Hierarchical Clustering: Learning clusters of CPT codes for improving surgical outcomes. MLHC 2017.
  5. Futoma, Joseph, Sanjay Hariharan, Mark Sendak, Nathan Brajer, Meredith Clement, Armando Bedoya, Cara O'Brien, and Katherine Heller. An Improved Multi-Output Gaussian Process RNN with Real-Time Validation for Early Sepsis Detection MLHC 2017.


  1. Elizabeth C Lorenzi, Zhifei Sun, Erich Huang, Ricardo Henao, and Katherine A Heller. Transfer Learning via Latent Factor Models for Predicting Surgical Complications. 2016 NIPS Workshop on Machine Learning for Health.
  2. Joseph Futoma, Mark Sendak, C. Blake Cameron, and Katherine Heller. Predicting Disease Progression with a Model for Multivariate Longitudinal Clinical Data Disease. MLHC 2016
  3. Joseph Futoma, Mark Sendak, C. Blake Cameron, and Katherine Heller. Scalable Joint Modeling of Longitudinal and Point Process Data for Disease Trajectory Prediction and Improving Management of Chronic Kidney Disease. UAI 2016
  4. Xi Tan, Syed A.Z. Naqvi, Alan Qi, Katherine Heller, Vinayak Rao. Content-based Modeling of Reciprocal Relationships using Hawkes and Gaussian Processes. UAI 2016
  5. Kai Fan, Yizhe Zhang, Ricardo Henao, and Katherine Heller. Triply Stochastic Variational Inference for Non-linear Beta Process Factor Analysis. ICDM 2016
  6. Kai Fan, Chunyuan Li, and Katherine Heller. A Unifying Variational Inference Framework for Hierarchical Graph-Coupled HMM with an Application to Influenza Infection. AAAI 2016


  1. Kai Fan and Katherine A. Heller. Scalable Non-linear Beta Process Factory Analysis. NIPS 2015
  2. Kai Fan, Ziteng Wang, Jeffrey Beck, James Kwok and Katherine A. Heller. Fast Second Order Stochastic Backpropagation for Variational Inference. NIPS 2015
  3. Xiangyu Wang, Fangjian Guo, Katherine A. Heller and David B. Dunson. Parallelizing MCMC with random partition trees. NIPS 2015 (GitHub)
  4. Kai Fan, Marisa Eisenberg, Alison Walsh, Allison Aiello and Katherine A. Heller. Hierarchical Graph-Coupled HMMs for Heterogeneous Personalized Health Data. KDD 2015
  5. Jiefeng Jiang, Jeffrey Beck, Katherine A. Heller, and Tobias Egner. An insula-frontostriatal network mediates flexible cognitive control by adaptively predicting changing control demands. Nature Communications 6 (2015).
  6. Fangjian Guo, Charles Blundell, Hanna M. Wallach and Katherine A. Heller. The Bayesian Echo Chamber: Modeling Influence in Conversations. AISTATS 2015 (GitHub)


  1. Jiefeng Jiang, Katherine A. Heller, and Tobias Egner. Bayesian modeling of flexible cognitive control. Neuroscience and Biobehavioral Reviews 46 (2014): 30-43.
  2. Shakir Mohamed, Katherine A. Heller, and Zoubin Ghahramani. Bayesian Approaches for Sparse Latent Variable Models: Reconsidering L1 Sparsity. Practical Applications of Sparse Modeling (2014): 203.


  1. Benjamin Letham, Cynthia Rudin, and Katherine A. Heller. Growing a list. Data Mining and Knowledge Discovery 27.3 (2013): 372-395.