《M2M-GAN》行人重识别的迁移学习

M2M-GAN: Many-to-Many Generative Adversarial Transfer Learning for Person
Re-Identification
和我毕设好像!

摘要

行人重识别任务中,在实际应用的场景下会遇到跨域不适应的问题,所以也可以说是现在深度学习方法的一个研究方向:跨域迁移学习。说点题外的,之前有看到旷视video的reid研究组说过,现在针对这类跨域适应其实可以通过一个很大的源域数据集就可以很好的使用在实际的场景下,类似于ImageNet这样的数据集,训练的模型,就可以很好的应用在实际的分类任务中,他们现在更关注于遮挡、部分身体、以及相同服装(学校,工厂)的行人重识别。那为什么还有很多人在做呢?我认为很大情况下是因为人工标签的昂贵,对于分类可以直接标注类别就可以了,但是对于行人重识别任务来讲,要标注一个庞大具有很好适应各种场景的数据是需要极其的努力,更何况标注出错带来的影响更大。
在行人重识别的迁移学习的场景下,源数据域是有标签的,但是目标数据域没有标签。但是在重识别里面源域和目标域都存在很多子数据域,比如基于摄像头的子域。为了要解决这种多域的问题,作者提出了一个多对多的生成对抗迁移方法,该方法同时考虑源域的多个子域和目标域的多个子域。通过实现的源域到目标域的转换,就可以利用源于的标签进行监督训练,从而实现跨域适应的作用。

引言

现有的行人重识别跨域迁移学习的两步法CDTL:将有标签的源域数据通过GAN转换为目标域数据,然后对转换后的数据进行监督学习。
所以第一步转换的能力决定了跨域适应的效果。
现有的CDTL方法比如有Deng 和Wei将其作为一个经典的一对一的CycleGAN。然后这样就忽视了很多个子域之间的差别,影响实际的识别效果。
所以就需要有一个能够关注子域转换的方法,但是如果进行每个子域的对应,对应一个有M个子域的源域和有N个的子域的目标域,就需要有MxN个CycleGAN。
图片说明

相关工作

有监督的:
首先是介绍了当前一些显著的reid的方法都是基于不同的损失函数,比如classification loss, verification loss and triplet loss。
然后就是基于部分身体特征的一些方法,有直接在特征图上面进行划分的,也有基于关键点标注的方法。还有基于spatial-temporal constraint时空约束的方法,可以达到market rank1 98的成绩。
跨域适应的:
采用领域分类和域混淆的方法。
基于GAN的风格迁移方法,目前是一对一或者一对二的方法,忽略了多样性。

提出的方法

给定有标注的数据域S和无标注的数据域T
S的子域对应各个摄像头
T的子域
要想实现多对多的转换,就是要实现的转换
如果不考虑子域的存在,cycleGAN将很难优化。这也是之前的风格转换的方法不能够得到很好效果的原因之一。
M2M-GAN目的也很简单,就是要让生成在生成的过程中关注要生成的目标子领域,同时也要知道从哪个领域生成的,所以输入就是这样的:
图片说明
整个的算法框架如下:
图片说明

损失函数

Adversarial Loss

图片说明
判别器D的目的就是能够很好的去判断目标域和根据源域生成的目标域图片的真假,而生成器G的就是能够生成能够骗过判别器的结果。

Reconstruction Loss

图片说明
CycleGAN里面的东西,能够很好的保证生成的结果足够的真实。

Mask-based Identity Preserve Loss

图片说明
根据Mask-RCNN的分割结果,来很好的保证行人的图像信息。这里M(.)就是一个输入的mask,将没有人的地方去除掉。

Sub-domain Classification Loss

图片说明
这里是为了保证生成的子领域的正确性,需要一个分类器进行监督。

整个优化目标

图片说明

实验结果

图片说明
总的来说,实验还是很丰富的,但是就是决定这是个工程项目,创新点不足吧。

全部评论

相关推荐

工作基本定了,比较满意。这深度学习真是难搞啊,论文看不懂,代码看不懂,不会改,创新点??? 还有一个月中期,怎么搞,已经抑郁了,没有小论文要求,我真是非常幸运了,不然就我这科研能力………… 感觉自己对自己失去信心,无时无刻不在否定自己取得一些成绩都是运气好罢了,真的能顺利毕业吗?严重内耗、无力感爆棚啊…………
AI牛可乐:哎呀,听起来你最近压力好大呀,毕业论文和工作的双重压力一定让你感觉像是背了两座山呢。深度学习确实挺复杂的,就像是要在一片迷雾中找到出路一样。但是,你已经很棒了,工作都定好了,这是很大的成就呢! 论文的事情,别太担心。每个人都有看不懂代码和论文的时候,这是正常的。至于创新点嘛,有时候灵感就像小猫一样,需要一点一点地哄出来的。你可以先试试把你看过的资料整理一下,说不定就能找到新的思路呢。 中期答辩前,你可以列个计划,一步一步来,把大问题分解成小问题,一点一点克服。记得,你不是一个人在战斗,你的导师、同学都是你的战友哦。 至于信心嘛,别否定自己,每个人都会有低谷的时候。你已经取得的成绩,都是你实力的体现。毕业不是问题,只要你不放弃,就一定能顺利走过去的。加油哦,我相信你可以的! 对了,如果你愿意的话,可以点击我的头像,我们私信聊,也许我能帮你出出主意,或者就是简单地聊聊天,分散一下你的注意力也好呀。🐮💪🌟
点赞 评论 收藏
分享
牛客868257804号:九个中铁八个中建
点赞 评论 收藏
分享
11-05 07:29
贵州大学 Java
点赞 评论 收藏
分享
评论
1
1
分享
牛客网
牛客企业服务