arrow
Return

Large scale tensor regression using kernels and variational inference

delete2021-11-09
delete1
delete
OA
AI
R
Robert Hu *
G
Geoff K. Nicholls
D
Dino Sejdinović
DOI:10.1007/s10994-021-06067-7delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
We outline an inherent flaw of tensor factorization models when latent factors are expressed as a function of side information and propose a novel method to mitigate this. We coin our methodology kernel fried tensor (KFT) and present it as a large-scale prediction and forecasting tool for high dimensional data. Our results show superior performance against LightGBM and Field aware factorization machines (FFM), two algorithms with proven track records, widely used in large-scale prediction. We also develop a variational inference framework for KFT which enables associating the predictions and forecasts with calibrated uncertainty estimates on several datasets.
Keywords:
Large scale prediction
Tensor
RKHS
Kernel methods
Variational inference
Bayesian
Uncertainty quantification

Journal

Machine Learning cover
Machine Learning
IF:
2.9
Papers:
2.6K
Citations:
3.4W

Organization

U
university of oxford
Scholars:
9.7W
Papers: 8.6W
Citations: 137