用物品关联信息增强推荐,让大模型更懂用户偏好。
ItemRAG: Item-Based Retrieval-Augmented Generation for LLM-Based Recommendation
- 基于物品而非用户历史检索相关商品
- 结合共购与语义信息提升检索准确性
- 特别适合冷启动商品推荐场景
近期,大型语言模型(LLMs)因其推理能力被广泛用于推荐系统,尤其在处理冷启动商品方面表现优异。现有主流方法通过提示目标用户的购买历史,从候选集推荐商品,并常采用检索增强生成(RAG)。但大多数RAG方法仅检索与目标用户相似用户的购买历史,这些历史常包含噪声或弱相关数据,对候选商品帮助有限。为此,我们提出ItemRAG,一种新型RAG方法,将检索重点从粗粒度的用户历史转向细粒度的物品级检索。ItemRAG通过检索与目标用户历史中每个物品或候选商品相关的物品,来增强其描述。为确保检索不仅语义相关且对推荐有用,ItemRAG融合共购信息与语义信息。通过精心组合,ItemRAG优先选择更具信息量的检索结果,同时显著提升冷启动商品推荐效果。大量实验表明,ItemRAG在标准和冷启动推荐设置下均持续优于现有RAG方法。补充材料、代码及数据集见https://github.com/kswoo97/ItemRAG。
原文摘要 · Abstract (English)
Recently, large language models (LLMs) have been widely used as recommender systems, owing to their reasoning capability and effectiveness in handling cold-start items. A common approach prompts an LLM with a target user's purchase history to recommend items from a candidate set, often enhanced with retrieval-augmented generation (RAG). Most existing RAG approaches retrieve purchase histories of users similar to the target user; however, these histories often contain noisy or weakly relevant information and provide little or no useful information for candidate items. To address these limitations, we propose ItemRAG, a novel RAG approach that shifts focus from coarse user-history retrieval to fine-grained item-level retrieval. ItemRAG augments the description of each item in the target user's history or the candidate set by retrieving items relevant to each. To retrieve items not merely semantically similar but informative for recommendation, ItemRAG leverages co-purchase information alongside semantic information. Especially, through their careful combination, ItemRAG prioritizes more informative retrievals and also benefits cold-start items. Through extensive experiments, we demonstrate that ItemRAG consistently outperforms existing RAG approaches under both standard and cold-start item recommendation settings. Supplementary materials, code, and datasets are provided at https://github.com/kswoo97/ItemRAG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。