Pre-trained Language Model and Knowledge Distillation for Lightweight Sequential Recommendation

Abstract: Sequential recommendation models user interests based on historical behaviors to provide personalized recommendation. Previous sequential recommendation algorithms primarily employ neural networks to extract features of user interests, achieving good performance. However, due to the recommendation system datasets sparsity, these algorithms often employ small-scale network frameworks, resulting in weaker generalization capability. Recently, a series of sequential recommendation algorithms based on large pre-trained LLMs have been proposed. Nonetheless, given the real-time demands of recommendation systems, the challenge remains in applying pre-trained LLMs for rapid recommendations in real scenarios. To address this, we propose a sequential recommendation algorithm based on a pre-trained LLM and knowledge distillation. The key of proposed algorithm is to transfer pre-trained knowledge across domains and achieve lightweight inference by knowledge distillation. The algorithm operates in two stages: in the first stage, we fine-tune the pre-trained LLM on the recommendation dataset to transfer the pre-trained knowledge to the recommendation task; in the second stage, we distill the trained LLM to transfer the learned knowledge to a lightweight model. Extensive experiments on multiple public recommendation datasets show that the proposed algorithm enhances recommendation accuracy and provide timely recommendation services.