News Recommendation with Category Description by a Large Language Model

Yuki Yada, Hayato Yamana
5/13/2024
cs.CLcs.AIcs.IR

Official code: yamanalab/gpt-augmented-news-recommendation (9 stars, Apache-2.0, PyTorch, last commit 2024-05-13).

Abstract

Personalized news recommendations are essential for online news platforms to assist users in discovering news articles that match their interests from a vast amount of online content. Appropriately encoded content features, such as text, categories, and images, are essential for recommendations. Among these features, news categories, such as tv-golden-globe, finance-real-estate, and news-politics, play an important role in understanding news content, inspiring us to enhance the categories' descriptions. In this paper, we propose a novel method that automatically generates informative category descriptions using a large language model (LLM) without manual effort or domain-specific knowledge and incorporates them into recommendation models as additional information. In our comprehensive experimental evaluations using the MIND dataset, our method successfully achieved 5.8% improvement at most in AUC compared with baseline approaches without the LLM's generated category descriptions for the state-of-the-art content-based recommendation models including NAML, NRMS, and NPA. These results validate the effectiveness of our approach. The code is available at https://github.com/yamanalab/gpt-augmented-news-recommendation.

View on arXivView PDF

Code Implementations(1)

PyTorchApache-2.0

News Recommendation with Category Description by a Large Language Model

92Shell, PythonMay 3, 20242 years agoApache-2.0
deep-learninglarge-language-modelsnews-recommendationpythonpytorch

Similar Papers

Cite this paper

@article{yada2024news,
  title  = {News Recommendation with Category Description by a Large Language Model},
  author = {Yuki Yada and Hayato Yamana},
  year   = {2024},
  eprint = {2405.13007},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.13007v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.13007.svg)](https://codeofpaper.com/paper/2405.13007)

Discussion