STAR: A Simple Training-free Approach for Recommendations using Large Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lee, Dong-Ho, Kraft, Adam, Jin, Long, Mehta, Nikhil, Xu, Taibai, Hong, Lichan, Chi, Ed H., Yi, Xinyang
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929720752340992
author Lee, Dong-Ho
Kraft, Adam
Jin, Long
Mehta, Nikhil
Xu, Taibai
Hong, Lichan
Chi, Ed H.
Yi, Xinyang
author_facet Lee, Dong-Ho
Kraft, Adam
Jin, Long
Mehta, Nikhil
Xu, Taibai
Hong, Lichan
Chi, Ed H.
Yi, Xinyang
contents Recent progress in large language models (LLMs) offers promising new approaches for recommendation system tasks. While the current state-of-the-art methods rely on fine-tuning LLMs to achieve optimal results, this process is costly and introduces significant engineering complexities. Conversely, methods that directly use LLMs without additional fine-tuning result in a large drop in recommendation quality, often due to the inability to capture collaborative information. In this paper, we propose a Simple Training-free Approach for Recommendation (STAR), a framework that utilizes LLMs and can be applied to various recommendation tasks without the need for fine-tuning, while maintaining high quality recommendation performance. Our approach involves a retrieval stage that uses semantic embeddings from LLMs combined with collaborative user information to retrieve candidate items. We then apply an LLM for pairwise ranking to enhance next-item prediction. Experimental results on the Amazon Review dataset show competitive performance for next item prediction, even with our retrieval stage alone. Our full method achieves Hits@10 performance of +23.8% on Beauty, +37.5% on Toys & Games, and -1.8% on Sports & Outdoors relative to the best supervised models. This framework offers an effective alternative to traditional supervised models, highlighting the potential of LLMs in recommendation systems without extensive training or custom architectures.
format Preprint
id arxiv_https___arxiv_org_abs_2410_16458
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle STAR: A Simple Training-free Approach for Recommendations using Large Language Models
Lee, Dong-Ho
Kraft, Adam
Jin, Long
Mehta, Nikhil
Xu, Taibai
Hong, Lichan
Chi, Ed H.
Yi, Xinyang
Information Retrieval
Artificial Intelligence
Machine Learning
Recent progress in large language models (LLMs) offers promising new approaches for recommendation system tasks. While the current state-of-the-art methods rely on fine-tuning LLMs to achieve optimal results, this process is costly and introduces significant engineering complexities. Conversely, methods that directly use LLMs without additional fine-tuning result in a large drop in recommendation quality, often due to the inability to capture collaborative information. In this paper, we propose a Simple Training-free Approach for Recommendation (STAR), a framework that utilizes LLMs and can be applied to various recommendation tasks without the need for fine-tuning, while maintaining high quality recommendation performance. Our approach involves a retrieval stage that uses semantic embeddings from LLMs combined with collaborative user information to retrieve candidate items. We then apply an LLM for pairwise ranking to enhance next-item prediction. Experimental results on the Amazon Review dataset show competitive performance for next item prediction, even with our retrieval stage alone. Our full method achieves Hits@10 performance of +23.8% on Beauty, +37.5% on Toys & Games, and -1.8% on Sports & Outdoors relative to the best supervised models. This framework offers an effective alternative to traditional supervised models, highlighting the potential of LLMs in recommendation systems without extensive training or custom architectures.
title STAR: A Simple Training-free Approach for Recommendations using Large Language Models
topic Information Retrieval
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2410.16458