Achieving Explainability for Plant Disease Classification with Disentangled Variational Autoencoders

Abstract

Agricultural image recognition tasks are becoming increasingly dependent on deep learning (DL); however, despite the excellent performance of DL, it is difficult to comprehend the type of logic or features of the input image it uses during decision making. Knowing the logic or features is highly crucial for result verification, algorithm improvement, training data improvement, and knowledge extraction. However, the explanations from the current heatmap-based algorithms are insufficient for the abovementioned requirements. To address this, this paper details the development of a classification and explanation method based on a variational autoencoder (VAE) architecture, which can visualize the variations of the most important features by visualizing the generated images that correspond to the variations of those features. Using the PlantVillage dataset, an acceptable level of explainability was achieved without sacrificing the classification accuracy. The proposed method can also be extended to other crops as well as other image classification tasks. Further, application systems using this method for disease identification tasks, such as the identification of potato blackleg disease, potato virus Y, and other image classification tasks, are currently being developed.

arXiv comment55 pages, 24 figures, 2 tables

References in corpus (14)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Deep learning in agriculture: A survey
• Striving for Simplicity: The All Convolutional Net
• Explaining NonLinear Classification Decisions with Deep Taylor Decomposition
• Man is to Computer Programmer as Woman is to Homemaker? Debiasing Word Embeddings
• An open access repository of images on plant health to enable the development of mobile disease diagnostics
• Disentangling by Factorising
• Concept Whitening for Interpretable Image Recognition
• Understanding disentangling in $\beta$-VAE
• Diagnosing and Enhancing VAE Models
• Improving Variational Autoencoder with Deep Feature Consistent and Generative Adversarial Training
• Discriminative Regularization for Generative Models
• Why are deep nets reversible: A simple theory, with implications for training
• Interpretable Image Recognition with Hierarchical Prototypes