深度共识学习
计算机视觉与模式识别
2021-03-16 v1 机器学习
摘要
生成式学习与判别式学习近来都借助深度神经网络(DNNs)取得了显著进展。对于结构化输入合成与结构化输出预测问题(例如分别为布局到图像合成与图像语义分割),它们通常被分开研究。本文提出深度共识学习(DCL)用于联合布局到图像合成与弱监督图像语义分割。前者由近期提出的LostGAN方法实现,后者通过引入一个推理网络作为第三方加入LostGAN的两方博弈实现。利用两种深度共识映射来促进三个网络的端到端训练:给定输入布局(一组目标边界框),生成器生成掩码(标签图)并用以辅助合成图像。推理网络推断合成图像的掩码。然后,在生成器生成的掩码与推理网络推断的掩码之间度量潜在共识。对于对应输入布局的真实图像,其掩码同样由推理网络计算,随后由生成器用于重建真实图像。然后,在真实图像与其重建图像之间度量数据共识。判别器仍通过为真实图像、其重建图像与合成图像计算真实度分数来扮演对抗者角色。实验中,我们的DCL在COCO-Stuff数据集上测试,取得了引人注目的布局到图像合成结果与弱监督图像语义分割结果。
引用
@article{arxiv.2103.08475,
title = {Deep Consensus Learning},
author = {Wei Sun and Tianfu Wu},
journal= {arXiv preprint arXiv:2103.08475},
year = {2021}
}
备注
Work in progress