Training Generative Adversarial Networks with Limited Data

被引:0
|
作者
Karras, Tero [1 ]
Aittala, Miika [1 ]
Hellsten, Janne [1 ]
Laine, Samuli [1 ]
Lehtinen, Jaakko [1 ,2 ]
Aila, Timo [1 ]
机构
[1] NVIDIA, Santa Clara, CA 95051 USA
[2] Aalto Univ, Espoo, Finland
关键词
D O I
暂无
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Training generative adversarial networks (GAN) using too little data typically leads to discriminator overfitting, causing training to diverge. We propose an adaptive discriminator augmentation mechanism that significantly stabilizes training in limited data regimes. The approach does not require changes to loss functions or network architectures, and is applicable both when training from scratch and when fine-tuning an existing GAN on another dataset. We demonstrate, on several datasets, that good results are now possible using only a few thousand training images, often matching StyleGAN2 results with an order of magnitude fewer images. We expect this to open up new application domains for GANs. We also find that the widely used CIFAR-10 is, in fact, a limited data benchmark, and improve the record FID from 5.59 to 2.42.
引用
收藏
页数:11
相关论文
共 50 条
  • [1] GenCo: Generative Co-training for Generative Adversarial Networks with Limited Data
    Cui, Kaiwen
    Huang, Jiaxing
    Luo, Zhipeng
    Zhang, Gongjie
    Zhan, Fangneng
    Lu, Shijian
    [J]. THIRTY-SIXTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FOURTH CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE / THE TWELVETH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2022, : 499 - 507
  • [2] DeLiGAN : Generative Adversarial Networks for Diverse and Limited Data
    Gurumurthy, Swaminathan
    Sarvadevabhatla, Ravi Kiran
    Babu, R. Venkatesh
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 4941 - 4949
  • [3] Regularizing Generative Adversarial Networks under Limited Data
    Tseng, Hung-Yu
    Jiang, Lu
    Liu, Ce
    Yang, Ming-Hsuan
    Yang, Weilong
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 7917 - 7927
  • [4] Generative Adversarial Networks for Black-Box API Attacks with Limited Training Data
    Shi, Yi
    Sagduyu, Yalin E.
    Davaslioglu, Kemal
    Li, Jason H.
    [J]. 2018 IEEE INTERNATIONAL SYMPOSIUM ON SIGNAL PROCESSING AND INFORMATION TECHNOLOGY (ISSPIT), 2018, : 453 - 458
  • [5] Improving the training performance of generative adversarial networks with limited data: Application to the generation of geological models
    Ranazzi, Paulo Henrique
    Luo, Xiaodong
    Sampaio, Marcio Augusto
    [J]. Computers and Geosciences, 2024, 193
  • [6] Exploring generative adversarial networks and adversarial training
    Sajeeda, Afia
    Hossain, B M Mainul
    [J]. International Journal of Cognitive Computing in Engineering, 2022, 3 : 78 - 89
  • [7] TRAINING GENERATIVE ADVERSARIAL NETWORKS WITH WEIGHTS
    Pantazis, Yannis
    Paul, Dipjyoti
    Fasoulakis, Michail
    Stylianou, Yannis
    [J]. 2019 27TH EUROPEAN SIGNAL PROCESSING CONFERENCE (EUSIPCO), 2019,
  • [8] Generative Adversarial Networks for Augmenting Training Data of Microscopic Cell Images
    Baniukiewicz, Piotr
    Lutton, E. Josiah
    Collier, Sharon
    Bretschneider, Till
    [J]. FRONTIERS IN COMPUTER SCIENCE, 2019, 1
  • [9] Can Generative Adversarial Networks help to overcome the limited data problem in segmentation
    Heilemann, Gerd
    Matthewman, Mark
    Kuess, Peter
    Goldner, Gregor
    Widder, Joachim
    Georg, Dietmar
    Zimmermann, Lukas
    [J]. ZEITSCHRIFT FUR MEDIZINISCHE PHYSIK, 2022, 32 (03): : 361 - 368
  • [10] Regularizing Label-Augmented Generative Adversarial Networks Under Limited Data
    Hou, Liang
    [J]. IEEE ACCESS, 2023, 11 (28966-28976) : 28966 - 28976