国内刊号:11-5602/TP
国际刊号:1673-9418
发布日期:
作者:武随烁,杨金福,单义,许兵兵
单位:1. 北京工业大学 信息学部,北京 1001242. 计算智能与智能系统北京市重点实验室,北京 100124
关键词:深度学习,生成对抗网络(GAN),生成模型,对抗学习,注意力机制
生成对抗网络(GAN)能够生成逼真的图像,已成为生成模型中的一个研究热点。针对生成对抗网络无法有效提取图像局部与全局特征间依赖关系以及各类别间的依赖关系,提出一种用于生成对抗网络的孪生注意力模型(TAGAN)。以孪生注意力机制为驱动,通过模拟局部与全局特征间的依赖关系以及各类别间依赖关系,对真实自然图像建模,创建逼真的非真实图像。孪生注意力机制包含特征注意力模型和通道注意力模型,特征注意力模型通过有选择地聚合特征,学习相似特征间的关联性,通道注意力模型通过整合各通道维度的相关特征,学习各通道的内部依赖关系。在MNIST、CIFAR10和CelebA64数据集上验证了所提出模型的有效性。
来源:2020年第5期
《计算机科学与探索》期刊编辑部