摘要
条件生成对抗网络(cGANs)因其对类别控制的灵活性和在复杂生成任务中的卓越质量而取得了显著的成功。典型的cGANs通过分解为两个较简单的子问题——边缘匹配和条件匹配——来解决联合分布匹配问题。本文提出了一种简单但有效的训练方法,即选择性聚焦学习,该方法强制判别器和生成器快速学习每个类别的简单样本,同时保持多样性。我们的关键思想是针对每个小批量数据,选择性地应用条件匹配和联合匹配。具体来说,我们首先选择使用判别器输出(真实样本和生成样本)的条件项排序后得分最高的样本。然后,我们使用仅条件匹配优化所选样本,而使用联合匹配优化其他样本。通过我们的玩具实验发现,由于判别器的内容感知优化,仅对某些样本应用条件匹配效果最佳。我们在ImageNet(64x64和128x128)、CIFAR-10、CIFAR-100数据集以及高斯混合、噪声标签设置下进行了实验,结果表明,所提出的方法在10次独立试验中显著(在FID方面高达35.18%)改善了所有指标。代码可在https://github.com/pnu-cvsp/Enhancing-Stability-in-Training-Conditional-GAN-via-Selective-Data-Matching获取。