Title :
Collaborative Kalman Filtering for Dynamic Matrix Factorization
Author :
Sun, J.Z. ; Parthasarathy, Dhruv ; Varshney, Kush R.
Author_Institution :
Dept. of Electr. Eng. & Comput. Sci., Massachusetts Inst. of Technol., Cambridge, MA, USA
Abstract :
We propose a new algorithm for estimation, prediction, and recommendation named the collaborative Kalman filter. Suited for use in collaborative filtering settings encountered in recommendation systems with significant temporal dynamics in user preferences, the approach extends probabilistic matrix factorization in time through a state-space model. This leads to an estimation procedure with parallel Kalman filters and smoothers coupled through item factors. Learning of global parameters uses the expectation-maximization algorithm. The method is compared to existing techniques and performs favorably on both generated data and real-world movie recommendation data.
Keywords :
Kalman filters; collaborative filtering; expectation-maximisation algorithm; learning (artificial intelligence); matrix decomposition; probability; recommender systems; state-space methods; collaborative Kalman filtering; dynamic matrix factorization; estimation algorithm; expectation-maximization algorithm; global parameter learning; item factors; parallel Kalman filters; prediction algorithm; probabilistic matrix factorization; recommendation systems; state-space model; temporal dynamics; user preferences; Collaboration; Estimation; Heuristic algorithms; Hidden Markov models; Kalman filters; Probabilistic logic; Signal processing algorithms; Collaborative filtering; Kalman filtering; expectation-maximization; learning; recommendation systems;
Journal_Title :
Signal Processing, IEEE Transactions on
DOI :
10.1109/TSP.2014.2326618