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 条
  • [41] Data Augmentation using Generative Adversarial Networks for Pneumonia classification in chest X-rays
    Bhagat, Vedant
    Bhaumik, Swapnil
    2019 FIFTH INTERNATIONAL CONFERENCE ON IMAGE INFORMATION PROCESSING (ICIIP 2019), 2019, : 574 - 579
  • [42] Implementing Data Augmentation Techniques Using Conditional Generative Adversarial Network-Based upon Chest X-Ray Images
    Sankaranarayanan, Suresh
    Khare, Akshat
    INTELLIGENT SYSTEMS AND APPLICATIONS, VOL 4, INTELLISYS 2024, 2024, 1068 : 531 - 541
  • [43] Generative adversarial network-based deep learning approach in classification of retinal conditions with optical coherence tomography images
    Ling-Chun Sun
    Shu-I. Pao
    Ke-Hao Huang
    Chih-Yuan Wei
    Ke-Feng Lin
    Ping-Nan Chen
    Graefe's Archive for Clinical and Experimental Ophthalmology, 2023, 261 : 1399 - 1412
  • [44] Generative adversarial network-based deep learning approach in classification of retinal conditions with optical coherence tomography images
    Sun, Ling-Chun
    Pao, Shu-I.
    Huang, Ke-Hao
    Wei, Chih-Yuan
    Lin, Ke-Feng
    Chen, Ping-Nan
    GRAEFES ARCHIVE FOR CLINICAL AND EXPERIMENTAL OPHTHALMOLOGY, 2023, 261 (05) : 1399 - 1412
  • [45] Conditional Generative Adversarial Network-Based Training Sample Set Improvement Model for the Semantic Segmentation of High-Resolution Remote Sensing Images
    Pan, Xin
    Zhao, Jian
    Xu, Jun
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (09): : 7854 - 7870
  • [46] An Enhanced Model for Detecting and Classifying Emergency Vehicles Using a Generative Adversarial Network (GAN)
    Shatnawi, Mo'ath
    Younes, Maram Bani
    VEHICLES, 2024, 6 (03): : 1114 - 1139
  • [47] A novel semi-supervised method for classification of power quality disturbance using generative adversarial network
    Jian, Xianzhong
    Wang, Xutao
    Jian, Xianzhong (jianxz@usst.edu.cn), 2021, IOS Press BV (40) : 3875 - 3885
  • [48] MR to CT synthesis with multicenter data in the pelvic area using a conditional generative adversarial network
    Boni, Kevin N. D. Brou
    Klein, John
    Vanquin, Ludovic
    Wagner, Antoine
    Lacornerie, Thomas
    Pasquier, David
    Reynaert, Nick
    PHYSICS IN MEDICINE AND BIOLOGY, 2020, 65 (07)
  • [49] Data Augmentation Using Generative Adversarial Networks for Multi-Class Segmentation of Lung Confocal IF Images
    Katsuma, Daiki
    Kawanaka, Hiroharu
    Prasath, V. B. Surya
    Aronow, Bruce J.
    JOURNAL OF ADVANCED COMPUTATIONAL INTELLIGENCE AND INTELLIGENT INFORMATICS, 2022, 26 (02) : 138 - 146
  • [50] Augmenting Seismic Data Using Generative Adversarial Network for Low-Cost MEMS Sensors
    Wu, Aming
    Shin, Juyong
    Ahn, Jae-Kwang
    Kwon, Young-Woo
    IEEE ACCESS, 2021, 9 : 167140 - 167153