A unifying variational inference framework for hierarchical graph-coupled HMM with an application to influenza infection

Conference Paper

© Copyright 2016, Association for the Advancement of Artificial Intelligence (www.aaai.org). All rights reserved. The Hierarchical Graph-Coupled Hidden Markov Model (hGCHMM) is a useful tool for tracking and predicting the spread of contagious diseases, such as influenza, by leveraging social contact data collected from individual wearable devices. However, the existing inference algorithms depend on the assumption that the infection rates are small in probability, typically close to 0. The purpose of this paper is to build a unified learning framework for latent infection state estimation for the hGCHMM, regardless of the infection rate and transition function. We derive our algorithm based on a dynamic auto-encoding variational inference scheme, thus potentially generalizing the hGCHMM to models other than those that work on highly contagious diseases. We experimentally compare our approach with previous Gibbs EM algorithms and standard variational method mean-field inference, on both semi-synthetic data and app collected epidemiological and social records.

Duke Authors

Cited Authors

  • Fan, K; Li, C; Heller, K

Published Date

  • January 1, 2016

Published In

  • 30th AAAI Conference on Artificial Intelligence, AAAI 2016

Start / End Page

  • 3828 - 3834

International Standard Book Number 13 (ISBN-13)

  • 9781577357605

Citation Source

  • Scopus