Selection GAN:基于级联语义引导下的多通道注意力选择图像翻译

一.研究背景

目前图像翻译问题的解决方案一般是基于Encoder-Decoder结构,即将原域图像编码后再解码到目标域中,然而这种方案在原域与目标域图像具有显著不同结构或重叠区域极少的情况下时翻译效果会大打折扣。作者发现之前利用语义图指导图像翻译的模型对于图像细节的翻译效果不佳,作者认为这是由于语义图一般是由深度预训练模型产生,并不能保证像素级的准确性。基于此作者提出了级联语义引导下的基于多通道注意力选择机制的图像翻译Selection GAN,其将图像翻译分为两个阶段,第一个阶段用于产生粗粒度级的翻译结果,第二个阶段通过多通道注意力选择机制产生更细致的结果(如图1所示)。
Selection GAN:基于级联语义引导下的多通道注意力选择图像翻译_第1张图片
图1.效果概览

二.模型介绍

Selection GAN将图像翻译的过程分为两个阶段(如图2所示),第一阶段作者提出了一个级联语义引导的生成器子网络(Gi),该网络是将原域图像(Ia)以及目标域语义图(Sg)级联后的结果翻译至目标域图像(Ig’),该生成图像Ig’作为语义生成器(Gs)的输入去生成目标域的语义图(Sg’)。为了保证Gi和Gs的生成效果,这里使用重建损失来限制生成器,即Ig≈Ig’,Sg≈Sg’,另外目标域语义图(Sg)是由深度预训练模型产生。
Selection GAN:基于级联语义引导下的多通道注意力选择图像翻译_第2张图片
图2.模型概览
经过第一阶段后模型产生了粗粒度级的翻译结果Ig’,在第二阶段中作者将该结果以及第一阶段生成的深度特征级联作为多通道注意力选择模型的输入(如图3所示),其目的是从更大的生成空间中产生更加细粒度级的翻译结果并且生成不确定映射去引导优化像素损失。
多通道注意力选择模型包括多规模空间池化和多通道注意力选择两个部分,作者选取第一阶段中Gi和Gs的最后一层卷积网络的输出作为深度特征Fi和Fs,并与第一阶段的输出Ig’和原域图像Ia级联为特征Fc,即,输入至多规模空间池化网络中,该网络对Fc进行不同规模的平均池化从而获取多规模的空间上下文特征。为了保留有用信息将经过不同规模池化后的特征与输入特征Fc相乘,该结果经过卷积后产生新的多规模特征Fc’并作为多通道注意力选择的输入。
作者认为普通的三通道RGB图像所包含的信息太少,因此作者在多通道注意力选择中通过卷积网络扩大图像的通道表示,并且结合注意力映射产生更合理的结果(Ig’’),另外注意力映射还需要学习产生不确定映射用于引导优化由预训练模型所带来的像素级误差对整个模型的影响。
Selection GAN:基于级联语义引导下的多通道注意力选择图像翻译_第3张图片
图3.多通道注意力选择模型
Selection GAN的优化目标主要分为三个部分:重建损失、对抗损失以及全变分正则化损失(如图4所示)。这里的重建损失包括三个部分:Ig≈Ig’,Sg≈Sg’(第一阶段)以及Ig≈Ig’’(第二阶段)。
在这里插入图片描述
图4.模型优化目标
作者认为不同于原始GAN,Selection GAN的鉴别器应该学习分辨来自不同域的成对图像是否彼此相关联,因此这里的对抗损失是用于区分真实图像对(Ia,Ig)以及虚假图像对(Ia,Ig’),如图5所示。
在这里插入图片描述
图5. 对抗损失

三.实验

在具体实验中对于Gi和Gs作者采用了U-Net架构,由于模型的目的是生成合理的翻译结果因此Gi采用了深层网络模型,Gs采用了潜层网络模型。对于鉴别器D作者采用了PatchGAN架构,用于生成于语义图的预训练模型采用了RefineNet架构。
作者主要在Dayton、CVUSA以及Ego2Top数据集上进行实验,其基线模型选取了Pix2Pix,X-Fork以及X-Seq进行对比,评价指标采用了SSIM,PSNR,SD以及KL散度进行衡量,实验结果如下。
Selection GAN:基于级联语义引导下的多通道注意力选择图像翻译_第4张图片
另外作者还选取了Inception Score进行衡量,结果如下。
Selection GAN:基于级联语义引导下的多通道注意力选择图像翻译_第5张图片

四.总结

Selection GAN旨在解决具有显著不同结构的图像翻译问题,其将翻译过程分为两个阶段,阶段一旨在捕获场景的语义结构,阶段二通过提出的多通道注意力选择模块关注更多外观细节。另外作者还提出了利用不确定性映射引导优化由预训练模型产生的语义图像素损失,以解决不准确的语义标签问题。在三个公共数据集的实验结果表明,该方法获得了比现有技术更好的结果。

你可能感兴趣的:(机器学习,图像翻译)