CVPR 2019 | 基于级联语义引导下的多通道注意力选择图像翻译
作者丨薛洁婷
学校丨北京交通大学硕士生
研究方向丨图像翻译
研究背景
目前图像翻译问题的解决方案一般是基于 Encoder-Decoder 结构,即将原域图像编码后再解码到目标域中,然而这种方案在原域与目标域图像具有显著不同结构或重叠区域极少的情况下时翻译效果会大打折扣。
作者发现之前利用语义图指导图像翻译的模型对于图像细节的翻译效果不佳,作者认为这是由于语义图一般是由深度预训练模型产生,并不能保证像素级的准确性。基于此作者提出了级联语义引导下的基于多通道注意力选择机制的图像翻译 Selection GAN,其将图像翻译分为两个阶段,第一个阶段用于产生粗粒度级的翻译结果,第二个阶段通过多通道注意力选择机制产生更细致的结果(如图 1 所示)。
▲ 图1. 效果概览
模型介绍
Selection GAN 将图像翻译的过程分为两个阶段(如图 2 所示),第一阶段作者提出了一个级联语义引导的生成器子网络
▲ 图2. 模型概览
经过第一阶段后模型产生了粗粒度级的翻译结果
多通道注意力选择模型包括多规模空间池化和多通道注意力选择两个部分,作者选取第一阶段中
作者认为普通的三通道 RGB 图像所包含的信息太少,因此作者在多通道注意力选择中通过卷积网络扩大图像的通道表示,并且结合注意力映射产生更合理的结果
▲ 图3. 多通道注意力选择模型
Selection GAN 的优化目标主要分为三个部分:重建损失、对抗损失以及全变分正则化损失(如图 4 所示)。这里的重建损失包括三个部分:
▲ 图 4. 模型优化目标
作者认为不同于原始 GAN,Selection GAN 的鉴别器应该学习分辨来自不同域的成对图像是否彼此相关联,因此这里的对抗损失是用于区分真实图像对
▲ 图5. 对抗损失
实验
在具体实验中对于
作者主要在 Dayton、CVUSA 以及 Ego2Top 数据集上进行实验,其基线模型选取了 Pix2Pix,X-Fork 以及 X-Seq 进行对比,评价指标采用了 SSIM,PSNR,SD 以及 KL 散度进行衡量,实验结果如下。
另外作者还选取了 Inception Score 进行衡量,结果如下。
总结
Selection GAN 旨在解决具有显著不同结构的图像翻译问题,其将翻译过程分为两个阶段,阶段一旨在捕获场景的语义结构,阶段二通过提出的多通道注意力选择模块关注更多外观细节。另外作者还提出了利用不确定性映射引导优化由预训练模型产生的语义图像素损失,以解决不准确的语义标签问题。在三个公共数据集的实验结果表明,该方法获得了比现有技术更好的结果。
点击以下标题查看更多往期内容:
让你的论文被更多人看到
如何才能让更多的优质内容以更短路径到达读者群体,缩短读者寻找优质内容的成本呢?答案就是:你不认识的人。
总有一些你不认识的人,知道你想知道的东西。PaperWeekly 或许可以成为一座桥梁,促使不同背景、不同方向的学者和学术灵感相互碰撞,迸发出更多的可能性。
PaperWeekly 鼓励高校实验室或个人,在我们的平台上分享各类优质内容,可以是最新论文解读,也可以是学习心得或技术干货。我们的目的只有一个,让知识真正流动起来。
📝 来稿标准:
• 稿件确系个人原创作品,来稿需注明作者个人信息(姓名+学校/工作单位+学历/职位+研究方向)
• 如果文章并非首发,请在投稿时提醒并附上所有已发布链接
• PaperWeekly 默认每篇文章都是首发,均会添加“原创”标志
📬 投稿邮箱:
• 投稿邮箱:hr@paperweekly.site
• 所有文章配图,请单独在附件中发送
• 请留下即时联系方式(微信或手机),以便我们在编辑发布时和作者沟通
🔍
现在,在「知乎」也能找到我们了
进入知乎首页搜索「PaperWeekly」
点击「关注」订阅我们的专栏吧
关于PaperWeekly
PaperWeekly 是一个推荐、解读、讨论、报道人工智能前沿论文成果的学术平台。如果你研究或从事 AI 领域,欢迎在公众号后台点击「交流群」,小助手将把你带入 PaperWeekly 的交流群里。
▽ 点击 | 阅读原文 | 下载论文 & 源码