Scalable model selection for belief networks


Conference Paper

© 2017 Neural information processing systems foundation. All rights reserved. We propose a scalable algorithm for model selection in sigmoid belief networks (SBNs), based on the factorized asymptotic Bayesian (FAB) framework. We derive the corresponding generalized factorized information criterion (gFIC) for the SBN, which is proven to be statistically consistent with the marginal log-likelihood. To capture the dependencies within hidden variables in SBNs, a recognition network is employed to model the variational distribution. The resulting algorithm, which we call FABIA, can simultaneously execute both model selection and inference by maximizing the lower bound of gFIC. On both synthetic and real data, our experiments suggest that FABIA, when compared to state-of-the-art algorithms for learning SBNs, (i) produces a more concise model, thus enabling faster testing; (ii) improves predictive performance; (iii) accelerates convergence; and (iv) prevents overfitting.

Duke Authors

Cited Authors

  • Song, Z; Muraoka, Y; Fujimaki, R; Carin, L

Published Date

  • January 1, 2017

Published In

Volume / Issue

  • 2017-December /

Start / End Page

  • 4610 - 4620

International Standard Serial Number (ISSN)

  • 1049-5258

Citation Source

  • Scopus