arXiv:2412.16502cs.LGcs.IR2024-12AAAI被引 6

通过时空知识蒸馏,提升外卖推荐系统对用户动态偏好的捕捉能力。

Spatial-Temporal Knowledge Distillation for Takeaway Recommendation

  • 分两阶段训练:先构建时空知识图,再蒸馏知识到序列模型
  • 在三个真实数据集上超越当前最优基线,显著提升推荐效果
  • 适合研究推荐系统、时空建模与知识蒸馏的学者和工程师

外卖推荐系统旨在基于用户历史购买行为预测其未来的外卖订单,以提升用户满意度并促进商家销售。现有方法多依赖辅助信息或知识图谱缓解用户购买序列稀疏问题,但仍面临两大挑战:(1)捕捉复杂地理空间信息下的动态用户偏好;(2)高效融合图结构与序列数据中的时空知识,同时保持低计算成本。本文提出一种新型两阶段时空知识蒸馏模型(STKDRec)。第一阶段预训练中,时空知识图编码器(STKG)用于从时空知识图中提取高阶时空依赖与协同关联。第二阶段采用时空Transformer(ST-Transformer),从序列视角全面建模用户对细粒度地理空间信息的动态偏好。引入时空知识蒸馏策略,将图结构知识迁移至ST-Transformer,实现来自知识图与序列数据的丰富知识自适应融合,同时降低计算开销。在三个真实数据集上的大量实验表明,STKDRec显著优于当前最优基线。

原文摘要 · Abstract (English)

The takeaway recommendation system aims to recommend users' future takeaway purchases based on their historical purchase behaviors, thereby improving user satisfaction and boosting merchant sales. Existing methods focus on incorporating auxiliary information or leveraging knowledge graphs to alleviate the sparsity issue of user purchase sequences. However, two main challenges limit the performance of these approaches: (1) capturing dynamic user preferences on complex geospatial information and (2) efficiently integrating spatial-temporal knowledge from both graphs and sequence data with low computational costs. In this paper, we propose a novel spatial-temporal knowledge distillation model for takeaway recommendation (STKDRec) based on the two-stage training process. Specifically, during the first pre-training stage, a spatial-temporal knowledge graph (STKG) encoder is trained to extract high-order spatial-temporal dependencies and collaborative associations from the STKG. During the second spatial-temporal knowledge distillation (STKD) stage, a spatial-temporal Transformer (ST-Transformer) is employed to comprehensively model dynamic user preferences on various types of fine-grained geospatial information from a sequential perspective. Furthermore, the STKD strategy is introduced to transfer graph-based spatial-temporal knowledge to the ST-Transformer, facilitating the adaptive fusion of rich knowledge derived from both the STKG and sequence data while reducing computational overhead. Extensive experiments on three real-world datasets show that STKDRec significantly outperforms the state-of-the-art baselines.

推荐系统时空建模知识蒸馏外卖推荐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。