Dynamic embeddings for interaction prediction
2021 (English)In: The Web Conference 2021 - Proceedings of the World Wide Web Conference, WWW 2021, Association for Computing Machinery, Inc , 2021, p. 1609-1618Conference paper, Published paper (Refereed)
Abstract [en]
In recommender systems (RSs), predicting the next item that a user interacts with is critical for user retention. While the last decade has seen an explosion of RSs aimed at identifying relevant items that match user preferences, there is still a range of aspects that could be considered to further improve their performance. For example, often RSs are centered around the user, who is modeled using her recent sequence of activities. Recent studies, however, have shown the effectiveness of modeling the mutual interactions between users and items using separate user and item embeddings. Building on the success of these studies, we propose a novel method called DeePRed that addresses some of their limitations. In particular, we avoid recursive and costly interactions between consecutive short-term embeddings by using long-term (stationary) embeddings as a proxy. This enable us to train DeePRed using simple mini-batches without the overhead of specialized mini-batches proposed in previous studies. Moreover, DeePRed's effectiveness comes from the aforementioned design and a multi-way attention mechanism that inspects user-item compatibility. Experiments show that DeePRed outperforms the best state-of-the-art approach by at least 14% of Mean Reciprocal Rank (MRR) on next item prediction task, while gaining more than an order of magnitude speedup over the best performing baselines. Although this study is mainly concerned with temporal interaction networks, we also show the power and flexibility of DeePRed by adapting it to the case of static interaction networks, substituting the short- and long-term aspects with local and global ones.
Place, publisher, year, edition, pages
Association for Computing Machinery, Inc , 2021. p. 1609-1618
Keywords [en]
Dynamic embeddings, Interaction prediction, Multi-way attention, Mutual RNN, Recommender systems, Forecasting, World Wide Web, Attention mechanisms, Best state, Interaction networks, Mean reciprocal ranks, Mutual interaction, Prediction tasks, Static interaction, Embeddings
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:kth:diva-309946DOI: 10.1145/3442381.3450020ISI: 000733621801054Scopus ID: 2-s2.0-85107936313OAI: oai:DiVA.org:kth-309946DiVA, id: diva2:1645898
Conference
2021 World Wide Web Conference, WWW 2021, 19 April 2021 through 23 April 2021, Ljubljana Slovenia
Note
Part of proceedings: ISBN 978-1-4503-8312-7
QC 20220517
2022-03-212022-03-212023-01-18Bibliographically approved