Macau: Scalable Bayesian Multi-relational Factorization with Side Information using MCMC
We propose Macau, a powerful and flexible Bayesian factorization method for heterogeneous data. Our model can factorize any set of entities and relations that can be represented by a relational model, including tensors and also multiple relations for each entity. Macau can also incorporate side information, specifically entity and relation features, which are crucial for predicting sparsely observed relations. Macau scales to millions of entity instances, hundred millions of observations, and sparse entity features with millions of dimensions. To achieve the scale up, we specially designed sampling procedure for entity and relation features that relies primarily on noise injection in linear regressions. We show performance and advanced features of Macau in a set of experiments, including challenging drug-protein activity prediction task.
Code (0)
등록된 구현이 없습니다.
Tasks
Activity PredictionRelationSimilar Papers 제목 키워드 기반
Stochastic Relational Models for Large-scale Dyadic Data using MCMC
Stochastic relational models provide a rich family of choices for learning and predicting dyadic data between two sets of entities. It generalizes matrix factorization to a supervised learning problem that utilizes attri…
Bayesian InferenceCollaborative FilteringModelling Relational Data using Bayesian Clustered Tensor Factorization
We consider the problem of learning probabilistic models for complex relational structures between various types of objects. A model can help us ``understand a dataset of relational facts in at least two ways, by findin…
ClusteringReducing the Rank in Relational Factorization Models by Including Observable Patterns
Tensor factorizations have become popular methods for learning from multi-relational data. In this context, the rank of a factorization is an important parameter that determines runtime as well as generalization ability.…
Highly Scalable Tensor Factorization for Prediction of Drug-Protein Interaction Type
The understanding of the type of inhibitory interaction plays an important role in drug design. Therefore, researchers are interested to know whether a drug has competitive or non-competitive interaction to particular pr…
Drug DesignVocal Bursts Type PredictionTemporal Collaborative Filtering with Bayesian Probabilistic Tensor Factorization
Real-world relational data are seldom stationary, yet traditional collaborative filtering algorithms generally rely on this assumption. Motivated by our sales prediction problem, we propose a factor-based algorithm tha…
Collaborative FilteringMovie RecommendationRecommendation Systems