Res-WGAN: Image Classification for Plant Small-scale Datasets

Mi Jiaqi,Hao xia,Yang Si,Yang Xiande,Gao Wanlin,Li Minzan,Zheng Lihua,Minjuan Wang
DOI: https://doi.org/10.21203/rs.3.rs-31967/v1
2020-01-01
Abstract:Abstract Background: Artificial identification of rare plants is an important yet challenging problem in plant taxonomy. Although deep learning-based method can accurately predict rare plant category from training samples, accuracy requirements of only few experts are satisfied due to the small sample size. Thus, effective data augmentation is vital to improve the expressive power and robustness of deep learning models, especially for plant small-sample classification tasks. Different from traditional methods, a generative adversarial network (GAN) can mimic the distribution of primary data and produce similar but nonidentical samples. Data augmentation for automated classification of rare plant samples with GAN has not been studied for a long time. Result: In this study, we present a novel GAN model, referred to as residual Wasserstein GAN (Res-WGAN), for extending rare plant dataset efficiently. With a premise of minimized size of training parameters, residual block serves as elementary building block to increase the model’s expression ability. A loss function based on Wasserstein distance with gradient penalty is used to ensure diversity of output. Moreover, we draw on the super-resolution GAN (SRGAN) and consider perceptual loss into the function. Perceptual loss guarantees similarity between generated samples and original samples in high-dimensional features. Benefiting from these improvements, the expanded dataset improves the classification accuracy on the residual neural network (ResNet), a classic convolutional neural network, and restrains the overfitting phenomenon effectively by using transfer learning. More similar experiments are presented in this paper to prove the practicality of the model. Conclusions: Our proposed method produces better diversity and sharpness of generated samples and more accurate classification results than other data augmentation methods. In addition, a user study confirms that it is an ideal alternative strategy for small-scale rare plant identification. Developing a robust and effective small-scale plant classification method to replace expert testimony is highly relevant for agricultural automation development.
What problem does this paper attempt to address?