Neural News Recommendation with Multi-Head Self-Attention

Code

• github.com/microsoft/recommenders
  tf★ 21,541
• github.com/anuscode/nrms
  pytorch★ 0
• github.com/wuch15/EMNLP2019-NRMS
  none★ 0
• github.com/aqweteddy/NRMS-Pytorch
  pytorch★ 0

Abstract

News recommendation can help users find interested news and alleviate information overload. Precisely modeling news and users is critical for news recommendation, and capturing the contexts of words and news is important to learn news and user representations. In this paper, we propose a neural news recommendation approach with multi-head self-attention (NRMS). The core of our approach is a news encoder and a user encoder. In the news encoder, we use multi-head self-attentions to learn news representations from news titles by modeling the interactions between words. In the user encoder, we learn representations of users from their browsed news and use multi-head self-attention to capture the relatedness between the news. Besides, we apply additive attention to learn more informative news and user representations by selecting important words and news. Experiments on a real-world dataset validate the effectiveness and efficiency of our approach.

Tasks

Reproductions