用GAN生成城市像素数据,大幅提升卫星图像分类准确率
Application of Generative Adversarial Network (GAN) for Synthetic Training Data Creation to improve performance of ANN Classifier for extracting Built-Up pixels from Landsat Satellite Imagery
- 构建简易GAN生成真实分布的建筑像素数据
- 分类准确率从93.31%提升至99.83%,一致性系数达0.9958
- 适合遥感图像小样本分类任务的研究者参考
基于低分辨率Landsat影像进行像素级分类训练时,由于单一类别纯像素数量有限,常导致神经网络难以达到预期精度。为解决这一问题,本文提出一种基于生成对抗网络(GAN)的合成数据生成方法,通过训练原始建筑像素样本,生成与原数据分布一致的合成像素。采用非参数柯尔莫哥洛夫-斯米诺夫检验和球散度分布等性检验,验证生成像素在各波段的边缘与联合分布与真实数据无显著差异。实验表明,将生成数据加入原始训练集后,人工神经网络(ANN)分类器的整体精度从0.9331提升至0.9983,卡帕系数从0.8277增至0.9958,性能持续显著改善。
原文摘要 · Abstract (English)
Training a neural network for pixel based classification task using low resolution Landsat images is difficult as the size of the training data is usually small due to less number of available pixels that represent a single class without any mixing with other classes. Due to this scarcity of training data, neural network may not be able to attain expected level of accuracy. This limitation could be overcome using a generative network that aims to generate synthetic data having the same distribution as the sample data with which it is trained. In this work, we have proposed a methodology for improving the performance of ANN classifier to identify built-up pixels in the Landsat$7$ image with the help of developing a simple GAN architecture that could generate synthetic training pixels when trained using original set of sample built-up pixels. To ensure that the marginal and joint distributions of all the bands corresponding to the generated and original set of pixels are indistinguishable, non-parametric Kolmogorov Smirnov Test and Ball Divergence based Equality of Distributions Test have been performed respectively. It has been observed that the overall accuracy and kappa coefficient of the ANN model for built-up classification have continuously improved from $0.9331$ to $0.9983$ and $0.8277$ to $0.9958$ respectively, with the inclusion of generated sets of built-up pixels to the original one.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。