paper-with-me

홈 › Papers

SLMRec: Distilling Large Language Models into Small for Sequential Recommendation

2024-05-28 · Wujiang Xu, Qitian Wu, Zujie Liang, Jiaojiao Han, Xuying Ning, Yunxiao Shi, Wenfang Lin, Yongfeng Zhang

Sequential Recommendation (SR) task involves predicting the next item a user is likely to interact with, given their past interactions. The SR models examine the sequence of a user's actions to discern more complex behavioral patterns and temporal dynamics. Recent research demonstrates the great impact of LLMs on sequential recommendation systems, either viewing sequential recommendation as language modeling or serving as the backbone for user representation. Although these methods deliver outstanding performance, there is scant evidence of the necessity of a large language model and how large the language model is needed, especially in the sequential recommendation scene. Meanwhile, due to the huge size of LLMs, it is inefficient and impractical to apply a LLM-based model in real-world platforms that often need to process billions of traffic logs daily. In this paper, we explore the influence of LLMs' depth by conducting extensive experiments on large-scale industry datasets. Surprisingly, our motivational experiments reveal that most intermediate layers of LLMs are redundant, indicating that pruning the remaining layers can still maintain strong performance. Motivated by this insight, we empower small language models for SR, namely SLMRec, which adopt a simple yet effective knowledge distillation method. Moreover, SLMRec is orthogonal to other post-training efficiency techniques, such as quantization and pruning, so that they can be leveraged in combination. Comprehensive experimental results illustrate that the proposed SLMRec model attains the best performance using only 13% of the parameters found in LLM-based recommendation models while simultaneously achieving up to 6.6x and 8.0x speedups in training and inference time costs, respectively. Besides, we provide a theoretical justification for why small language models can perform comparably to large language models in SR.

📄 PDF Abstract BibTeX arXiv:2405.17890

Code (1)

wujiangxu/slmrec 공식 구현 pytorch

Tasks

Knowledge DistillationLanguage ModelingLanguage ModellingLarge Language ModelQuantizationRecommendation SystemsSequential Recommendation

Methods 이 논문이 사용한 방법론

Pruning 설명 없음
Knowledge Distillation A very simple way to improve the performance of almost any machine learning algorithm is to train many different models on the same data and then to average their predictions.…

Similar Papers 제목 키워드 기반

Mind's Mirror: Distilling Self-Evaluation Capability and Comprehensive Thinking from Large Language Models

2023-11-15 · Weize Liu, Guocong Li, Kai Zhang, Bang Du 외

Large language models (LLMs) have achieved remarkable advancements in natural language processing. However, the massive scale and computational demands of these models present formidable challenges when considering their…

Transfer Learning

Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizes

2023-05-03 · Cheng-Yu Hsieh, Chun-Liang Li, Chih-Kuan Yeh, Hootan Nakhost 외

Deploying large language models (LLMs) is challenging because they are memory inefficient and compute-intensive for practical applications. In reaction, researchers train smaller task-specific models by either finetuning…

Distilling LLM Agent into Small Models with Retrieval and Code Tools

2025-05-23 · Minki Kang, Jongwon Jeong, Seanie Lee, Jaewoong Cho 외

Large language models (LLMs) excel at complex reasoning tasks but remain computationally expensive, limiting their practical deployment. To address this, recent works have focused on distilling reasoning capabilities int…

Action GenerationDomain GeneralizationRetrieval

Self-Supervised Learning for Multimedia Recommendation

2022-06-30 · IEEE Transactions on Multimedia (TMM) 2022 6 · Zhulin Tao, Xiaohao Liu, Yewei Xia, Xiang Wang 외

Learning representations for multimedia content is critical for multimedia recommendation. Current representation learning methods roughly fall into two groups: (1) using the historical interactions to create ID embeddin…

Contrastive LearningData AugmentationMultimedia recommendationMulti-modal Recommendation+2

Distilling Large Language Models into Tiny and Effective Students using pQRNN

2021-01-21 · Prabhu Kaliamoorthi, Aditya Siddhant, Edward Li, Melvin Johnson

Large pre-trained multilingual models like mBERT, XLM-R achieve state of the art results on language understanding tasks. However, they are not well suited for latency critical applications on both servers and edge devic…

Data AugmentationSemantic ParsingXLM-R