Benefits of using blended generative adversarial network images to augment classification model training data sets

被引:2
|
作者
McCloskey, Benjamin J. [1 ]
Cox, Bruce A. [1 ,3 ]
Champagne, Lance [1 ]
Bihl, Trevor J. [2 ]
机构
[1] Inst Technol, Dept Operat Sci, Wright Patterson AFB, OH USA
[2] Res Lab, Sensors Directorate, Wright Patterson AFB, OH USA
[3] Inst Technol, Dept Operat Sci, 2950 Hobson Way, Wright Patterson AFB, OH 45433 USA
来源
JOURNAL OF DEFENSE MODELING AND SIMULATION-APPLICATIONS METHODOLOGY TECHNOLOGY-JDMS | 2023年
关键词
Generative adversarial networks; GANs; synthetic images; object classification; class imbalance; training data sets;
D O I
10.1177/15485129231170225
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Object detection algorithms have reached nearly superhuman levels within the last decade; however, these algorithms require large diverse training data sets to ensure their operational performance matches performance demonstrated during testing. The collection and human labeling of such data sets can be expensive and, in some cases, such as Intelligence, Surveillance and Reconnaissance of rare events it may not even be feasible. This research proposes a novel method for creating additional variability within the training data set by utilizing multiple models of generative adversarial networks producing both high- and low-quality synthetic images of vehicles and inserting those images alongside images of real vehicles into real backgrounds. This research demonstrates a 17.90% increase in mean absolute percentage error, on average, compared to the YOLOv4-Tiny Model trained on the original non-augmented training set as well as a 14.44% average improvement in the average intersection over union rate. In addition, our research adds to a small, but growing, body of literature indicating that the inclusion of low-quality images into training data sets is beneficial to the performance of computer vision models.
引用
收藏
页数:14
相关论文
共 50 条
  • [31] 3-D Hybrid CNN Combined With 3-D Generative Adversarial Network for Wetland Classification With Limited Training Data
    Jamali, Ali
    Mahdianpari, Masoud
    Mohammadimanesh, Fariba
    Brisco, Brian
    Salehi, Bahram
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2022, 15 : 8095 - 8108
  • [32] Training data independent image registration using generative adversarial networks and domain adaptation
    Mahapatra, Dwarikanath
    Ge, Zongyuan
    PATTERN RECOGNITION, 2020, 100
  • [33] CT synthesis from MR images using frequency attention conditional generative adversarial network
    Wei, Kexin
    Kong, Weipeng
    Liu, Liheng
    Wang, Jian
    Li, Baosheng
    Zhao, Bo
    Li, Zhenjiang
    Zhu, Jian
    Yu, Gang
    COMPUTERS IN BIOLOGY AND MEDICINE, 2024, 170 (170)
  • [34] Synthetic attack data generation model applying generative adversarial network for intrusion detection
    Kumar, Vikash
    Sinha, Ditipriya
    COMPUTERS & SECURITY, 2023, 125
  • [35] Attention-Enhanced Voice Portrait Model Using Generative Adversarial Network
    Mao, Jingyi
    Zhou, Yuchen
    Wang, Yifan
    Li, Junyu
    Liu, Ziqing
    Bu, Fanliang
    CMC-COMPUTERS MATERIALS & CONTINUA, 2024, 79 (01): : 837 - 855
  • [36] An improved multi-scale gradient generative adversarial network for enhancing classification of colorectal cancer histological images
    Jiang, Liwen
    Huang, Shuting
    Luo, Chaofan
    Zhang, Jiangyu
    Chen, Wenjing
    Liu, Zhenyu
    FRONTIERS IN ONCOLOGY, 2023, 13
  • [37] Time-series Anonymization of Tabular Health Data using Generative Adversarial Network
    Hashemi, Atiye Sadat
    Etminani, Kobra
    Soliman, Amira
    Hamed, Omar
    Lundstrom, Jens
    2023 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, IJCNN, 2023,
  • [38] Data Augmentation for X-Ray Prohibited Item Images Using Generative Adversarial Networks
    Yang, Jinfeng
    Zhao, Zihao
    Zhang, Haigang
    Shi, Yihua
    IEEE ACCESS, 2019, 7 : 28894 - 28902
  • [39] Prediction of Typhoon Track and Intensity Using a Generative Adversarial Network With Observational and Meteorological Data
    Ruettgers, Mario
    Jeon, Soohwan
    Lee, Sangseung
    You, Donghyun
    IEEE ACCESS, 2022, 10 : 48434 - 48446
  • [40] Data Augmentation for Imbalanced HRRP Recognition Using Deep Convolutional Generative Adversarial Network
    Song, Yiheng
    Li, Yang
    Wang, Yanhua
    Hu, Cheng
    IEEE ACCESS, 2020, 8 : 201686 - 201695