用大模型架构但不用文本,轻量推荐系统性能超群。
Architecture is All You Need: Improving LLM Recommenders by Dropping the Text
- 用物品编号代替文本分词,保留大模型架构
- 模型规模仅为原模型的1/100,精度仍更优
- 适合资源有限、追求高效推荐的场景
近年来,大量研究将大型预训练语言模型(PLMs)应用于推荐系统,并在常见基准数据集上展现出强劲性能。基于PLM的推荐模型具备灵活提示设计、无限可推荐项词汇量及通过海量文本预训练获得的通用世界知识。然而,由于模型体积庞大、计算成本高,实际部署困难;且微调以提升协同信号表现可能损害模型的世界知识与泛化能力。本文提出一种新型推荐模型:保留大语言模型(LLM)的架构特征,但减少层数与维度,并用唯一标识单个内容项的离散令牌替代传统子词分词。实验表明,该简化方法在仅需极小规模与计算开销的情况下,显著优于传统序列推荐模型和基于PLM的推荐模型。结果表明,大模型在推荐系统中的核心优势来自其架构设计,而非预训练获取的世界知识。
原文摘要 · Abstract (English)
In recent years, there has been an explosion of interest in the applications of large pre-trained language models (PLMs) to recommender systems, with many studies showing strong performance of PLMs on common benchmark datasets. PLM-based recommender models benefit from flexible and customizable prompting, an unlimited vocabulary of recommendable items, and general ``world knowledge'' acquired through pre-training on massive text corpora. While PLM-based recommenders show promise in settings where data is limited, they are hard to implement in practice due to their large size and computational cost. Additionally, fine-tuning PLMs to improve performance on collaborative signals may degrade the model's capacity for world knowledge and generalizability. We propose a recommender model that uses the architecture of large language models (LLMs) while reducing layer count and dimensions and replacing the text-based subword tokenization of a typical LLM with discrete tokens that uniquely represent individual content items. We find that this simplified approach substantially outperforms both traditional sequential recommender models and PLM-based recommender models at a tiny fraction of the size and computational complexity of PLM-based models. Our results suggest that the principal benefit of LLMs in recommender systems is their architecture, rather than the world knowledge acquired during extensive pre-training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。