arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.08098cs.CVcs.IR

基于Transformer的图神经网络用于服装搭配生成

Transformer-based Graph Neural Networks for Outfit Generation

University of Florence(佛罗伦萨大学)

Federico Becattini, Federico Maria Teotini, Alberto Del Bimbo

AI总结:

本文提出基于Transformer的图神经网络TGNN,利用多头自注意力在图结构中建模服装间关系,通过从种子单品迭代选择最兼容服装实现穿搭生成,并在兼容性估计上取得最优结果。

摘要:

推荐互补的服装单品以组成一套穿搭是一个日益受到关注的过程,但它需要对时尚趋势和视觉美学有精细的理解。以往的工作主要侧重于通过评分视觉吸引力来进行推荐,并将服装表示为有序序列或成对兼容单品的集合。这限制了对服装之间关系的充分利用。我们尝试通过利用集合中单品的图表示来弥合穿搭推荐与生成之间的差距。本文开展的工作试图通过从已有的穿搭集合中发现新的具有吸引力的穿搭,在穿搭推荐与生成之间架起一座桥梁。我们提出了一种基于Transformer的架构,命名为TGNN,它利用多头自注意力来捕捉图中服装单品之间的关系,作为卷积图神经网络中的消息传递步骤。具体而言,从一个种子(即一件或多件服装)开始,穿搭生成通过迭代地选择与先前所选服装最兼容的服装来完成。在两个不同的数据集上进行了大量实验,证明了该模型执行种子穿搭生成的能力,并在兼容性估计任务上取得了最先进的结果。

英文摘要:

Suggesting complementary clothing items to compose an outfit is a process of emerging interest, yet it involves a fine understanding of fashion trends and visual aesthetics. Previous works have mainly focused on recommendation by scoring visual appeal and representing garments as ordered sequences or as collections of pairwise-compatible items. This limits the full usage of relations among clothes. We attempt to bridge the gap between outfit recommendation and generation by leveraging a graph-based representation of items in a collection. The work carried out in this paper, tries to build a bridge between outfit recommendation and generation, by discovering new appealing outfits starting from a collection of pre-existing ones. We propose a transformer-based architecture, named TGNN, which exploits multi-headed self attention to capture relations between clothing items in a graph as a message passing step in Convolutional Graph Neural Networks. Specifically, starting from a seed, i.e.~one or more garments, outfit generation is performed by iteratively choosing the garment that is most compatible with the previously chosen ones. Extensive experimentations are conducted with two different datasets, demonstrating the capability of the model to perform seeded outfit generation as well as obtaining state of the art results on compatibility estimation tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑