用大模型分步推理筛选候选,提升推荐精度。
CORONA: A Coarse-to-Fine Framework for Graph-based Recommendation with Large Language Models
- 大模型分步推理:先基于用户画像找相关项,再结合购买历史细化范围。
- 实验显示召回率和NDCG平均提升18.6%和18.4%。
- 适合想用大模型增强推荐系统的研究者或工程师。
推荐系统从海量候选中筛选用户可能感兴趣的内容。现有方法多用图神经网络(GNN)捕捉高阶交互关系,而大语言模型(LLM)虽具强大能力,但以往仅用于重排序或数据增强,未充分参与候选过滤过程。为此,本文提出一种细粒度框架CORONA,利用LLM在候选过滤中的推理能力:首先,基于用户画像进行偏好推理,生成查询以从交互图中检索相关用户与项目;其次,结合前步结果及目标用户的购买历史,进行意图推理,进一步缩小交互子图范围;最后,使用GNN从提取的子图中捕获高阶协同过滤信息,完成增强检索。该框架在多个数据集上验证,平均召回率提升18.6%,NDCG提升18.4%,性能达到当前最优。
原文摘要 · Abstract (English)
Recommender systems (RSs) are designed to retrieve candidate items a user might be interested in from a large pool. A common approach is using graph neural networks (GNNs) to capture high-order interaction relationships. As large language models (LLMs) have shown strong capabilities across domains, researchers are exploring their use to enhance recommendation. However, prior work limits LLMs to re-ranking results or dataset augmentation, failing to utilize their power during candidate filtering - which may lead to suboptimal performance. Instead, we propose to leverage LLMs' reasoning abilities during the candidate filtering process, and introduce Chain Of Retrieval ON grAphs (CORONA) to progressively narrow down the range of candidate items on interaction graphs with the help of LLMs: (1) First, LLM performs preference reasoning based on user profiles, with the response serving as a query to extract relevant users and items from the interaction graph as preference-assisted retrieval; (2) Then, using the information retrieved in the previous step along with the purchase history of target user, LLM conducts intent reasoning to help refine an even smaller interaction subgraph as intent-assisted retrieval; (3) Finally, we employ a GNN to capture high-order collaborative filtering information from the extracted subgraph, performing GNN-enhanced retrieval to generate the final recommendation results. The proposed framework leverages the reasoning capabilities of LLMs during the retrieval process, while seamlessly integrating GNNs to enhance overall recommendation performance. Extensive experiments on various datasets and settings demonstrate that our proposed CORONA achieves state-of-the-art performance with an 18.6% relative improvement in recall and an 18.4% relative improvement in NDCG on average.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。