Multi-axis Gating MLP with Bidirectional Feature Extraction for Sequence Recommendation
Huang Shaowei, Xiangping Wu, Ke Wang, Qingqing Gao, Xiaopeng Li · 2022 IEEE Intl Conf on Dependable, Autonomic and Secure Computing, Intl Conf on Pervasive Intelligence and Computing, Intl Conf on Cloud and Big Data Computing, Intl Conf on Cyber Science and Technology Congress (DASC/PiCom/CBDCom/CyberSciTech) · 2022
The sequence recommendation task uses neural network models such as GRU and Self-Attention to extract potential preference characteristics from the user’s behavior data, This makes the recommendation more accurate. However, they use the sequence information of items to build embedded features, which may destroy the representation of interaction between items. Therefore, based on the traditional RNN network, we use multi-axis gating unit to capture the user behavior characteristics inside and across channels. In our experiments, we found that multi-axis gating unit can significantly improve the performance of sequence recommendation model. Compared with Self-Attention model, MLP architecture has less parameters and lower time complexity. Specifically, we propose a spatial gating unit MLP model for sequential recommendation tasks, which uses multi axis gating units to capture the behavior characteristics of multiple users in the spatial dimension. A large number of experiments on multiple real data sets show that our proposed method is superior to RNN and Self-Attention based methods.