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 条
  • [1] ECGAN: An Improved Conditional Generative Adversarial Network With Edge Detection to Augment Limited Training Data for the Classification of Remote Sensing Images With High Spatial Resolution
    Sui, Baikai
    Jiang, Tao
    Zhang, Zhen
    Pan, Xinliang
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2021, 14 : 1311 - 1325
  • [2] Generative adversarial network augmentation for solving the training data imbalance problem in crop classification
    Shumilo, Leonid
    Okhrimenko, Anton
    Kussul, Nataliia
    Drozd, Sofiia
    Shkalikov, Oleh
    REMOTE SENSING LETTERS, 2023, 14 (11) : 1131 - 1140
  • [3] Generative adversarial network based synthetic data training model for lightweight convolutional neural networks
    Ishfaq Hussain Rather
    Sushil Kumar
    Multimedia Tools and Applications, 2024, 83 : 6249 - 6271
  • [4] Generative adversarial network based synthetic data training model for lightweight convolutional neural networks
    Rather, Ishfaq Hussain
    Kumar, Sushil
    MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 83 (2) : 6249 - 6271
  • [5] Classification of Optical Coherence Tomography Images Using Generative Adversarial Networks
    Aghaei, S. M. H. Seyed
    Rashno, A.
    Fadaei, S.
    INTERNATIONAL JOURNAL OF ENGINEERING, 2025, 38 (02): : 389 - 399
  • [6] Improving the performance of multimedia pedestrian classification with images synthesized using a deep convolutional generative adversarial network
    Chuho Yi
    Jungwon Cho
    Multimedia Tools and Applications, 2021, 80 : 34697 - 34712
  • [7] Improving the performance of multimedia pedestrian classification with images synthesized using a deep convolutional generative adversarial network
    Yi, Chuho
    Cho, Jungwon
    MULTIMEDIA TOOLS AND APPLICATIONS, 2021, 80 (26-27) : 34697 - 34712
  • [8] Dehaze of Cataractous Retinal Images Using an Unpaired Generative Adversarial Network
    Luo, Yuhao
    Chen, Kun
    Liu, Lei
    Liu, Jicheng
    Mao, Jianbo
    Ke, Genjie
    Sun, Mingzhai
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2020, 24 (12) : 3374 - 3383
  • [9] Generative Adversarial Network Design for Data Augmentation for Copro-parasite Images
    Gutierrez-Velazquez, Miguel
    Chacon-Murguia, Mario, I
    Ramirez-Quintana, Juan A.
    Arzate-Quintana, Carlos
    Corral-Saenz, Alma D.
    2021 IEEE INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND APPLIED NETWORK TECHNOLOGIES (ICMLANT II), 2021, : 120 - 124
  • [10] Poststack Seismic Data Compression Using a Generative Adversarial Network
    dos Santos Ribeiro, Kevyn Swhants
    Schiavon, Ana Paula
    Navarro, Joao Paulo
    Vieira, Marcelo Bernardes
    Villela, Saulo Moraes
    Cruz E Silva, Pedro Mario
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19