中文

S2SRec2:基于食谱的篮子完成集合到集合推荐

机器学习 2025-07-15 v1

摘要

在 grocery 电商中,客户常依据饮食偏好构建食材篮子,但缺乏专业知识以创建完整的餐点。利用食谱知识为基于部分篮子推荐互补食材对于提升烹饪体验至关重要。传统的食谱完成方法通常仅预测单个缺失食材,采用留一损失策略。然而,这些方法在两个关键方面不足:(i)它们未能反映实际场景,其中通常需要多个食材;(ii)它们忽略了缺失食材之间的关系。为此,我们将篮子完成重新表述为集合到集合(S2S)推荐问题,输入不完整篮子后系统预测一组互补食材。我们引入 S2SRec2,一个基于 Set Transformer 的集合到集合食材推荐框架,并在多任务学习范式下训练。S2SRec2 同时学习:(i)从现有食材表示中检索缺失食材;(ii)预测后评估篮子完整性。这些任务相互优化,确保准确检索和连贯的篮子完成。在大规模食谱数据集上的实验和定性分析表明,S2SRec2 显著优于单目标基线,为提升购物体验和激发烹饪创意提供了有前景的方案。

关键词

引用

@article{arxiv.2507.09101,
  title  = {S2SRec2: Set-to-Set Recommendation for Basket Completion with Recipe},
  author = {Yanan Cao and Omid Memarrast and Shiqin Cai and Sinduja Subramaniam and Evren Korpeoglu and Kannan Achan},
  journal= {arXiv preprint arXiv:2507.09101},
  year   = {2025}
}