DM-GAN: CNN hybrid vits for training GANs under limited data

被引:2
|
作者
Yan, Longquan [1 ]
Yan, Ruixiang [2 ]
Chai, Bosong [3 ]
Geng, Guohua [1 ]
Zhou, Pengbo [4 ]
Gao, Jian [1 ]
机构
[1] Northwest Univ, Sch Informat Sci & Technol, Xian 710119, Peoples R China
[2] Tianjin Univ, Coll Intelligence & Comp, Tianjin 300350, Peoples R China
[3] Zhejiang Univ, Sch Comp Sci & Technol, Hangzhou 310013, Peoples R China
[4] Beijing Normal Univ, Sch Informat Sci & Technol, Beijing 100875, Peoples R China
基金
中国国家自然科学基金;
关键词
GAN; Few-shot; Vision transformer; Proprietary artifact image;
D O I
10.1016/j.patcog.2024.110810
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Generative adversarial network (GAN) training demands substantial data and computational resources. This paper aims to explore an economical approach for generating novel images with limited image data, addressing the challenge of data scarcity. Our contributions involve resolving the few-shot image generation challenge through the development of an unsupervised hybrid generative adversarial network named DM-GAN. We introduce a lightweight hybrid module (DC-Vit) comprising convolution and visual transformation, merging local and global features to enhance image perception, expressiveness, and ensure stable image generation. Additionally, a multi-scale adaptive skip connection module is incorporated to effectively mitigate the feature loss problem arising from inter-layer jumps, thereby producing more complete and regular images. To enhance the texture learning process and improve the quality and realism of synthesized images, we integrate the gray conjugate matrix into the loss function. Empirical evaluations are conducted on small sample datasets at various resolutions, including publicly accessible collections of art paintings, real-life photographs, and proprietary artifact image datasets. The experimental results unequivocally demonstrate the qualitative and quantitative superiority of our model over existing methods, underscoring its efficacy and robustness.
引用
收藏
页数:9
相关论文
共 50 条
  • [1] FreGAN: Exploiting Frequency Components for Training GANs under Limited Data
    Yang, Mengping
    Wang, Zhe
    Chi, Ziqiu
    Zhang, Yanbing
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 35 (NEURIPS 2022), 2022,
  • [2] ScoreMix: A Scalable Augmentation Strategy for Training GANs With Limited Data
    Cao, Jie
    Luo, Mandi
    Yu, Junchi
    Yang, Ming-Hsuan
    He, Ran
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (07) : 8920 - 8935
  • [3] Re-GAN: Data-Efficient GANs Training via Architectural Reconfiguration
    Saxena, Divya
    Cao, Jiannong
    Xu, Jiahao
    Kulshrestha, Tarun
    2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 16230 - 16240
  • [4] Optimized Fault Diagnosis Algorithm under GAN and CNN Hybrid Model
    Zhu, Xiaobo
    Ye, Yunlong
    WIRELESS COMMUNICATIONS & MOBILE COMPUTING, 2022, 2022
  • [5] D3T- GAN: Data-Dependent Domain Transfer GANs for Image Generation with Limited Data
    Wu, Xintian
    Wang, Huanyu
    Wu, Yiming
    Li, Xi
    ACM TRANSACTIONS ON MULTIMEDIA COMPUTING COMMUNICATIONS AND APPLICATIONS, 2023, 19 (04)
  • [6] Learnable GAN Regularization for Improving Training Stability in Limited Data Paradigm
    Singh, Nakul (nakul692k@gmail.com), 1600, Springer Science and Business Media Deutschland GmbH (2010 CCIS):
  • [7] Deceive D: Adaptive Pseudo Augmentation for GAN Training with Limited Data
    Jiang, Liming
    Dai, Bo
    Wu, Wayne
    Loy, Chen Change
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 34 (NEURIPS 2021), 2021, 34
  • [8] Learnable GAN Regularization for Improving Training Stability in Limited Data Paradigm
    Singh, Nakul
    Sandhan, Tushar
    COMPUTER VISION AND IMAGE PROCESSING, CVIP 2023, PT II, 2024, 2010 : 542 - 554
  • [9] DigGAN: Discriminator gradIent Gap Regularization for GAN Training with Limited Data
    Fang, Tiantian
    Sun, Ruoyu
    Schwing, Alex
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 35 (NEURIPS 2022), 2022,
  • [10] Application of GAN for Reducing Data Imbalance under Limited Dataset
    Adke, Gaurav
    PROCEEDINGS OF THE 17TH INTERNATIONAL JOINT CONFERENCE ON COMPUTER VISION, IMAGING AND COMPUTER GRAPHICS THEORY AND APPLICATIONS (VISAPP), VOL 4, 2022, : 60 - 68