基于超图表示的服装兼容性预测模型

摘要：

针对现有服装兼容性模型都集中探究成对单品之间的兼容性这一问题,提出一种基于超图表示的服装兼容性预测模型.该模型首先基于现有数据集中时尚服装的不同类别和时尚服装间的搭配关系构建了一个服装超图,其中每个节点表示一件衣服,每条超边表示多件衣服组成的套装.为了更好地从超图中推断服装的兼容性,该模型将超图转换为传统图,并利用图神经网络模拟节点之间的复杂交互.最后引入注意力机制计算服装的兼容性得分,增强模型的预测能力.实验结果表明,在服装填空任务和服装兼容性预测两个服装搭配任务上,该模型分别达到了 77.29％和96.23％的准确率,较其他基线模型有显著的提升.

关键词：服装兼容性、超图表示、图神经网络、注意力机制

所属期刊栏目：39

分类号：TP393.04(计算技术、计算机技术)

资助基金：天津市自然科学基金资助项目20JCQNJC01490

在线出版日期：2022-08-16（万方平台首次上网日期，不代表论文的发表时间）

页数：共5页

页码：2506-2510