中文

通过二值化表示熵(BRE)正则化改进GAN训练

机器学习 2018-05-11 v1 机器学习

摘要

我们提出一种新颖的正则化器以改进生成对抗网络(GANs)的训练。其动机在于,当判别器D以恰当方式分散其模型容量时,给予生成器G的学习信号更具信息性与多样性。这反过来帮助G更好地探索并发现真实数据流形,同时避免因D的错误外推而导致的大幅不稳定跳跃。我们的正则化器通过鼓励D的选定内部层上的二值激活模式具有高联合熵,来引导修正线性判别器D更好地分配其模型容量。在合成数据与真实数据集上的实验结果表明,GAN训练的稳定性与收敛速度得到改善,且样本质量更高。该方法在半监督学习中还带来了更高的分类准确率。

关键词

引用

@article{arxiv.1805.03644,
  title  = {Improving GAN Training via Binarized Representation Entropy (BRE) Regularization},
  author = {Yanshuai Cao and Gavin Weiguang Ding and Kry Yik-Chau Lui and Ruitong Huang},
  journal= {arXiv preprint arXiv:1805.03644},
  year   = {2018}
}

备注

Published as a conference paper at the 6th International Conference on Learning Representations (ICLR 2018)