{"title":"Self-supervised contrastive learning for implicit collaborative filtering","authors":"Shipeng Song , Bin Liu , Fei Teng , Tianrui Li","doi":"10.1016/j.engappai.2024.109563","DOIUrl":null,"url":null,"abstract":"<div><div>Recommendation systems are a critical application of artificial intelligence (AI), driving personalized user experiences across various platforms. Recent advancements in contrastive learning-based recommendation algorithms have led to significant progress in self-supervised recommendation. A key method in this field is Bayesian Personalized Ranking (BPR), which has become a dominant approach for implicit collaborative filtering. However, the challenge of false-positive and false-negative examples in implicit feedback continues to hinder accurate preference learning. In this study, we introduce an efficient self-supervised contrastive learning framework that enhances the supervisory signal by incorporating positive feature augmentation and negative label augmentation. Our theoretical analysis reveals that this approach is equivalent to maximizing the likelihood estimation with latent variables representing user interest centers. Additionally, we present a novel negative label augmentation technique that selects unlabeled examples based on their relative ranking positions, enabling efficient augmentation with constant time complexity. Validation on the MovieLens-100k, MovieLens-1M, Yahoo!-R3, Yelp2018, and Gowalla datasets demonstrates that our method achieves over a 5% improvement in precision compared to the widely used BPR optimization objective, while maintaining comparable runtime efficiency.</div></div>","PeriodicalId":50523,"journal":{"name":"Engineering Applications of Artificial Intelligence","volume":"139 ","pages":"Article 109563"},"PeriodicalIF":7.5000,"publicationDate":"2024-11-07","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":null,"platform":"Semanticscholar","paperid":null,"PeriodicalName":"Engineering Applications of Artificial Intelligence","FirstCategoryId":"94","ListUrlMain":"https://www.sciencedirect.com/science/article/pii/S0952197624017214","RegionNum":2,"RegionCategory":"计算机科学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q1","JCRName":"AUTOMATION & CONTROL SYSTEMS","Score":null,"Total":0}
引用次数: 0
Abstract
Recommendation systems are a critical application of artificial intelligence (AI), driving personalized user experiences across various platforms. Recent advancements in contrastive learning-based recommendation algorithms have led to significant progress in self-supervised recommendation. A key method in this field is Bayesian Personalized Ranking (BPR), which has become a dominant approach for implicit collaborative filtering. However, the challenge of false-positive and false-negative examples in implicit feedback continues to hinder accurate preference learning. In this study, we introduce an efficient self-supervised contrastive learning framework that enhances the supervisory signal by incorporating positive feature augmentation and negative label augmentation. Our theoretical analysis reveals that this approach is equivalent to maximizing the likelihood estimation with latent variables representing user interest centers. Additionally, we present a novel negative label augmentation technique that selects unlabeled examples based on their relative ranking positions, enabling efficient augmentation with constant time complexity. Validation on the MovieLens-100k, MovieLens-1M, Yahoo!-R3, Yelp2018, and Gowalla datasets demonstrates that our method achieves over a 5% improvement in precision compared to the widely used BPR optimization objective, while maintaining comparable runtime efficiency.
期刊介绍:
Artificial Intelligence (AI) is pivotal in driving the fourth industrial revolution, witnessing remarkable advancements across various machine learning methodologies. AI techniques have become indispensable tools for practicing engineers, enabling them to tackle previously insurmountable challenges. Engineering Applications of Artificial Intelligence serves as a global platform for the swift dissemination of research elucidating the practical application of AI methods across all engineering disciplines. Submitted papers are expected to present novel aspects of AI utilized in real-world engineering applications, validated using publicly available datasets to ensure the replicability of research outcomes. Join us in exploring the transformative potential of AI in engineering.