/ibp_vi

VI implementation for inference of the IBP

Primary LanguageJupyter Notebook

Variational Inference for the IBP posterior

The VI implementation is based from Doshi-Velez et al 2009.

The notebooks contains an implementation of matrix factorisation using the IBP prior over latent features in the dictionary. The implementation contains finite approximations based on a Beta-Bernoulli model from Griffiths and Ghahramani in ibp_vi.ipynb and a finite stick-breaking contruction from Teh et al in the notebook ibp_vi_sb.ipynb.