Leveraging Guided Backpropagation to Select Convolutional Neural Networks for Plant Classification

4Citations
Citations of this article
16Readers
Mendeley users who have this article in their library.
Get full text

Abstract

The development of state-of-the-art convolutional neural networks (CNN) has allowed researchers to perform plant classification tasks previously thought impossible and rely on human judgment. Researchers often develop complex CNN models to achieve better performances, introducing over-parameterization and forcing the model to overfit on a training dataset. The most popular process for evaluating overfitting in a deep learning model is using accuracy and loss curves. Train and loss curves may help understand the performance of a model but do not provide guidance on how the model could be modified to attain better performance. In this article, we analyzed the relation between the features learned by a model and its capacity and showed that a model with higher representational capacity might learn many subtle features that may negatively affect its performance. Next, we showed that the shallow layers of a deep learning model learn more diverse features than the ones learned by the deeper layers. Finally, we propose SSIM cut curve, a new way to select the depth of a CNN model by using the pairwise similarity matrix between the visualization of the features learned at different depths by using Guided Backpropagation. We showed that our proposed method could potentially pave a new way to select a better CNN model.

Cite

CITATION STYLE

APA

Mostafa, S., Mondal, D., Beck, M. A., Bidinosti, C. P., Henry, C. J., & Stavness, I. (2022). Leveraging Guided Backpropagation to Select Convolutional Neural Networks for Plant Classification. Frontiers in Artificial Intelligence, 5. https://doi.org/10.3389/frai.2022.871162

Register to see more suggestions

Mendeley helps you to discover research relevant for your work.

Already have an account?

Save time finding and organizing research with Mendeley

Sign up for free