通过任务感知检索增强,提升动态推荐系统泛化能力
Task-Aware Retrieval Augmentation for Dynamic Recommendation
- 用任务感知机制自动筛选相关历史子图,构建特定任务数据集
- 在多个大规模数据集上显著优于现有方法,准确率与泛化性俱佳
- 适合需要捕捉用户偏好演变的实时推荐场景
动态推荐系统旨在通过建模时间序列用户-项目交互,提供个性化建议。近期研究利用预训练动态图神经网络(GNN)在时序快照图上学习用户-项目表示,但微调阶段常因预训练与微调间的时间差异导致泛化问题,难以捕捉用户偏好的演化。为此,我们提出TarDGR,一种任务感知的检索增强框架,通过引入任务感知评估机制识别语义相关的历史子图,构建无需人工标注的任务特定数据集;并设计基于图变换器的任务感知模型,融合语义与结构编码以评估子图相关性。推理时,TarDGR检索并融合任务感知子图与查询子图,丰富表示,缓解时间泛化问题。在多个大规模动态图数据集上的实验表明,TarDGR持续优于当前最优方法,实证证明其在准确率和泛化能力上的优越性。
原文摘要 · Abstract (English)
Dynamic recommendation systems aim to provide personalized suggestions by modeling temporal user-item interactions across time-series behavioral data. Recent studies have leveraged pre-trained dynamic graph neural networks (GNNs) to learn user-item representations over temporal snapshot graphs. However, fine-tuning GNNs on these graphs often results in generalization issues due to temporal discrepancies between pre-training and fine-tuning stages, limiting the model's ability to capture evolving user preferences. To address this, we propose TarDGR, a task-aware retrieval-augmented framework designed to enhance generalization capability by incorporating task-aware model and retrieval-augmentation. Specifically, TarDGR introduces a Task-Aware Evaluation Mechanism to identify semantically relevant historical subgraphs, enabling the construction of task-specific datasets without manual labeling. It also presents a Graph Transformer-based Task-Aware Model that integrates semantic and structural encodings to assess subgraph relevance. During inference, TarDGR retrieves and fuses task-aware subgraphs with the query subgraph, enriching its representation and mitigating temporal generalization issues. Experiments on multiple large-scale dynamic graph datasets demonstrate that TarDGR consistently outperforms state-of-the-art methods, with extensive empirical evidence underscoring its superior accuracy and generalization capabilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。