The Performance Improvement of U-Net Model for Landcover Semantic Segmentation through Data Augmentation

被引:8
作者
Baek, Won-Kyung [1 ]
Lee, Moung-Jin [2 ]
Jung, Hyung-Sup [3 ,4 ]
机构
[1] Univ Seoul, Dept Geoinformat, Seoul, South Korea
[2] Korea Environm Inst, Ctr Environm Data Strategy, Sejong, South Korea
[3] Univ Seoul, Dept Geoinformat, Seoul, South Korea
[4] Univ Seoul, Dept Smart Cities, Seoul, South Korea
关键词
Landcover; Semantic segmentation; U-Net; Data augmentation; CLASSIFICATION;
D O I
10.7780/kjrs.2022.38.6.2.8
中图分类号
TP7 [遥感技术];
学科分类号
081102 ; 0816 ; 081602 ; 083002 ; 1404 ;
摘要
Recently, a number of deep-learning based land cover segmentation studies have been introduced. Some studies denoted that the performance of land cover segmentation deteriorated due to insufficient training data. In this study, we verified the improvement of land cover segmentation performance through data augmentation. U-Net was implemented for the segmentation model. And 2020 satellite-derived landcover dataset was utilized for the study data. The pixel accuracies were 0.905 and 0.923 for U-Net trained by original and augmented data respectively. And the mean F1 scores of those models were 0.720 and 0.775 respectively, indicating the better performance of data augmentation. In addition, F1 scores for building, road, paddy field, upland field, forest, and unclassified area class were 0.770, 0.568, 0.433, 0.455, 0.964, and 0.830 for the U-Net trained by original data. It is verified that data augmentation is effective in that the F1 scores of every class were improved to 0.838, 0.660, 0.791, 0.530, 0.969, and 0.860 respectively. Although, we applied data augmentation without considering class balances, we find that data augmentation can mitigate biased segmentation performance caused by data imbalance problems from the comparisons between the performances of two models. It is expected that this study would help to prove the importance and effectiveness of data augmentation in various image processing fields.
引用
收藏
页码:1663 / 1676
页数:14
相关论文
共 29 条
  • [1] AI Hub, 2020, 2020 Satellite-derived landcover dataset
  • [2] Baek W.-K., 2022, Phase Unwrapping Using ModifiOU-Net Regression Model: Focusing on Network Structure and Training Data Optimization
  • [3] Classification of Natural and Artificial Forests from KOMPSAT-3/3A/5 Images Using Deep Neural Network
    Baek, Won-Kyung
    Lee, Yong-Suk
    Park, Sung-Hwan
    Jung, Hyung-Sup
    [J]. KOREAN JOURNAL OF REMOTE SENSING, 2021, 37 (06) : 1965 - 1974
  • [4] Performance Comparison of Oil Spill and Ship Classification from X-Band Dual- and Single-Polarized SAR Image Using Support Vector Machine, Random Forest, and Deep Neural Network
    Baek, Won-Kyung
    Jung, Hyung-Sup
    [J]. REMOTE SENSING, 2021, 13 (16)
  • [5] Oil Spill Detection of Kerch Strait in November 2007 from Dual-Polarized TerraSAR-X Image Using Artificial and Convolutional Neural Network Regression Models
    Baek, Won-Kyung
    Jung, Hyung-Sup
    Kim, Daeseong
    [J]. JOURNAL OF COASTAL RESEARCH, 2020, : 137 - 144
  • [6] Baek WK, 2017, KOREAN J REMOTE SENS, V33, P889, DOI 10.7780/kjrs.2017.33.5.3.11
  • [7] Choi D, 2020, Arxiv, DOI arXiv:1910.05446
  • [8] The Unreasonable Effectiveness of Data
    Halevy, Alon
    Norvig, Peter
    Pereira, Fernando
    [J]. IEEE INTELLIGENT SYSTEMS, 2009, 24 (02) : 8 - 12
  • [9] Integrative Data Augmentation with U-Net Segmentation Masks Improves Detection of Lymph Node Metastases in Breast Cancer Patients
    Jin, Yong Won
    Jia, Shuo
    Ashraf, Ahmed Bilal
    Hu, Pingzhao
    [J]. CANCERS, 2020, 12 (10) : 1 - 13
  • [10] Survey on deep learning with class imbalance
    Johnson, Justin M.
    Khoshgoftaar, Taghi M.
    [J]. JOURNAL OF BIG DATA, 2019, 6 (01)