Generative adversarial networks for realistic synthesis of hyperspectral samples
Abstract
This work addresses the scarcity of annotated hyperspectral data required to train deep neural networks. Especially, we investigate generative adversarial networks and their application to the synthesis of consistent labeled spectra. By training such networks on public datasets, we show that these models are not only able to capture the underlying distribution, but also to generate genuine-looking and physically plausible spectra. Moreover, we experimentally validate that the synthetic samples can be used as an effective data augmentation strategy. We validate our approach on several public hyperspectral datasets using a variety of deep classifiers.
Generative adversarial networks for realistic synthesis of hyperspectral samples
Nicolas Audebert ^{1, 2}, Bertrand Le Saux ^{1}, Sébastien Lefèvre ^{2} ^{†}^{†}thanks: N. Audebert’s work is funded by the Total-ONERA project NAOMI. |
^{1} ONERA, The French Aerospace Lab, F-91761 Palaiseau, France |
^{2} Univ. Bretagne-Sud, UMR 6074, IRISA, F-56000 Vannes, France |
Index Terms— hyperspectral image classification, generative models, deep learning, data augmentation.
1 Introduction
Data augmentation consists of introducing unobserved samples into the optimization process of a statistical model [1]. Since the reintroduction of Convolutional Neural Networks (CNN) for image classification [2], this practice has been critical to avoid overfitting of deep networks. Therefore, as annotated hyperspectral data is scarce, overfitting is an even more common pitfall compared to multimedia image processing. Although recent efforts have been made to use CNN for hyperspectral image classification [3, 4, 5, 6], successes are limited to small datasets that do not leverage the generalization capacity of deep networks.
To this end, recent works have started to investigate data augmentation as a way to artificially enlarge the quantity of annotated samples. For example, [7] suggested a model of relighting to simulate the same hyperspectral pixel under different illuminations. With a more data-driven approach, [8] introduced a label propagation strategy to incorporate observed but unlabeled samples to the training set. However, these methods require either unlabeled samples or physics-related assumptions and modeling. On the other hand, [9] introduced generative models in hyperspectral image processing by using variational autoencoders to find the endmembers composition of spectral mixtures.
In this work, we introduce a way to artificially synthesize new annotated hyperspectral samples using a purely data-driven approach based on generative adversarial networks [10]. More specifically, we use a GAN to approximate the distribution of the observed hyperspectral samples and to generate new plausible samples that can be used to train deep networks. Our method can exploit both labeled and unlabeled samples and is validated on several datasets covering both aerial and satellite sensors over rural and urban areas.
2 Generative models
The Generative Adversarial Network framework has been introduced in [10]. It uses deep neural networks to approximate an unknown data distribution based on its observations. The idea is to generate new samples of a given distribution by training a generator to map random noise from the latent space to the distribution. However, the target distribution is observed only on some data points and we wish to use the generator to create new data points that also still belong to the underlying distribution. To this end, the generator is trained to approximate the distribution using an adversarial objective function. This function is obtained by introducing another network called the discriminator – or critic. The discriminator learns to infer whether a given sample belongs to the true or the fake distribution, i.e. if the sample belongs to the training set or was created by the generator. The discriminator is trained for a few steps, and then the generator is optimized to fool the critic, i.e. to generate samples that are indistinguishable to the discriminator.
Several flavors of GAN have been introduced that use various objective functions. In this work, we use a generator and a discriminator in the Wasserstein GAN [11] fashion, trained with the gradient penalty from [12]. However, this GAN setup alone only makes it possible to infer a global distribution. In our case, we wish to condition the output of the generator w.r.t. the hyperspectral classes. More specifically, we want our generator to take as an input a random noise and class label, so that it learns to generate a sample beloging to the specified class. This is called conditioning a GAN. To do so, we introduce an additional classifier network . This classifier adds a conditional penalty on the generated distribution by enforcing that the generated spectra are classified in the same class as the conditional label that was given. The whole framework is illustrated in Fig. 1. While and can be trained without label knowledge (i.e. unsupervised training), needs annotated samples to learn.
3 Experimental setup
We train our GAN on four datasets: Pavia University and Pavia Center (urban aerial scenes at 1.3m/px and 103 bands), Indian Pines (agricultural scene at 20m/px and 224 bands) and Botswana datasets (swamps, acquired by the Hyperion sensor at 30m/px with 242 bands). We use atmospheric correction when available and we normalize the reflectance between . As we try to approximate individual hyperspectral pixels with no spatial context, we use 4-layers deep fully connected networks with 512 neurons for , and using the leaky ReLU non-linearity [13]. is followed by a sigmoid activation and outputs a vector which length equals the expected number of bands, while has as many outputs as classes and only has one output.
Optimization is done for all three networks using the RMSprop stochastic gradient descent policy. The GAN is trained for 100,000 iterations, with and being trained twice per iteration before optimizing ’s weights.
4 Spectra analysis
In this section, we aim to investigate the physical plausibility of the synthetic spectra. Especially, we compare the real and fake distributions under several criteria. To this end, we train two GANs on random samples from the Pavia University and Indian Pines datasets.
We can visually assess the quality of the generated spectra by comparing their statistical moments, e.g. plotting the mean spectra and their standard deviation (Fig. 2). As can be seen, the spectral shapes are accurately learned by the GAN. However, we can immediately identify two potential shortcomings. First, the fake mean spectra appear noisier than the true spectra, which means that the GAN overfitted on some specific features that are common to only a subset of the real spectra. Second, the fake standard deviation is lower than the real one, which means that fake spectra are less diverse than the real ones. Both of those signs point to a form of overfitting called mode collapse [14].
To learn more about how this overfitting actually impacts the distribution of the fake samples in the spectral space, we apply Principal Component Analysis (PCA) to map the spectra into a 2D space ( Fig. 3). The clusters formed by the different classes are reproduced truthfully by the synthesized samples. However, there are slight deformations that show that the GAN failed to capture some specificities of each class.
Split | Random (uniform) | Disjoint | ||
---|---|---|---|---|
Train Test | Real | Fake | Real | Fake |
Real | 89.5 | 98.3 | 87.2 | 98.8 |
Fake | 87.8 | 99.2 | 79.4 | 99.9 |
We can form an intuition on how the fake distribution respects the class boundaries of the real spectra by training a linear Support Vector Machine (SVM) on the latter and applying it on the former. The SVM will learn the best separating hyperplanes from the true distribution. Hopefully, these hyperplanes should separate the synthesized spectra with the same accuracy. If the accuracy is significantly lower, then the GAN learned unrealistic samples; if it is significantly higher, then the GAN learned samples too similar to the center of each class cluster, i.e. suffered from mode collapse. Results are presented in Table 1. We consider two train/test splits: either 3% of randomly selected annotated samples or two disjoint halves of the image, i.e. spatially disjoint sets of 50% of the pixels. In the unsupervised setting, we also use the unlabeled samples. As expected, it is easier for the SVM to separate the fake data than the real samples. However, training on fake samples only still reach encouraging accuracies, only between 2% and 8% under the reference real/real setting. This means that although the synthesized spectra are concentrated around the main mode of each class, they still are representative of their class.
Finally, as GANs map a latent noise space to the signal space, it is possible to explore the spectral manifold by interpolating between two noise vectors. Within a fixed class, it allows to generate spectra between two arbitrary points of the latent space. However, it is also possible to interpolate between two classes to generate intermediate spectra that do not necessarily belong to one specific class. This is illustrated in Fig. 4. There is a continuous progression between the origin and target vectors, which is especially interesting in the inter-class interpolation. The generator learns to perform realistic mixing of several materials, which is the reverse of the unmixing task. Dictionary learning, nearest-neighbors or reversibility approaches such as [9] could be used to retrieve the material mixing if an exhaustive panel of synthetic mixes has been generated.
Dataset | PaviaU | PaviaC | Botswana | Indian Pines | |||||
---|---|---|---|---|---|---|---|---|---|
Classifier | Augmentation | 3% (r) | 50% (s) | 3% (r) | 50% (s) | 3% (r) | 50% (s) | 3% (r) | 50% (s) |
NN 1D | 92.72 | 86.22 | 98.93 | 96.26 | 86.90 | 84.87 | 79.44 | 74.00 | |
GAN | 92.95 | 86.47 | 99.00 | 96.26 | 87.72 | 84.60 | 80.01 | 74.81 | |
ss-GAN | 93.12 | 87.20 | 98.93 | 96.70 | 88.40 | 85.27 | 80.42 | 74.58 |
5 Data augmentation
Considering that the synthesized hyperspectral samples are both realistic and diverse, we suggest to use the fake spectra to augment pre-existing hyperspectral datasets. We test this idea on several datasets: Indian Pines (aerial, rural), Pavia University (aerial, peri-urban), Pavia Center (aerial, urban) and Botswana (satellite, rural). Results in the supervised and semi-supervised settings are illustrated in Table 2. Augmenting the dataset with fake samples marginally increase the classification accuracy when the GAN is trained only on annotated samples. This is expected as the samples would hardly bring new information compared to the true samples. However, training the GAN in a semi-supervised fashion allows us to augment the dataset with fake samples that come from an approximation of the global distribution, including knowledge of how unlabeled samples look like. It therefore increases the model generalization ability, especially in the case where the training and testing set are disjoint.
It is worth noting that increasing drastically the number of fake samples does not increase further the classification accuracy and even degrades it beyond a certain point. We speculate that the introduction of too many approximative samples hurt the model’s classification ability.
6 Discussion
In this work, we presented a method based on Generative Adversarial Networks to generate an arbitrary large number of hyperspectral samples matching the distribution of any dataset, annotated or not. Through a data-driven analysis, we showed that the obtained spectra are plausible as they respect the statistical properties of the real samples. By interpolating between vectors in the latent space, we show that it is possible to synthesize any arbitrary combination of classes, i.e. to perform realistic spectral mixing. This is especially interesting as this could form the basis of data-driven unmixing techniques, e.g. by using a dictionary of synthetic spectra. Finally, we showed that incorporating synthetic samples can serve as a data augmentation strategy for hyperspectral datasets, with positive accuracy improvements on the Indian Pines, Pavia University, Pavia Center and Botswana datasets.
This opens the door to new possibilities in hyperspectral data synthesis and manipulation based on generative models, e.g. domain adaptation by learning the transfer function between two sensors, unmixing by disentangling spectra in the latent domain or hyperspectral data augmentation for deep learning purposes.
References
- [1] David A. van Dyk and Xiao-Li Meng, “The Art of Data Augmentation,” J. Comput. Graph. Stat., 2012.
- [2] A. Krizhevsky et al., “ImageNet Classification with Deep Convolutional Neural Networks,” in Advances in Neural Information Processing Systems (NIPS), 2012.
- [3] Y. Chen et al., “Deep Feature Extraction and Classification of Hyperspectral Images Based on Convolutional Neural Networks,” IEEE Transactions on Geoscience and Remote Sensing, vol. 54, no. 10, 2016.
- [4] K. Makantasis et al., “Deep supervised learning for hyperspectral data classification through convolutional neural networks,” in IEEE International Geoscience and Remote Sensing Symposium (IGARSS), 2015.
- [5] V. Slavkovikj et al., “Hyperspectral Image Classification with Convolutional Neural Networks,” in 23rd ACM International Conference on Multimedia, 2015.
- [6] H. Lee and H. Kwoon, “Contextual deep CNN based hyperspectral classification,” in IEEE International Geoscience and Remote Sensing Symposium (IGARSS), 2016.
- [7] L. Windrim et al., “Hyperspectral CNN Classification with Limited Training Samples,” arXiv:1611.09007 [cs], 2016.
- [8] J. Acquarelli et al., “Cnn and Data Augmentation for Spectral-Spatial Classification of Hyperspectral Images,” arXiv:1711.05512 [cs], 2017.
- [9] I. Gemp et al., “Inverting Variational Autoencoders for Improved Generative Accuracy,” in NIPS Workshop on Advances in Approximate Bayesian Inference, 2017.
- [10] I. Goodfellow et al., “Generative Adversarial Nets,” in Advances in Neural Information Processing Systems (NIPS), 2014.
- [11] M. Arjovsky et al., “Wasserstein Generative Adversarial Networks,” in Proceedings of the International Conference on Machine Learning (ICML), 2017.
- [12] I. Gulrajani et al., “Improved Training of Wasserstein GANs,” in Advances in Neural Information Processing Systems (NIPS), 2017.
- [13] A. L. Maas et al., “Rectifier nonlinearities improve neural network acoustic models,” in ICML Workshop on Deep Learning for Audio, Speech and Language Processing, 2013.
- [14] T. Salimans et al., “Improved Techniques for Training GANs,” in Advances in Neural Information Processing Systems (NIPS), 2016.