用多模态嵌入和语义ID提升大模型在序列推荐中的表现
Empowering Large Language Model for Sequential Recommendation via Multimodal Embeddings and Semantic IDs
- 融合多模态嵌入与量化语义ID,缓解嵌入坍塌问题
- 在三个数据集上超越基线,最佳提升达12.3%
- 适合关注大模型推荐系统优化的研究者
序列推荐(SR)旨在基于用户历史交互捕捉其动态兴趣和序列模式。近期大型语言模型(LLM)的强大能力推动了其在SR中的应用。然而,现有基于LLM的SR方法存在两大关键挑战:1)引入预训练协同嵌入时出现嵌入坍塌;2)使用语义ID时量化嵌入发生灾难性遗忘。这些问题限制了模型可扩展性并导致推荐性能不佳。为此,基于Llama3-8B-instruct等大模型,我们提出新型框架MME-SID,通过整合多模态嵌入与量化嵌入缓解嵌入坍塌。同时,设计多模态残差量化变分自编码器(MM-RQ-VAE),以最大均值差异为重构损失,结合对比学习对齐特征,有效保留模态内距离信息并捕捉跨模态相关性。为缓解灾难性遗忘,采用训练好的多模态码嵌入进行初始化。最后,通过LoRA以多模态频率感知方式高效微调LLM。在三个公开数据集上的大量实验验证了MME-SID的优越性能,得益于其对嵌入坍塌和遗忘的有效抑制。代码与数据集已公开:https://github.com/Applied-Machine-Learning-Lab/MME-SID。
原文摘要 · Abstract (English)
Sequential recommendation (SR) aims to capture users' dynamic interests and sequential patterns based on their historical interactions. Recently, the powerful capabilities of large language models (LLMs) have driven their adoption in SR. However, we identify two critical challenges in existing LLM-based SR methods: 1) embedding collapse when incorporating pre-trained collaborative embeddings and 2) catastrophic forgetting of quantized embeddings when utilizing semantic IDs. These issues dampen the model scalability and lead to suboptimal recommendation performance. Therefore, based on LLMs like Llama3-8B-instruct, we introduce a novel SR framework named MME-SID, which integrates multimodal embeddings and quantized embeddings to mitigate embedding collapse. Additionally, we propose a Multimodal Residual Quantized Variational Autoencoder (MM-RQ-VAE) with maximum mean discrepancy as the reconstruction loss and contrastive learning for alignment, which effectively preserve intra-modal distance information and capture inter-modal correlations, respectively. To further alleviate catastrophic forgetting, we initialize the model with the trained multimodal code embeddings. Finally, we fine-tune the LLM efficiently using LoRA in a multimodal frequency-aware fusion manner. Extensive experiments on three public datasets validate the superior performance of MME-SID thanks to its capability to mitigate embedding collapse and catastrophic forgetting. The implementation code and datasets are publicly available for reproduction: https://github.com/Applied-Machine-Learning-Lab/MME-SID.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。