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 条
  • [21] Cross-Scene Classification of Hyperspectral Images via Generative Adversarial Network in Latent Space
    Yang, Yahan
    Xu, Yang
    Wu, Zebin
    Wang, Biqi
    Wei, Zhihui
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [22] ARTIFICIAL BANDWIDTH EXTENSION USING A CONDITIONAL GENERATIVE ADVERSARIAL NETWORK WITH DISCRIMINATIVE TRAINING
    Sautter, Jonas
    Faubel, Friedrich
    Buck, Markus
    Schmidt, Gerhard
    2019 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2019, : 7005 - 7009
  • [23] SolarGAN: Multivariate Solar Data Imputation Using Generative Adversarial Network
    Zhang, Wenjie
    Luo, Yonghong
    Zhang, Ying
    Srinivasan, Dipti
    IEEE TRANSACTIONS ON SUSTAINABLE ENERGY, 2021, 12 (01) : 743 - 746
  • [24] Data augmentation using generative adversarial networks for images and biomarkers in medicine and neuroscience
    Yahaya, Maizan Syamimi Meor
    Teo, Jason
    FRONTIERS IN APPLIED MATHEMATICS AND STATISTICS, 2023, 9
  • [25] Surgical Tool Segmentation Using Generative Adversarial Networks With Unpaired Training Data
    Zhang, Zhongkai
    Rosa, Benoit
    Nageotte, Florent
    IEEE ROBOTICS AND AUTOMATION LETTERS, 2021, 6 (04) : 6266 - 6273
  • [26] Transferemble: a classification method for the detection of fake satellite images created with deep convolutional generative adversarial network
    Surucu, Selim
    Diri, Banu
    JOURNAL OF ELECTRONIC IMAGING, 2023, 32 (04)
  • [27] SIGAN: Spectral Index Generative Adversarial Network for Data Augmentation in Multispectral Remote Sensing Images
    Singh, Abhishek
    Bruzzone, Lorenzo
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [28] Efficient Classification of Imbalanced Natural Disasters Data Using Generative Adversarial Networks for Data Augmentation
    Eltehewy, Rokaya
    Abouelfarag, Ahmed
    Saleh, Sherine Nagy
    ISPRS INTERNATIONAL JOURNAL OF GEO-INFORMATION, 2023, 12 (06)
  • [29] Classification of Breast Cancer Histopathological Images Using Discriminative Patches Screened by Generative Adversarial Networks
    Man, Rui
    Yang, Ping
    Xu, Bowen
    IEEE ACCESS, 2020, 8 : 155362 - 155377
  • [30] Few-Shot Hyperspectral Image Classification Using Relational Generative Adversarial Network
    Wei, Wei
    Tong, Lei
    Guo, Baoqing
    Zhou, Jun
    Xiao, Chuangbai
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62