提出轻量级知识蒸馏方法,提升图卷积推荐系统的增量训练效率。
Light distillation for Incremental Graph Convolution Collaborative Filtering
- 直接从历史交互中蒸馏用户对物品的偏好得分,减少参数和计算开销。
- 在两个数据集上训练时间减少1.5到9.5倍,召回率比微调方法提升超5%。
- 适合需要快速更新、资源受限的实时推荐场景,如电商与短视频平台。
推荐系统目前依赖海量数据,在提升用户体验方面发挥关键作用。图卷积网络(GCNs)因其捕捉复杂关系信息的能力,成为推荐系统中的高效模型。然而,数据持续增长使GCN训练成本过高。为解决此问题,增量训练新数据块已成为重要研究方向。现有基于知识蒸馏的方法虽能缓解灾难性遗忘,但引入额外参数,模型复杂度高,在增量设置下训练耗时过长,难以实际部署。本文提出一种轻量级偏好驱动蒸馏方法,直接从历史交互中蒸馏用户对物品的偏好得分,显著降低增量训练时间,且性能损失可忽略。在两个通用数据集上的实验表明,该方法相比现有方法训练时间减少1.5至9.5倍,相较于微调方法,Recall@20分别提升5.41%和10.64%。
原文摘要 · Abstract (English)
Recommender systems presently utilize vast amounts of data and play a pivotal role in enhancing user experiences. Graph Convolution Networks (GCNs) have surfaced as highly efficient models within the realm of recommender systems due to their ability to capture extensive relational information. The continuously expanding volume of data may render the training of GCNs excessively costly. To tackle this problem, incrementally training GCNs as new data blocks come in has become a vital research direction. Knowledge distillation techniques have been explored as a general paradigm to train GCNs incrementally and alleviate the catastrophic forgetting problem that typically occurs in incremental settings. However, we argue that current methods based on knowledge distillation introduce additional parameters and have a high model complexity, which results in unrealistic training time consumption in an incremental setting and thus difficult to actually deploy in the real world. In this work, we propose a light preference-driven distillation method to distill the preference score of a user for an item directly from historical interactions, which reduces the training time consumption in the incremental setting significantly without noticeable loss in performance. The experimental result on two general datasets shows that the proposed method can save training time from 1.5x to 9.5x compared to the existing methods and improves Recall@20 by 5.41% and 10.64% from the fine-tune method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。