Improving Content Recommendation: Knowledge Graph-Based Semantic Contrastive Learning for Diversity and Cold-Start Users

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Kim, Yejin, Rome, Scott, Foley, Kevin, Nankani, Mayur, Melamed, Rimon, Morales, Javier, Yadav, Abhay, Peifer, Maria, Hamidian, Sardar, Huang, H. Howie
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913286574833664
author Kim, Yejin
Rome, Scott
Foley, Kevin
Nankani, Mayur
Melamed, Rimon
Morales, Javier
Yadav, Abhay
Peifer, Maria
Hamidian, Sardar
Huang, H. Howie
author_facet Kim, Yejin
Rome, Scott
Foley, Kevin
Nankani, Mayur
Melamed, Rimon
Morales, Javier
Yadav, Abhay
Peifer, Maria
Hamidian, Sardar
Huang, H. Howie
contents Addressing the challenges related to data sparsity, cold-start problems, and diversity in recommendation systems is both crucial and demanding. Many current solutions leverage knowledge graphs to tackle these issues by combining both item-based and user-item collaborative signals. A common trend in these approaches focuses on improving ranking performance at the cost of escalating model complexity, reducing diversity, and complicating the task. It is essential to provide recommendations that are both personalized and diverse, rather than solely relying on achieving high rank-based performance, such as Click-through Rate, Recall, etc. In this paper, we propose a hybrid multi-task learning approach, training on user-item and item-item interactions. We apply item-based contrastive learning on descriptive text, sampling positive and negative pairs based on item metadata. Our approach allows the model to better understand the relationships between entities within the knowledge graph by utilizing semantic information from text. It leads to more accurate, relevant, and diverse user recommendations and a benefit that extends even to cold-start users who have few interactions with items. We perform extensive experiments on two widely used datasets to validate the effectiveness of our approach. Our findings demonstrate that jointly training user-item interactions and item-based signals using synopsis text is highly effective. Furthermore, our results provide evidence that item-based contrastive learning enhances the quality of entity embeddings, as indicated by metrics such as uniformity and alignment.
format Preprint
id arxiv_https___arxiv_org_abs_2403_18667
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Improving Content Recommendation: Knowledge Graph-Based Semantic Contrastive Learning for Diversity and Cold-Start Users
Kim, Yejin
Rome, Scott
Foley, Kevin
Nankani, Mayur
Melamed, Rimon
Morales, Javier
Yadav, Abhay
Peifer, Maria
Hamidian, Sardar
Huang, H. Howie
Information Retrieval
Computation and Language
Addressing the challenges related to data sparsity, cold-start problems, and diversity in recommendation systems is both crucial and demanding. Many current solutions leverage knowledge graphs to tackle these issues by combining both item-based and user-item collaborative signals. A common trend in these approaches focuses on improving ranking performance at the cost of escalating model complexity, reducing diversity, and complicating the task. It is essential to provide recommendations that are both personalized and diverse, rather than solely relying on achieving high rank-based performance, such as Click-through Rate, Recall, etc. In this paper, we propose a hybrid multi-task learning approach, training on user-item and item-item interactions. We apply item-based contrastive learning on descriptive text, sampling positive and negative pairs based on item metadata. Our approach allows the model to better understand the relationships between entities within the knowledge graph by utilizing semantic information from text. It leads to more accurate, relevant, and diverse user recommendations and a benefit that extends even to cold-start users who have few interactions with items. We perform extensive experiments on two widely used datasets to validate the effectiveness of our approach. Our findings demonstrate that jointly training user-item interactions and item-based signals using synopsis text is highly effective. Furthermore, our results provide evidence that item-based contrastive learning enhances the quality of entity embeddings, as indicated by metrics such as uniformity and alignment.
title Improving Content Recommendation: Knowledge Graph-Based Semantic Contrastive Learning for Diversity and Cold-Start Users
topic Information Retrieval
Computation and Language
url https://arxiv.org/abs/2403.18667