Attention-over-Attention Field-Aware Factorization Machine

Factorization Machine (FM) has been a popular approach in supervised predictive tasks, such as click-through rate prediction and recommender systems, due to its great performance and efficiency. Recently, several variants of FM have been proposed to improve its performance. However, most of the state-of-the-art prediction algorithms neglected the field information of features, and they also failed to discriminate the importance of feature interactions due to the problem of redundant features. In this paper, we present a novel algorithm called Attention-over-Attention Field-aware Factorization Machine (AoAFFM) for better capturing the characteristics of feature interactions. Specifically, we propose the fieldaware embedding layer to exploit the field information of features, and combine it with the attention-over-attention mechanism to learn both feature-level and interaction-level attention to estimate the weight of feature interactions. Experimental results show that the proposed AoAFFM improves FM and FFM with large margin, and outperforms state-of-the-art algorithms on three public benchmark datasets.

Authors

• Zhibo Wang School of Cyber Science and Technology, Zhejiang University, P.R. China
• Jinxin Ma
• Yongquan Zhang
• Qian Wang
• Ju Ren Department of Computer Science and Technology, Tsinghua University
• Peng Sun

Keywords

No keywords are indexed for this paper.