Deep collective matrix factorization for augmented multi-view learning

被引:8
作者
Mariappan, Ragunathan [1 ]
Rajan, Vaibhav [1 ]
机构
[1] Natl Univ Singapore, Sch Comp, Singapore, Singapore
关键词
Collective Matrix Factorization; Deep learning; Augmented multi-view learning; Bayesian optimization; Recommendation; Gene-disease prioritization; GENOME-WIDE ASSOCIATION; DISEASE GENES; IDENTIFICATION; PREDICTION; NETWORK; NORM;
D O I
10.1007/s10994-019-05801-6
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Learning by integrating multiple heterogeneous data sources is a common requirement in many tasks. Collective Matrix Factorization (CMF) is a technique to learn shared latent representations from arbitrary collections of matrices. It can be used to simultaneously complete one or more matrices, for predicting the unknown entries. Classical CMF methods assume linearity in the interaction of latent factors which can be restrictive and fails to capture complex non-linear interactions. In this paper, we develop the first deep-learning based method, called dCMF, for unsupervised learning of multiple shared representations, that can model such non-linear interactions, from an arbitrary collection of matrices. We address optimization challenges that arise due to dependencies between shared representations through multi-task Bayesian optimization and design an acquisition function adapted for collective learning of hyperparameters. Our experiments show that dCMF significantly outperforms previous CMF algorithms in integrating heterogeneous data for predictive modeling. Further, on two tasks-recommendation and prediction of gene-disease association-dCMF outperforms state-of-the-art matrix completion algorithms that can utilize auxiliary sources of information.
引用
收藏
页码:1395 / 1420
页数:26
相关论文
共 56 条
  • [1] Andrienko G., 2013, Introduction, P1
  • [2] [Anonymous], 2014, ADAM METHOD STOCHAST
  • [3] [Anonymous], PROCEEDINGS OF THE 2
  • [4] [Anonymous], 2016, DEEP LEARNING
  • [5] [Anonymous], 2017, F1000Res
  • [6] [Anonymous], 2016, P UAI
  • [7] [Anonymous], 2011, P 24 ADV NEUR INF PR
  • [8] Representation Learning: A Review and New Perspectives
    Bengio, Yoshua
    Courville, Aaron
    Vincent, Pascal
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2013, 35 (08) : 1798 - 1828
  • [9] Bergstra J, 2012, J MACH LEARN RES, V13, P281
  • [10] Bouchard G., 2013, Proceedings of the 16th International Conference on Artificial Intelligence and Statistics, P144