Sequential Recommendation via Adaptive Robust Attention with Multi-dimensional Embeddings
Pang, Linsey, Raffiee, Amir Hossein, Liu, Wei, Lundgaard, Keld
–arXiv.org Artificial Intelligence
Sequential recommendation models have achieved state-of-the-art performance using self-attention mechanism. It has since been found that moving beyond only using item ID and positional embeddings leads to a significant accuracy boost when predicting the next item. In recent literature, it was reported that a multi-dimensional kernel embedding with temporal contextual kernels to capture users' diverse behavioral patterns results in a substantial performance improvement. In this study, we further improve the sequential recommender model's robustness and generalization by introducing a mix-attention mechanism with a layer-wise noise injection (LNI) regularization. We refer to our proposed model as adaptive robust sequential recommendation framework (ADRRec), and demonstrate through extensive experiments that our model outperforms existing self-attention architectures.
arXiv.org Artificial Intelligence
Sep-8-2024
- Country:
- North America > United States
- New York > New York County > New York City (0.04)
- Oceania > Australia (0.04)
- North America > United States
- Genre:
- Research Report > New Finding (0.34)
- Industry:
- Information Technology (0.30)
- Technology: