Repository logo
 

Comparative Study of Inference Methods for Bayesian Nonnegative Matrix Factorisation.

Accepted version
Peer-reviewed

Loading...
Thumbnail Image

Type

Conference Object

Change log

Authors

Brouwer, Thomas 
Frellsen, Jes 
Liò, Pietro 

Abstract

In this paper, we study the trade-offs of different inference approaches for Bayesian matrix factorisation methods, which are commonly used for predicting missing values, and for finding patterns in the data. In particular, we consider Bayesian nonnegative variants of matrix factorisation and tri-factorisation, and compare non-probabilistic inference, Gibbs sampling, variational Bayesian inference, and a maximum-a-posteriori approach. The variational approach is new for the Bayesian nonnegative models. We compare their convergence, and robustness to noise and sparsity of the data, on both synthetic and real-world datasets. Furthermore, we extend the models with the Bayesian automatic relevance determination prior, allowing the models to perform automatic model selection, and demonstrate its efficiency. Code and data related to this chapter are available at: https://github.com/ThomasBrouwer/BNMTF_ARD.

Description

Keywords

Journal Title

ECML/PKDD (1)

Conference Name

Joint European Conference on Machine Learning and Knowledge Discovery in Databases

Journal ISSN

Volume Title

10534

Publisher

Springer
Sponsorship
This work was supported by the UK Engineering and Physical Sciences Research Council (EPSRC), grant reference EP/M506485/1. JF acknowledge funding from the Danish Council for Independent Research 0602-02909B.