Enhancing Factorization Machines with Generalized Metric Learning (extended Abstract).

Yangyang Guo,Zhiyong Cheng,Jiazheng Jing,Yanpeng Lin,Liqiang Nie,Meng Wang
DOI: https://doi.org/10.1109/icde55515.2023.00304
2023-01-01
Abstract:Factorization Machines (FMs) have been proven effective in modeling the second-order interactions between attributes. Nevertheless, existing methods often fall short of learning the complex and rich intra-attribute feature interactions, which are abundantly exhibited in real-world recommendation data. To address this problem, we propose an FM framework equipped with generalized metric learning to better handle the essential feature correlations. In particular, based on this framework, we present a Mahalanobis distance and a deep neural network method, which can effectively model the linear and non-linear correlations between features, respectively. Besides, we design an efficient approach for simplifying the model functions. Experiments on several benchmark datasets demonstrate that our proposed framework outperforms several state-of-the-art baselines by a significant margin.
What problem does this paper attempt to address?