Generative adversarial network augmentation for solving the training data imbalance problem in crop classification

被引:5
|
作者
Shumilo, Leonid [1 ]
Okhrimenko, Anton [2 ]
Kussul, Nataliia [2 ,3 ,4 ]
Drozd, Sofiia [2 ]
Shkalikov, Oleh [2 ]
机构
[1] Univ Maryland, Dept Geog Sci, College Pk, MD 20742 USA
[2] Natl Tech Univ Ukraine, Igor Sikorsky Kyiv Polytech Inst, Dept Math Modelling & Data Anal, Kiev, Ukraine
[3] Natl Acad Sci Ukraine, Space Res Inst, Dept Space Informat Technol & Syst, Kiev, Ukraine
[4] State Space Agcy Ukraine, Kiev, Ukraine
基金
新加坡国家研究基金会;
关键词
Crop Classification; Generative Adversarial Networks; Training Data Generation; Data Set Imbalance; U-Net;
D O I
10.1080/2150704X.2023.2275551
中图分类号
TP7 [遥感技术];
学科分类号
081102 ; 0816 ; 081602 ; 083002 ; 1404 ;
摘要
Deep learning models offer great potential for advancing land monitoring using satellite data. However, they face challenges due to imbalanced real-world data distributions of land cover and crop types, hindering scalability and transferability. This letter presents a novel data augmentation method employing Generative Adversarial Neural Networks (GANs) with pixel-to-pixel transformation (pix2pix). This approach generates realistic synthetic satellite images with artificial ground truth masks, even for rare crop class distributions. It enables the creation of additional minority class samples, enhancing control over training data balance and outperforming traditional augmentation methods. Implementing this method improved the overall map accuracy (OA) and intersection over union (IoU) by 1.5% and 2.1%, while average crop type classes' user accuracy (UA) and producer accuracies (PA), as well as IoU, were improved by 11.2%, 6.4% and 10.2%.
引用
收藏
页码:1131 / 1140
页数:10
相关论文
共 50 条
  • [11] Features kept generative adversarial network data augmentation strategy for hyperspectral image classification
    Zhang, Mingyang
    Wang, Zhaoyang
    Wang, Xiangyu
    Gong, Maoguo
    Wu, Yue
    Li, Hao
    PATTERN RECOGNITION, 2023, 142
  • [12] Cancer classification with data augmentation based on generative adversarial networks
    Wei, Kaimin
    Li, Tianqi
    Huang, Feiran
    Chen, Jinpeng
    He, Zefan
    FRONTIERS OF COMPUTER SCIENCE, 2022, 16 (02)
  • [13] Cancer classification with data augmentation based on generative adversarial networks
    Kaimin Wei
    Tianqi Li
    Feiran Huang
    Jinpeng Chen
    Zefan He
    Frontiers of Computer Science, 2022, 16
  • [14] Cancer classification with data augmentation based on generative adversarial networks
    WEI Kaimin
    LI Tianqi
    HUANG Feiran
    CHEN Jinpeng
    HE Zefan
    Frontiers of Computer Science, 2022, 16 (02)
  • [15] Generalization classification regularization generative adversarial network for machinery fault diagnostics under data imbalance
    Lin, Cuiying
    Kong, Yun
    Huan, Guoyu
    Han, Qinkai
    Dong, Mingming
    Liu, Hui
    Chu, Fulei
    RELIABILITY ENGINEERING & SYSTEM SAFETY, 2025, 256
  • [16] Data Augmentation Computing Model Based on Generative Adversarial Network
    Weng, Yu
    Zhou, Haiwen
    IEEE ACCESS, 2019, 7 : 64223 - 64233
  • [17] Data augmentation in hotspot detection based on generative adversarial network
    Wang, Shuhan
    Gai, Tianyang
    Qu, Tong
    Ma, Bojie
    Su, Xiaojing
    Dong, Lisong
    Zhang, Libin
    Xu, Peng
    Su, Yajuan
    Wei, Yayi
    JOURNAL OF MICRO-NANOPATTERNING MATERIALS AND METROLOGY-JM3, 2021, 20 (03):
  • [18] Robust Data Augmentation Generative Adversarial Network for Object Detection
    Lee, Hyungtak
    Kang, Seongju
    Chung, Kwangsue
    SENSORS, 2023, 23 (01)
  • [19] Generative Adversarial Network for Class-Conditional Data Augmentation
    Lee, Jeongmin
    Yoon, Younkyoung
    Kwon, Junseok
    APPLIED SCIENCES-BASEL, 2020, 10 (23): : 1 - 15
  • [20] Data Augmentation Based on Generative Adversarial Networks for Endoscopic Image Classification
    Park, Hyun-Cheol
    Hong, In-Pyo
    Poudel, Sahadev
    Choi, Chang
    IEEE ACCESS, 2023, 11 : 49216 - 49225