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 条
  • [1] Data Augmentation for Intra-class Imbalance with Generative Adversarial Network
    Hase, Natsuki
    Ito, Seiya
    Kaneko, Naoshi
    Sumi, Kazuhiko
    FOURTEENTH INTERNATIONAL CONFERENCE ON QUALITY CONTROL BY ARTIFICIAL VISION, 2019, 11172
  • [2] Improving the Recognition Accuracy by Solving the Inherent Data Imbalance Problem of ErrP with Generative Adversarial Network
    Jia, Yaguang
    Tao, Tangfei
    Xu, Guanghua
    Li, Min
    Zhang, Sicong
    Han, Chengcheng
    Wu, Qingqiang
    Pei, Jinju
    Lv, Xiaoqing
    Shi, Zhilei
    2023 20TH INTERNATIONAL CONFERENCE ON UBIQUITOUS ROBOTS, UR, 2023, : 228 - 232
  • [3] Data Augmentation Using Generative Adversarial Network for Environmental Sound Classification
    Madhu, Aswathy
    Kumaraswamy, Suresh
    2019 27TH EUROPEAN SIGNAL PROCESSING CONFERENCE (EUSIPCO), 2019,
  • [4] LATENT PRESERVING GENERATIVE ADVERSARIAL NETWORK FOR IMBALANCE CLASSIFICATION
    Dam, Tanmoy
    Ferdaus, Md Meftahul
    Pratama, Mahardhika
    Anavatti, Sreenatha G.
    Jayavelu, Senthilnath
    Abbass, Hussein
    2022 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, ICIP, 2022, : 3712 - 3716
  • [5] Generative Adversarial Network for Data Augmentation and Substitution
    Stankovic, Marko
    Bacanin, Nebojsa
    Zivkovic, Miodrag
    Jovanovic, Luka
    Sarac, Marko
    Antonijevic, Milos
    2024 ZOOMING INNOVATION IN CONSUMER TECHNOLOGIES CONFERENCE, ZINC 2024, 2024, : 7 - 12
  • [6] Experimental Assessment of the Performance of Data Augmentation with Generative Adversarial Networks in the Image Classification Problem
    Karadag, Ozge Oztimur
    Cicek, Ozlem Erdas
    2019 INNOVATIONS IN INTELLIGENT SYSTEMS AND APPLICATIONS CONFERENCE (ASYU), 2019, : 48 - 51
  • [7] Latent Posterior Based Generative Adversarial Network for Imbalance Classification
    He, Xinlin
    Cai, Meng
    Li, Jianxun
    PROCEEDINGS OF THE 32ND 2020 CHINESE CONTROL AND DECISION CONFERENCE (CCDC 2020), 2020, : 3449 - 3454
  • [8] Generative adversarial network based data augmentation to improve cervical cell classification model
    Yu, Suxiang
    Zhang, Shuai
    Wang, Bin
    Dun, Hua
    Xu, Long
    Huang, Xin
    Shi, Ermin
    Feng, Xinxing
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2021, 18 (02) : 1740 - 1752
  • [9] BrainNetGAN: Data Augmentation of Brain Connectivity Using Generative Adversarial Network for Dementia Classification
    Li, Chao
    Wei, Yiran
    Chen, Xi
    Schonlieb, Carola-Bibiane
    DEEP GENERATIVE MODELS, AND DATA AUGMENTATION, LABELLING, AND IMPERFECTIONS, 2021, 13003 : 103 - 111
  • [10] Data Augmentation using Generative Adversarial Network for Gastrointestinal Parasite Microscopy Image Classification
    Pacompia Machaca, Mila Yoselyn
    Mayta Rosas, Milagros Lizet
    Castro-Gutierrez, Eveling
    Talavera Diaz, Henry Abraham
    Vasquez Huerta, Victor Luis
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2020, 11 (11) : 765 - 771