Back to Search
Start Over
Contrastive Learning and Deep Fusion Recommendation Model Based on ID Features
- Source :
- IEEE Access, Vol 12, Pp 163001-163015 (2024)
- Publication Year :
- 2024
- Publisher :
- IEEE, 2024.
-
Abstract
- In recent years, the application of deep learning in recommendation systems has achieved breakthrough progress. Neural networks have captured the complex nonlinear relationships between user behaviors and item characteristics. As indispensable elements, ID (Identity) features are widely used to construct user-item association models. The embedding of ID features plays a crucial role in neural network-based recommendation models, enhancing the model’s learning ability by converting them into one-hot vectors as input to the deep learning network. ID features are discrete and high-dimensional sparse, containing critical information for reasoning in recommendation systems. However, most models use the same embedding method to encode all features, leading to inaccurate representation of ID features and affecting the model’s generalization ability. This paper proposes a Contrastive Learning and Deep Fusion Recommendation Model based on ID features (CDM-ID) to address this issue. This model independently encodes ID features first, then accurately re-encodes ID and attribute features through a contrastive learning module, generating high-quality inputs for subsequent feature interaction modules. Finally, it achieves feature fusion through an attention mechanism, significantly improving the performance of the recommendation model. Experiments conducted on three public datasets demonstrate that CDM-ID achieves an average improvement of 8.95% and 5.17% in AUC and F1 scores, respectively, compared to the various baseline models. Additionally, it reduces MAE and RMSE by an average of 9.62% and 12.75%, respectively, further validating the effectiveness and superiority of the proposed model. The contributions of this paper lie in optimizing ID feature embeddings, enhancing embedding accuracy, generating multi-order interaction features, and applying an attention mechanism for feature fusion, thereby significantly improving the performance and prediction accuracy of recommendation systems.
Details
- Language :
- English
- ISSN :
- 21693536
- Volume :
- 12
- Database :
- Directory of Open Access Journals
- Journal :
- IEEE Access
- Publication Type :
- Academic Journal
- Accession number :
- edsdoj.429d7e8d03ac451ebd9693e8152f9906
- Document Type :
- article
- Full Text :
- https://doi.org/10.1109/ACCESS.2024.3491824