用集合到集合模型推荐缺失食材,让购物篮更完整。
S2SRec2: Set-to-Set Recommendation for Basket Completion with Recipe
- 将补全购物篮建模为集合到集合的推荐问题,考虑多食材协同。
- 在真实数据集上,推荐准确率提升18.6%,且能预测篮子完整性。
- 适合做智能厨艺助手或电商补货系统的研究与开发人员。
在生鲜电商中,用户常根据饮食偏好构建食材篮,却缺乏完成整餐的能力。利用食谱知识为部分篮子推荐互补食材,对提升烹饪体验至关重要。传统方法通常采用留一法预测单一缺失食材,存在两大缺陷:(i) 无法反映实际中常需多个食材的场景;(ii) 忽略缺失食材间的相互关系。为此,我们将篮子补全重新定义为集合到集合(S2S)推荐问题,提出 S2SRec2 框架,基于 Set Transformer 并采用多任务学习训练。该框架联合学习:(i) 从已有食材表示中检索缺失食材;(ii) 预测补全后篮子的完整性。两项任务共同优化,确保推荐精准且搭配合理。在大规模食谱数据集上的实验及定性分析表明,S2SRec2 显著优于单目标基线,在 Top-10 推荐准确率上提升 18.6%,为改善生鲜购物和激发烹饪创意提供有效方案。
原文摘要 · Abstract (English)
In grocery e-commerce, customers often build ingredient baskets guided by dietary preferences but lack the expertise to create complete meals. Leveraging recipe knowledge to recommend complementary ingredients based on a partial basket is essential for improving the culinary experience. Traditional recipe completion methods typically predict a single missing ingredient using a leave-one-out strategy. However, they fall short in two key aspects: (i) they do not reflect real-world scenarios where multiple ingredients are often needed, and (ii) they overlook relationships among the missing ingredients themselves. To address these limitations, we reformulate basket completion as a set-to-set (S2S) recommendation problem, where an incomplete basket is input into a system that predicts a set of complementary ingredients. We introduce S2SRec2, a set-to-set ingredient recommendation framework based on a Set Transformer and trained in a multitask learning paradigm. S2SRec2 jointly learns to (i) retrieve missing ingredients from the representation of existing ones and (ii) assess basket completeness after prediction. These tasks are optimized together, enforcing accurate retrieval and coherent basket completion. Experiments on large-scale recipe datasets and qualitative analyses show that S2SRec2 significantly outperforms single-target baselines, offering a promising approach to enhance grocery shopping and inspire culinary creativity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。