Topic Modeling on Health Journals With Regularized Variational Inference

Authors

  • Robert Giaquinto University of Minnesota
  • Arindam Banerjee University of Minnesota

Keywords:

machine learning, topic modeling, graphical model, regularized variational inference, healthcare

Abstract

Topic modeling enables exploration and compact representation of a corpus. The CaringBridge (CB) dataset is a massive collection of journals written by patients and caregivers during a health crisis. Topic modeling on the CB dataset, however, is challenging due to the asynchronous nature of multiple authors writing about their health journeys. To overcome this challenge we introduce the Dynamic Author-Persona topic model (DAP), a probabilistic graphical model designed for temporal corpora with multiple authors. The novelty of the DAP model lies in its representation of authors by a persona---where personas capture the propensity to write about certain topics over time. Further, we present a regularized variational inference (RVI) algorithm, which we use to encourage the DAP model's personas to be distinct. Our results show significant improvements over competing topic models---particularly after regularization, and highlight the DAP model's unique ability to capture common journeys shared by different authors.

Downloads

Published

2018-04-29

How to Cite

Giaquinto, R., & Banerjee, A. (2018). Topic Modeling on Health Journals With Regularized Variational Inference. Proceedings of the AAAI Conference on Artificial Intelligence, 32(1). Retrieved from https://ojs.aaai.org/index.php/AAAI/article/view/11745